fix: allow a single validation model
This commit is contained in:
@@ -14,13 +14,21 @@ class ValidationHarnessTests(unittest.TestCase):
|
||||
self.assertEqual(validators, ["validator-a", "validator-b"])
|
||||
self.assertTrue(harness)
|
||||
|
||||
def test_harness_requires_two_distinct_validators(self):
|
||||
body = api.ChatRequest(primary_model="primary", validator_models=["validator-a"], harness=True)
|
||||
def test_harness_requires_one_distinct_validator(self):
|
||||
body = api.ChatRequest(primary_model="primary", validator_models=[], harness=True)
|
||||
with patch.object(api, "_require_installed_model", side_effect=lambda name: name):
|
||||
with self.assertRaises(api.HTTPException) as context:
|
||||
api._harness_models(body)
|
||||
self.assertEqual(context.exception.status_code, 400)
|
||||
|
||||
def test_harness_accepts_one_validator(self):
|
||||
body = api.ChatRequest(primary_model="primary", validator_models=["validator-a"], harness=True)
|
||||
with patch.object(api, "_require_installed_model", side_effect=lambda name: name):
|
||||
primary, validators, harness = api._harness_models(body)
|
||||
self.assertEqual(primary, "primary")
|
||||
self.assertEqual(validators, ["validator-a"])
|
||||
self.assertTrue(harness)
|
||||
|
||||
def test_chat_route_returns_one_compiled_answer(self):
|
||||
body = api.ChatRequest(
|
||||
primary_model="primary",
|
||||
|
||||
Reference in New Issue
Block a user