Yusaki/MiroFish
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
666ghj 2026-01-16 09:00:10 +08:00
parent ca24ee09d1
commit 56b8babf17
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
backend/app/services/zep_graph_memory_updater.py
View file

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