feat(ZepGraphMemoryUpdater): add platform display name mapping and logging enhancements.

This commit is contained in:
666ghj 2026-01-16 09:00:10 +08:00
parent ca24ee09d1
commit 56b8babf17

View file

@ -215,6 +215,12 @@ class ZepGraphMemoryUpdater:
# 批量发送大小(每个平台累积多少条后发送)
BATCH_SIZE = 5
# 平台名称映射(用于控制台显示)
PLATFORM_DISPLAY_NAMES = {
'twitter': '世界1',
'reddit': '世界2',
}
# 发送间隔(秒),避免请求过快
SEND_INTERVAL = 0.5
@ -261,6 +267,10 @@ class ZepGraphMemoryUpdater:
logger.info(f"ZepGraphMemoryUpdater 初始化完成: graph_id={graph_id}, batch_size={self.BATCH_SIZE}")
def _get_platform_display_name(self, platform: str) -> str:
"""获取平台的显示名称"""
return self.PLATFORM_DISPLAY_NAMES.get(platform.lower(), platform)
def start(self):
"""启动后台工作线程"""
if self._running:
@ -403,7 +413,8 @@ class ZepGraphMemoryUpdater:
self._total_sent += 1
self._total_items_sent += len(activities)
logger.info(f"成功批量发送 {len(activities)}{platform}活动到图谱 {self.graph_id}")
display_name = self._get_platform_display_name(platform)
logger.info(f"成功批量发送 {len(activities)}{display_name}活动到图谱 {self.graph_id}")
logger.debug(f"批量内容预览: {combined_text[:200]}...")
return
@ -433,7 +444,8 @@ class ZepGraphMemoryUpdater:
with self._buffer_lock:
for platform, buffer in self._platform_buffers.items():
if buffer:
logger.info(f"发送{platform}平台剩余的 {len(buffer)} 条活动")
display_name = self._get_platform_display_name(platform)
logger.info(f"发送{display_name}平台剩余的 {len(buffer)} 条活动")
self._send_batch_activities(buffer, platform)
# 清空所有缓冲区
for platform in self._platform_buffers: