Skip to content

Commit 96d02b5

Browse files
committed
fix cut message
1 parent 01014ec commit 96d02b5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/agent/custom_massage_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def cut_messages(self):
9595
"""Get current message list, potentially trimmed to max tokens"""
9696
diff = self.history.total_tokens - self.max_input_tokens
9797
i = 1 # start from 1 to keep system message in history
98-
while diff > 0:
98+
while diff > 0 and i < len(self.history.messages):
9999
self.history.remove_message(i)
100100
diff = self.history.total_tokens - self.max_input_tokens
101101
i += 1

tests/test_browser_use.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ async def test_browser_use_custom_v2():
249249

250250
llm = utils.get_llm_model(
251251
provider="deepseek",
252-
model_name="deepseek-chat",
252+
model_name="deepseek-reasoner",
253253
temperature=0.8
254254
)
255255

0 commit comments

Comments
 (0)