diff --git a/tests/models/api/test_llm.py b/tests/models/api/test_llm.py index 9db702a..acf5f2b 100644 --- a/tests/models/api/test_llm.py +++ b/tests/models/api/test_llm.py @@ -72,7 +72,7 @@ def test_llm_request(self): self.assertEqual(len(valid_request.history), len(test_history)) self.assertEqual(len(valid_request.to_completion_kwargs()['messages']), 2 * valid_request.max_history + 2) - self.assertEqual(valid_request.to_completion_kwargs()['messages'][-1], + self.assertEqual(valid_request.to_completion_kwargs()['messages'][-1]['content'], test_query) # Valid explicit streaming