Provide Gemini with the list of IDs in correction request

This commit is contained in:
sebastien committed 2026-10-10 17:19:06 +02:00
1 parent f978310c4c
commit 2d0f7a219f
9 files changed
+242 -9

No files matched your search

+2 -1
View File
@@ -1775,8 +1775,9 @@ class StandardCliTests(unittest.TestCase):
args,
api_client=Mock(),
)
with patch.object(module.prompting, "make_prompt", return_value="prompt"):
with patch.object(module.prompting, "make_prompt", return_value="prompt") as make_prompt:
self.assertEqual(module.run_configured(args), 0)
make_prompt.assert_called_once_with(evaluation, "Ex 1", expected_ids=["01"])
lines = (evaluation / "batch_requests_flash.jsonl").read_text().splitlines()
self.assertEqual(len(lines), 1)
self.assertEqual(json.loads(lines[0])["key"], str(image))