fix context len

browser-use · Jan 25, 2025 · 01014ec · 01014ec
1 parent 2f923c5
commit 01014ec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/agent/custom_agent.py b/src/agent/custom_agent.py
@@ -97,6 +97,8 @@ def __init__(
         )
         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