APEX/BrowserUse_and_ComputerUse_skills
5
0
Fork 0
Code Issues 6 Pull requests 1 Projects 1 Releases Packages Wiki Activity Actions

fix: handle dict and object tool_calls in agent loop

Browse source 
vLLM's ToolCallTranslator returns tool_calls as dicts, while
OpenAI API returns them as objects with .id, .function.name etc.
Normalize both formats in the agent loop.
This commit is contained in:
dmahan93 2026-03-09 21:21:49 -05:00 • committed by teknium1
parent 1f9e7cd659
commit 93333387d6
1 changed files with 31 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

45
environments/agent_loop.py
View file

@ -279,21 +279,32 @@ class HermesAgentLoop:
pass # Fall through to no tool calls pass # Fall through to no tool calls
if assistant_msg.tool_calls: if assistant_msg.tool_calls:
# Normalize tool calls to dicts — they may come as objects
# (OpenAI API) or dicts (vLLM ToolCallTranslator).
def _tc_to_dict(tc):
if isinstance(tc, dict):
return {
"id": tc.get("id", f"call_{uuid.uuid4().hex[:8]}"),
"type": "function",
"function": {
"name": tc.get("function", {}).get("name", tc.get("name", "")),
"arguments": tc.get("function", {}).get("arguments", tc.get("arguments", "{}")),
},
}
return {
"id": tc.id,
"type": "function",
"function": {
"name": tc.function.name,
"arguments": tc.function.arguments,
},
}
# Build the assistant message dict for conversation history # Build the assistant message dict for conversation history
msg_dict: Dict[str, Any] = { msg_dict: Dict[str, Any] = {
"role": "assistant", "role": "assistant",
"content": assistant_msg.content or "", "content": assistant_msg.content or "",
"tool_calls": [ "tool_calls": [_tc_to_dict(tc) for tc in assistant_msg.tool_calls],
{
"id": tc.id,
"type": "function",
"function": {
"name": tc.function.name,
"arguments": tc.function.arguments,
},
}
for tc in assistant_msg.tool_calls
],
} }
# Preserve reasoning_content for multi-turn chat template handling # Preserve reasoning_content for multi-turn chat template handling
@ -306,8 +317,13 @@ class HermesAgentLoop:
# Execute each tool call via hermes-agent's dispatch # Execute each tool call via hermes-agent's dispatch
for tc in assistant_msg.tool_calls: for tc in assistant_msg.tool_calls:
tool_name = tc.function.name # Handle both object (OpenAI) and dict (vLLM) formats
tool_args_raw = tc.function.arguments if isinstance(tc, dict):
tool_name = tc.get("function", {}).get("name", tc.get("name", ""))
tool_args_raw = tc.get("function", {}).get("arguments", tc.get("arguments", "{}"))
else:
tool_name = tc.function.name
tool_args_raw = tc.function.arguments
# Validate tool name # Validate tool name
if tool_name not in self.valid_tool_names: if tool_name not in self.valid_tool_names:
@ -418,10 +434,11 @@ class HermesAgentLoop:
pass pass
# Add tool response to conversation # Add tool response to conversation
tc_id = tc.get("id", "") if isinstance(tc, dict) else tc.id
messages.append( messages.append(
{ {
"role": "tool", "role": "tool",
"tool_call_id": tc.id, "tool_call_id": tc_id,
"content": tool_result, "content": tool_result,
} }
) )