From 3b13046745438f12a472c37b7c69459897a5d5bb Mon Sep 17 00:00:00 2001 From: lfy79001 <843265183@qq.com> Date: Sat, 16 Mar 2024 01:40:41 +0800 Subject: [PATCH] add claude3 agent code --- mm_agents/agent.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/mm_agents/agent.py b/mm_agents/agent.py index dcf0919..7dd45b3 100644 --- a/mm_agents/agent.py +++ b/mm_agents/agent.py @@ -534,17 +534,17 @@ class PromptAgent: ) if response.status_code != 200: - if response.json()['error']['code'] == "context_length_exceeded": - logger.error("Context length exceeded. Retrying with a smaller context.") - payload["messages"] = payload["messages"][-1:] - retry_response = requests.post( - "https://api.anthropic.com/v1/messages", - headers=headers, - json=payload - ) - if retry_response.status_code != 200: - logger.error("Failed to call LLM: " + retry_response.text) - return "" + # if response.json()['error']['code'] == "context_length_exceeded": + # logger.error("Context length exceeded. Retrying with a smaller context.") + # payload["messages"] = payload["messages"][-1:] + # retry_response = requests.post( + # "https://api.anthropic.com/v1/messages", + # headers=headers, + # json=payload + # ) + # if retry_response.status_code != 200: + # logger.error("Failed to call LLM: " + retry_response.text) + # return "" logger.error("Failed to call LLM: " + response.text) time.sleep(5)