From 01014ecfd009682e0b83289a77938bc51922276f Mon Sep 17 00:00:00 2001
From: vincent <vincent19901026@gmail.com>
Date: Sat, 25 Jan 2025 23:53:58 +0800
Subject: [PATCH] fix context len

---
 src/agent/custom_agent.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/agent/custom_agent.py b/src/agent/custom_agent.py
index df307f0..e501d0c 100644
--- 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