lzy/sci-gui-agent-benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add claude3 agent code

Browse Source
This commit is contained in:
lfy79001
2024-03-16 01:40:41 +08:00
parent 017dde8966
commit 3b13046745
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
mm_agents/agent.py
View File

@@ -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)