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