Provide Gemini with the list of IDs in correction request
This commit is contained in:
1 parent
f978310c4c
commit
2d0f7a219f
9 files changed
+242
-9
No files matched your search
@@ -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))
|
||||
|
||||
Reference in new issue
Block a user