hotfix

Author: Dylan Huang

eval_protocol/pytest/default_agent_rollout_processor.py

@@ -58,7 +58,7 @@ async def call_agent(self) -> str:
         self.append_message_and_log(message)
         if message.tool_calls:
             # Create tasks for all tool calls to run them in parallel
-            tool_tasks = []
+            tool_tasks: List[asyncio.Task[tuple[str, List[TextContent]]]] = []
             for tool_call in message.tool_calls:
                 tool_call_id = tool_call.id
                 tool_name = tool_call.function.name
@@ -70,7 +70,7 @@ async def call_agent(self) -> str:
                 tool_tasks.append(task)
 
             # Execute all tool calls in parallel
-            tool_results: List[tuple[str, List[TextContent]]] = await asyncio.gather(*tool_tasks)
+            tool_results = await asyncio.gather(*tool_tasks)
 
             # Add all tool results to messages (they will be in the same order as tool_calls)
             for tool_call, (tool_call_id, content) in zip(message.tool_calls, tool_results):
@@ -126,7 +126,7 @@ async def default_agent_rollout_processor(
     async def process_row(row: EvaluationRow) -> EvaluationRow:
         """Process a single row with agent rollout."""
         agent = Agent(
-            model=config.completion_params.model, row=row, config_path=config.mcp_config_path, logger=config.logger
+            model=config.completion_params["model"], row=row, config_path=config.mcp_config_path, logger=config.logger
         )
         try:
             await agent.setup()