fix: mypy linter

2026-05-03 17:08:03 +08:00 · 2025-01-08 21:11:42 +08:00
parent fb309462ad
commit b7d168ac59
27 changed files with 62 additions and 45 deletions
--- a/api/core/agent/cot_agent_runner.py
+++ b/api/core/agent/cot_agent_runner.py
@ -119,7 +119,7 @@ class CotAgentRunner(BaseAgentRunner, ABC):
                callbacks=[],
            )

-            usage_dict = {}
+            usage_dict: dict[str, Optional[LLMUsage]] = {}
            react_chunks = CotAgentOutputParser.handle_react_stream_output(chunks, usage_dict)
            scratchpad = AgentScratchpadUnit(
                agent_response="",