fix: patch reasoning models (#2703)

Co-authored-by: Charles Packer <packercharles@gmail.com>
This commit is contained in:
Kevin Lin
2025-06-11 17:20:04 -07:00
committed by GitHub
parent e9ee2b0766
commit 58c4448235

View File

@@ -60,8 +60,11 @@ class OpenAIStreamingInterface:
def get_tool_call_object(self) -> ToolCall:
"""Useful for agent loop"""
function_name = self.last_flushed_function_name if self.last_flushed_function_name else self.function_name_buffer
tool_call_id = self.last_flushed_function_id if self.last_flushed_function_id else self.function_id_buffer
if not tool_call_id:
raise ValueError("No tool call ID available")
return ToolCall(
id=self.last_flushed_function_id,
id=tool_call_id,
function=FunctionCall(arguments=self.current_function_arguments, name=function_name),
)