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

feat(workflow): store token usage in message table

Browse Source
This commit is contained in:
Eternity
2026-01-27 18:52:51 +08:00
parent c5a794f1b5
commit 8fb9e779a6
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -528,7 +528,8 @@ class WorkflowService:
self.conversation_service.add_message( self.conversation_service.add_message(
conversation_id=conversation_id_uuid, conversation_id=conversation_id_uuid,
role=message["role"], role=message["role"],
content=message["content"] content=message["content"],
meta_data=None if message["role"] == "user" else {"usage": token_usage}
) )
logger.info(f"Workflow Run Success, " logger.info(f"Workflow Run Success, "
f"execution_id: {execution.execution_id}, message count: {len(final_messages)}") f"execution_id: {execution.execution_id}, message count: {len(final_messages)}")
@@ -678,7 +679,8 @@ class WorkflowService:
self.conversation_service.add_message( self.conversation_service.add_message(
conversation_id=conversation_id_uuid, conversation_id=conversation_id_uuid,
role=message["role"], role=message["role"],
content=message["content"] content=message["content"],
meta_data=None if message["role"] == "user" else {"usage": token_usage}
) )
logger.info(f"Workflow Run Success, " logger.info(f"Workflow Run Success, "
f"execution_id: {execution.execution_id}, message count: {len(final_messages)}") f"execution_id: {execution.execution_id}, message count: {len(final_messages)}")