Formatting (#2750)

tests/llms/test_deepseek.py

@@ -31,12 +31,12 @@ def test_deepseek_llm_base_url():
     # case3: with config.deepseek_base_url
     config_base_url = "https://api.config.com/v1/"
     config = BaseLlmConfig(
-        model="deepseek-chat", 
-        temperature=0.7, 
-        max_tokens=100, 
-        top_p=1.0, 
-        api_key="api_key", 
-        deepseek_base_url=config_base_url
+        model="deepseek-chat",
+        temperature=0.7,
+        max_tokens=100,
+        top_p=1.0,
+        api_key="api_key",
+        deepseek_base_url=config_base_url,
     )
     llm = DeepSeekLLM(config)
     assert str(llm.client.base_url) == config_base_url
@@ -99,16 +99,16 @@ def test_generate_response_with_tools(mock_deepseek_client):
     response = llm.generate_response(messages, tools=tools)
 
     mock_deepseek_client.chat.completions.create.assert_called_once_with(
-        model="deepseek-chat", 
-        messages=messages, 
-        temperature=0.7, 
-        max_tokens=100, 
-        top_p=1.0, 
-        tools=tools, 
-        tool_choice="auto"
+        model="deepseek-chat",
+        messages=messages,
+        temperature=0.7,
+        max_tokens=100,
+        top_p=1.0,
+        tools=tools,
+        tool_choice="auto",
     )
 
     assert response["content"] == "I've added the memory for you."
     assert len(response["tool_calls"]) == 1
     assert response["tool_calls"][0]["name"] == "add_memory"
-    assert response["tool_calls"][0]["arguments"] == {"data": "Today is a sunny day."}
+    assert response["tool_calls"][0]["arguments"] == {"data": "Today is a sunny day."}