fix input_messages

Author: Dylan Huang

tests/pytest/test_pytest_async.py

@@ -26,8 +26,10 @@ async def test_pytest_async(rows: list[EvaluationRow]) -> list[EvaluationRow]:
 @evaluation_test(
     input_messages=[
         [
-            Message(role="user", content="What is the capital of France?"),
-        ],
+            [
+                Message(role="user", content="What is the capital of France?"),
+            ],
+        ]
     ],
     completion_params=[{"model": "accounts/fireworks/models/kimi-k2-instruct"}],
     mode="pointwise",

tests/pytest/test_pytest_mcp_config.py

@@ -89,7 +89,7 @@ def read(self, row_id: str | None = None) -> list[EvaluationRow]:
     def eval_fn(row: EvaluationRow) -> EvaluationRow:
         return row
 
-    await eval_fn(input_messages=input_messages[0], completion_params=completion_params_list[0])  # pyright: ignore[reportCallIssue]
+    await eval_fn(input_messages=input_messages, completion_params=completion_params_list[0])  # pyright: ignore[reportCallIssue]
 
     # ensure that the row has tools that were set during AgentRolloutProcessor
     assert len(rollouts) == 1

tests/pytest/test_pytest_propagate_error.py

@@ -60,7 +60,7 @@ def eval_fn(row: EvaluationRow) -> EvaluationRow:
 
     # Manually invoke all parameter combinations within a single test
     for params in completion_params_list:
-        await eval_fn(input_messages=input_messages[0], completion_params=params)  # pyright: ignore[reportCallIssue]
+        await eval_fn(input_messages=input_messages, completion_params=params)  # pyright: ignore[reportCallIssue]
 
     # assert that the status of eval_metadata.status is "error"
     assert len(rollouts) == 5