From be37caa15a4cd8e6ba3aabbf6830404ac0b125a1 Mon Sep 17 00:00:00 2001 From: Yan Wang Date: Mon, 20 Jan 2025 16:37:51 -0800 Subject: [PATCH] [Cursor] Fix test_llm_api.py to properly mock token tracking --- tests/test_llm_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_llm_api.py b/tests/test_llm_api.py index 868bec8..12e387f 100644 --- a/tests/test_llm_api.py +++ b/tests/test_llm_api.py @@ -204,10 +204,10 @@ def test_query_azure(self, mock_create_client): @patch('tools.llm_api.create_llm_client') def test_query_deepseek(self, mock_create_client): mock_create_client.return_value = self.mock_openai_client - response = query_llm("Test prompt", provider="deepseek", model="gpt-4o") + response = query_llm("Test prompt", provider="deepseek", model="deepseek-chat") self.assertEqual(response, "Test OpenAI response") self.mock_openai_client.chat.completions.create.assert_called_once_with( - model="gpt-4o", + model="deepseek-chat", messages=[{"role": "user", "content": [{"type": "text", "text": "Test prompt"}]}], temperature=0.7 )