fix context len

2025-01-25 23:53:58 +08:00
parent 2f923c50d6
commit 01014ecfd0
1 changed files with 2 additions and 0 deletions
--- a/src/agent/custom_agent.py
+++ b/src/agent/custom_agent.py
@@ -97,6 +97,8 @@ class CustomAgent(Agent):
        )
        if self.llm.model_name in ["deepseek-reasoner"]:
            self.use_function_calling = False
+            # TODO: deepseek-reasoner only support 64000 context
+            self.max_input_tokens = 64000
        else:
            self.use_function_calling = True
        self.add_infos = add_infos