From f6d33a899840f6cc80727fea7caf9c3c540968ed Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 31 May 2024 09:03:28 -0700
Subject: [PATCH] ci/cd clarifai is an unstable provider, retry reqs

---
 litellm/tests/test_clarifai_completion.py | 3 +++
 1 file changed, 3 insertions(+)

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"),
             )