Skip to content

Commit e1733f9

Browse files
authored
Claude support
Claude support
2 parents 65247ad + 58b3220 commit e1733f9

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

interpreter/core/llm/run_tool_calling_llm.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,22 @@ def run_tool_calling_llm(llm, request_params):
5858

5959
request_params["messages"] = [m for m in request_params["messages"] if m != None]
6060

61+
new_messages = []
62+
for i, message in enumerate(request_params["messages"]):
63+
new_messages.append(message)
64+
if "tool_calls" in message:
65+
tool_call_id = message["tool_calls"][0]["id"]
66+
if not any(
67+
m
68+
for m in request_params["messages"]
69+
if m.get("role") == "function" and m.get("tool_call_id") == tool_call_id
70+
):
71+
new_messages.append(
72+
{"role": "tool", "tool_call_id": tool_call_id, "content": ""}
73+
)
74+
75+
request_params["messages"] = new_messages
76+
6177
# Add OpenAI's recommended function message
6278
# request_params["messages"][0][
6379
# "content"

0 commit comments

Comments
 (0)