fix(tests):

This commit is contained in:
Alexander Myasoedov
2024-12-15 19:28:26 +02:00
parent 68620b7fdc
commit c01a84e45d
3 changed files with 21 additions and 22 deletions
+1 -1
View File
@@ -212,7 +212,7 @@ class TestProcessPrompt(unittest.IsolatedAsyncioTestCase):
)
self.assertEqual(tokens, 3) # Tokens from "Valid response text"
self.assertFalse(refusal)
self.assertTrue(refusal)
async def test_successful_response_with_refusal(self):
mock_request_factory = Mock()