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

Merge pull request #377 from SuanmoSuanyangTechnology/fix/workflow-memory-write

Browse Source 
fix(workflow): align token usage fields and relax memory write
This commit is contained in:
Mark
2026-02-09 20:22:35 +08:00
committed by GitHub
parent ac6d9a39ec d57e3b3f64
commit 55e2baa865
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/app/core/workflow/nodes/memory/config.py
View File

@@ -45,7 +45,7 @@ class MemoryReadNodeConfig(BaseNodeConfig):
class MemoryWriteNodeConfig(BaseNodeConfig): class MemoryWriteNodeConfig(BaseNodeConfig):
message: str = Field( message: str = Field(
... default=""
) )
messages: list[MessageConfig] = Field( messages: list[MessageConfig] = Field(

4
api/app/core/workflow/nodes/parameter_extractor/node.py
View File

@@ -32,8 +32,8 @@ class ParameterExtractorNode(BaseNode):
usage = self.response_metadata.get('token_usage') usage = self.response_metadata.get('token_usage')
if usage: if usage:
return { return {
"prompt_tokens": usage.get('prompt_tokens', 0), "prompt_tokens": usage.get('input_tokens', 0),
"completion_tokens": usage.get('completion_tokens', 0), "completion_tokens": usage.get('output_tokens', 0),
"total_tokens": usage.get('total_tokens', 0) "total_tokens": usage.get('total_tokens', 0)
} }
return None return None

4
api/app/core/workflow/nodes/question_classifier/node.py
View File

@@ -32,8 +32,8 @@ class QuestionClassifierNode(BaseNode):
usage = self.response_metadata.get('token_usage') usage = self.response_metadata.get('token_usage')
if usage: if usage:
return { return {
"prompt_tokens": usage.get('prompt_tokens', 0), "prompt_tokens": usage.get('input_tokens', 0),
"completion_tokens": usage.get('completion_tokens', 0), "completion_tokens": usage.get('output_tokens', 0),
"total_tokens": usage.get('total_tokens', 0) "total_tokens": usage.get('total_tokens', 0)
} }
return None return None