fix cut message

This commit is contained in:
vincent
2025-01-26 00:01:16 +08:00
parent 01014ecfd0
commit 96d02b5224
2 changed files with 2 additions and 2 deletions

View File

@@ -95,7 +95,7 @@ class CustomMassageManager(MessageManager):
"""Get current message list, potentially trimmed to max tokens"""
diff = self.history.total_tokens - self.max_input_tokens
i = 1 # start from 1 to keep system message in history
while diff > 0:
while diff > 0 and i < len(self.history.messages):
self.history.remove_message(i)
diff = self.history.total_tokens - self.max_input_tokens
i += 1

View File

@@ -249,7 +249,7 @@ async def test_browser_use_custom_v2():
llm = utils.get_llm_model(
provider="deepseek",
model_name="deepseek-chat",
model_name="deepseek-reasoner",
temperature=0.8
)