Fix test_litellm_client

Giskard-AI · Nov 21, 2024 · cab45a1 · cab45a1
1 parent 77e6a4f
commit cab45a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/llm/test_llm_client.py b/tests/llm/test_llm_client.py
@@ -60,7 +60,7 @@ def test_litellm_client(completion):
     completion.return_value = DEMO_OPENAI_RESPONSE
     client = Mock()
     client.chat.completions.create.return_value = DEMO_OPENAI_RESPONSE
-    res = LiteLLMClient("gpt-4o", {"api_key": "api_key"}).complete(
+    res = LiteLLMClient("gpt-4o", True, completion_params={"api_key": "api_key"}).complete(
         [ChatMessage(role="system", content="Hello")], temperature=0.11, max_tokens=1
     )