diff --git a/litellm/tests/test_clarifai_completion.py b/litellm/tests/test_clarifai_completion.py index ce0c6ecfea8..400ed8a2bed 100644 --- a/litellm/tests/test_clarifai_completion.py +++ b/litellm/tests/test_clarifai_completion.py @@ -47,6 +47,7 @@ def test_completion_clarifai_claude_2_1(): try: response = completion( model="clarifai/anthropic.completion.claude-2_1", + num_retries=3, messages=messages, max_tokens=10, temperature=0.1, @@ -66,6 +67,7 @@ def test_completion_clarifai_mistral_large(): response: ModelResponse = completion( model="clarifai/mistralai.completion.mistral-small", messages=messages, + num_retries=3, max_tokens=10, temperature=0.78, ) @@ -91,6 +93,7 @@ def test_async_completion_clarifai(): response = await acompletion( model="clarifai/openai.chat-completion.GPT-4", messages=messages, + num_retries=3, timeout=10, api_key=os.getenv("CLARIFAI_API_KEY"), )