From 0eb899c0871804f6afe9144438c56d7da0a957eb Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 5 Jan 2024 17:58:59 +0530
Subject: [PATCH] (test) hosted ollama - retry 3 times

---
 litellm/tests/test_completion.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index a3ee1183a..fe07e4493 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -749,10 +749,14 @@ def test_completion_ollama_hosted():
             model="ollama/phi",
             messages=messages,
             max_tokens=10,
+            num_retries=3,
+            timeout=90,
             api_base="https://test-ollama-endpoint.onrender.com",
         )
         # Add any assertions here to check the response
         print(response)
+    except Timeout as e:
+        pass
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")