phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-10 04:08:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Ensure a tool call can be converted before adding to buffer

Browse source 
Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-02-14 23:27:10 -05:00
parent ab2b46e528
commit a67a575b27
No known key found for this signature in database
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/providers/utils/inference/openai_compat.py
View file

@ -427,10 +427,14 @@ def convert_tool_call(
"""
Convert a ChatCompletionMessageToolCall tool call to either a
ToolCall or UnparseableToolCall. Returns an UnparseableToolCall
if the tool call is not valid JSON.
if the tool call is not valid ToolCall.
"""
try:
arguments = json.loads(tool_call.function.arguments)
valid_tool_call = ToolCall(
call_id=tool_call.id,
tool_name=tool_call.function.name,
arguments=json.loads(tool_call.function.arguments),
)
except Exception as e:
return UnparseableToolCall(
call_id=tool_call.id or "",
@ -438,8 +442,4 @@ def convert_tool_call(
arguments=tool_call.function.arguments or "",
)
return ToolCall(
call_id=tool_call.id,
tool_name=tool_call.function.name,
arguments=arguments,
)
return valid_tool_call