From 42d092520933a7ec8bd737cb1855d5b6c84d8397 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 13 Nov 2023 16:34:59 -0800
Subject: [PATCH] test(test_streaming.py): adding retries to streaming tests

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

diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py
index 2493524ea68..314b8736b6e 100644
--- a/litellm/tests/test_streaming.py
+++ b/litellm/tests/test_streaming.py
@@ -14,7 +14,8 @@ import litellm
 from litellm import completion, acompletion, AuthenticationError, BadRequestError, RateLimitError, ModelResponse
 
 litellm.logging = False
-litellm.set_verbose = False
+litellm.set_verbose = True
+litellm.num_retries = 3
 litellm.cache = None
 
 score = 0
@@ -350,6 +351,7 @@ def test_completion_cohere_stream_bad_key():
 
 def test_completion_azure_stream():
     try:
+        litellm.set_verbose = True
         messages = [
             {"role": "system", "content": "You are a helpful assistant."},
             {
@@ -372,7 +374,7 @@ def test_completion_azure_stream():
         print(f"completion_response: {complete_response}")
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")
-# test_completion_azure_stream() 
+test_completion_azure_stream() 
 
 def test_completion_claude_stream():
     try: