Skip to content

Commit 52b46a7

Browse files
author
Dylan Huang
committed
fix agent rollout
1 parent ff63b04 commit 52b46a7

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

eval_protocol/pytest/default_agent_rollout_processor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@ async def _call_model(
7575
self, messages: list[Message], tools: Optional[list[ChatCompletionToolParam]]
7676
) -> ChatCompletionMessage:
7777
messages = [message.model_dump() if hasattr(message, "model_dump") else message for message in messages]
78+
tools = [{"function": tool["function"].model_dump(), "type": "function"} for tool in tools] if tools else []
7879
response = await self._policy._make_llm_call(messages=messages, tools=tools)
7980
return response["choices"][0]["message"]
8081

0 commit comments

Comments
 (0)