Skip to content

Commit 1c63ce2

Browse files
authored
[Cursor] Fix test_llm_api.py to properly mock token tracking (#32)
1 parent e709428 commit 1c63ce2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_llm_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,10 +204,10 @@ def test_query_azure(self, mock_create_client):
204204
@patch('tools.llm_api.create_llm_client')
205205
def test_query_deepseek(self, mock_create_client):
206206
mock_create_client.return_value = self.mock_openai_client
207-
response = query_llm("Test prompt", provider="deepseek", model="gpt-4o")
207+
response = query_llm("Test prompt", provider="deepseek", model="deepseek-chat")
208208
self.assertEqual(response, "Test OpenAI response")
209209
self.mock_openai_client.chat.completions.create.assert_called_once_with(
210-
model="gpt-4o",
210+
model="deepseek-chat",
211211
messages=[{"role": "user", "content": [{"type": "text", "text": "Test prompt"}]}],
212212
temperature=0.7
213213
)

0 commit comments

Comments
 (0)