Fix test error

NeonGeckoCom · Dec 19, 2024 · 449dcca · 449dcca
1 parent 9dd2ac2
commit 449dcca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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