fix: cot agent token usage is empty (#4474)

2024-05-17 14:45:20 +08:00
parent c2a8fa91b1
commit 528faceb35
2 changed files with 4 additions and 2 deletions
--- a/api/core/agent/cot_agent_runner.py
+++ b/api/core/agent/cot_agent_runner.py
@@ -121,7 +121,7 @@ class CotAgentRunner(BaseAgentRunner, ABC):
                raise ValueError("failed to invoke llm")
            
            usage_dict = {}
-            react_chunks = CotAgentOutputParser.handle_react_stream_output(chunks)
+            react_chunks = CotAgentOutputParser.handle_react_stream_output(chunks, usage_dict)
            scratchpad = AgentScratchpadUnit(
                agent_response='',
                thought='',