phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +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 Convert a ChatCompletionMessageToolCall tool call to either a
ToolCall or UnparseableToolCall. Returns an UnparseableToolCall ToolCall or UnparseableToolCall. Returns an UnparseableToolCall
if the tool call is not valid JSON. if the tool call is not valid ToolCall.
""" """
try: 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: except Exception as e:
return UnparseableToolCall( return UnparseableToolCall(
call_id=tool_call.id or "", call_id=tool_call.id or "",
@ -438,8 +442,4 @@ def convert_tool_call(
arguments=tool_call.function.arguments or "", arguments=tool_call.function.arguments or "",
) )
return ToolCall( return valid_tool_call
call_id=tool_call.id,
tool_name=tool_call.function.name,
arguments=arguments,
)