From 6f9d3fc3bc7a913bbcc89e315faca570d67885b9 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 5 Jan 2024 18:02:20 +0530
Subject: [PATCH] (ci/cd) retry hosted ollama + stream test 3 times

---
 litellm/tests/test_streaming.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py
index 9a668fdeece..398704525c8 100644
--- a/litellm/tests/test_streaming.py
+++ b/litellm/tests/test_streaming.py
@@ -306,6 +306,8 @@ def test_completion_ollama_hosted_stream():
             model="ollama/phi",
             messages=messages,
             max_tokens=10,
+            num_retries=3,
+            timeout=90,
             api_base="https://test-ollama-endpoint.onrender.com",
             stream=True,
         )