Skip to content

Commit

Permalink
fix: add function_name in tool_calls payload if it's missing (#243)
Browse files Browse the repository at this point in the history
add function_name in payload if it's missing
  • Loading branch information
yzld2002 authored Sep 23, 2024
1 parent fe87e81 commit 93be51a
Showing 1 changed file with 12 additions and 6 deletions.
18 changes: 12 additions & 6 deletions vision_agent/tools/tool_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -208,19 +208,25 @@ def _call_post(
if files:
files_in_b64 = [(file[0], b64encode(file[1]).decode("utf-8")) for file in files]
try:
if files is not None:
response = session.post(url, data=payload, files=files)
else:
response = session.post(url, json=payload)

# make sure function_name is in the payload so we can display it
tool_call_trace_payload = (
payload
if "function_name" in payload
else {**payload, **{"function_name": function_name}}
)
tool_call_trace = ToolCallTrace(
endpoint_url=url,
request=payload,
request=tool_call_trace_payload,
response={},
error=None,
files=files_in_b64,
)

if files is not None:
response = session.post(url, data=payload, files=files)
else:
response = session.post(url, json=payload)

if response.status_code != 200:
tool_call_trace.error = Error(
name="RemoteToolCallFailed",
Expand Down

0 comments on commit 93be51a

Please sign in to comment.