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

feat(muti agent): non-streaming output sub-nodes do not record the generated content of the conversation

Browse Source
This commit is contained in:
谢俊男
2026-01-15 11:55:00 +08:00
parent ea9a36a60a
commit 59f24fb5b4
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

@@ -245,7 +245,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
) -> Dict[str, Any]: ) -> Dict[str, Any]:
"""执行试运行（使用 LangChain Agent） """执行试运行（使用 LangChain Agent）
@@ -435,7 +436,7 @@ class DraftRunService:
elapsed_time = time.time() - start_time elapsed_time = time.time() - start_time
# 8. 保存会话消息 # 8. 保存会话消息
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

@@ -1327,7 +1327,8 @@ class MultiAgentOrchestrator:
web_search=web_search, web_search=web_search,
memory=memory, memory=memory,
storage_type=storage_type, storage_type=storage_type,
user_rag_memory_id=user_rag_memory_id user_rag_memory_id=user_rag_memory_id,
sub_agent=True
) )
return result return result