fix context len

vvincent1234 · vvincent1234 · commit 01014ecfd009 · 2025-01-25T23:53:58.000+08:00
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

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,8 @@ def __init__(`
`97`	`97`	`)`
`98`	`98`	`if self.llm.model_name in ["deepseek-reasoner"]:`
`99`	`99`	`self.use_function_calling = False`
	`100`	`+ # TODO: deepseek-reasoner only support 64000 context`
	`101`	`+ self.max_input_tokens = 64000`
`100`	`102`	`else:`
`101`	`103`	`self.use_function_calling = True`
`102`	`104`	`self.add_infos = add_infos`