From 449dcca6d67bdfc3c6d8ed7ba357e3003c72c5de Mon Sep 17 00:00:00 2001 From: Daniel McKnight Date: Wed, 18 Dec 2024 16:07:12 -0800 Subject: [PATCH] Fix test error --- tests/models/api/test_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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