fix test

mayinghan · mayinghan · commit 2d6e64e183dc · 2025-10-28T12:05:14.000-07:00
diff --git a/tests/pytest/helper/gsm8k_to_evaluation_row.py b/tests/pytest/helper/gsm8k_to_evaluation_row.py
@@ -6,7 +6,7 @@
 def gsm8k_to_evaluation_row(data: List[Dict[str, Any]]) -> List[EvaluationRow]:
     return [
         EvaluationRow(
-            messages=[Message(role="user", content=row["user_query"])], ground_truth=row["ground_truth_for_eval"]
+            messages=[Message(**msg) for msg in row["messages"]], ground_truth=row["ground_truth_for_eval"]
         )
         for row in data
     ]

Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,7 @@`
`6`	`6`	`def gsm8k_to_evaluation_row(data: List[Dict[str, Any]]) -> List[EvaluationRow]:`
`7`	`7`	`return [`
`8`	`8`	`EvaluationRow(`
`9`		`- messages=[Message(role="user", content=row["user_query"])], ground_truth=row["ground_truth_for_eval"]`
	`9`	`+ messages=[Message(**msg) for msg in row["messages"]], ground_truth=row["ground_truth_for_eval"]`
`10`	`10`	`)`
`11`	`11`	`for row in data`
`12`	`12`	`]`