adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] converfsation message duplicate entry problem

Browse Source
This commit is contained in:
Mark
2026-01-05 19:03:28 +08:00
parent 43dd31d0c9
commit ebdf4e4c5e
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/app/services/draft_run_service.py
View File

@@ -454,7 +454,8 @@ class DraftRunService:
storage_type: Optional[str] = None, storage_type: Optional[str] = None,
user_rag_memory_id: Optional[str] = None, user_rag_memory_id: Optional[str] = None,
web_search: bool = True, # 布尔类型默认值 web_search: bool = True, # 布尔类型默认值
memory: bool = True # 布尔类型默认值 memory: bool = True, # 布尔类型默认值
sub_agent: bool = False # 是否是作为子Agent运行
) -> AsyncGenerator[str, None]: ) -> AsyncGenerator[str, None]:
"""执行试运行（流式返回，使用 LangChain Agent） """执行试运行（流式返回，使用 LangChain Agent）
@@ -619,7 +620,7 @@ class DraftRunService:
elapsed_time = time.time() - start_time elapsed_time = time.time() - start_time
# 10. 保存会话消息 # 10. 保存会话消息
if agent_config.memory and agent_config.memory.get("enabled"): if not sub_agent and agent_config.memory and agent_config.memory.get("enabled"):
await self._save_conversation_message( await self._save_conversation_message(
conversation_id=conversation_id, conversation_id=conversation_id,
user_message=message, user_message=message,

3
api/app/services/multi_agent_orchestrator.py
View File

@@ -1267,7 +1267,8 @@ class MultiAgentOrchestrator:
storage_type=storage_type, storage_type=storage_type,
user_rag_memory_id=user_rag_memory_id, user_rag_memory_id=user_rag_memory_id,
web_search=web_search, web_search=web_search,
memory=memory memory=memory,
sub_agent=True
): ):
yield event yield event