Skip to content

Commit 19e8272

Browse files
author
Dylan Huang
committed
fix test_pytest_default_agent_rollout_processor.py
1 parent 7291feb commit 19e8272

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

eval_protocol/pytest/default_agent_rollout_processor.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -75,11 +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]
79-
response = await self._policy._make_llm_call(
80-
messages=messages,
81-
tools=tools,
82-
)
78+
response = await self._policy._make_llm_call(messages=messages, tools=tools)
8379
return response["choices"][0]["message"]
8480

8581
async def _execute_tool_call(self, tool_call_id: str, tool_name: str, tool_args_dict: dict) -> tuple[str, str]:

0 commit comments

Comments
 (0)