From cab45a1d50635a8f86807d9d0be974b436c82cbc Mon Sep 17 00:00:00 2001 From: Henrique Chaves Date: Thu, 21 Nov 2024 00:16:53 -0300 Subject: [PATCH] Fix test_litellm_client --- tests/llm/test_llm_client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llm/test_llm_client.py b/tests/llm/test_llm_client.py index f21f841216..b06a75b78e 100644 --- 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 )