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

fix(workflow): persist citations in conversation message meta_data

Browse Source
This commit is contained in:
wxy
2026-04-03 14:36:43 +08:00
parent 5db59bc9cf
commit d1f0258065
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/app/services/workflow_service.py
View File

@@ -728,12 +728,15 @@ class WorkflowService:
filtered_citations = ( filtered_citations = (
citations if isinstance(citation_cfg, dict) and citation_cfg.get("enabled") else [] citations if isinstance(citation_cfg, dict) and citation_cfg.get("enabled") else []
) )
assistant_meta = {"usage": token_usage, "audio_url": None}
if filtered_citations:
assistant_meta["citations"] = filtered_citations
self.conversation_service.add_message( self.conversation_service.add_message(
message_id=message_id, message_id=message_id,
conversation_id=conversation_id_uuid, conversation_id=conversation_id_uuid,
role="assistant", role="assistant",
content=assistant_message, content=assistant_message,
meta_data={"usage": token_usage, "audio_url": None, "citations": filtered_citations} meta_data=assistant_meta
) )
self.update_execution_status( self.update_execution_status(
execution.execution_id, execution.execution_id,
@@ -923,12 +926,15 @@ class WorkflowService:
filtered_citations = ( filtered_citations = (
citations if isinstance(citation_cfg, dict) and citation_cfg.get("enabled") else [] citations if isinstance(citation_cfg, dict) and citation_cfg.get("enabled") else []
) )
assistant_meta = {"usage": token_usage, "audio_url": None}
if filtered_citations:
assistant_meta["citations"] = filtered_citations
self.conversation_service.add_message( self.conversation_service.add_message(
message_id=message_id, message_id=message_id,
conversation_id=conversation_id_uuid, conversation_id=conversation_id_uuid,
role="assistant", role="assistant",
content=assistant_message, content=assistant_message,
meta_data={"usage": token_usage, "audio_url": None, "citations": filtered_citations} meta_data=assistant_meta
) )
self.update_execution_status( self.update_execution_status(
execution.execution_id, execution.execution_id,