Support evaluation-local configuration and names overrides
This commit is contained in:
1 parent
0bb6f7d97c
commit
d4404e4c27
36 files changed
+409
-184
No files matched your search
@@ -153,7 +153,7 @@ class NameGridTests(unittest.TestCase):
|
||||
self.assertEqual(len(requests), 4)
|
||||
self.assertEqual(requests[0].kwargs['model'], 'custom-name-model')
|
||||
self.assertEqual(requests[0].kwargs['config'].thinking_config.thinking_level.value, 'MEDIUM')
|
||||
self.assertEqual(requests[1].kwargs['model'], labels.MODEL_ID)
|
||||
self.assertEqual(requests[1].kwargs['model'], labels.config.MODEL_FOR_LABEL_ID)
|
||||
for request in requests[1:3]:
|
||||
text = request.kwargs['contents'][0].parts[1].text
|
||||
self.assertNotIn('Alice', text)
|
||||
|
||||
Reference in new issue
Block a user