From b54bbf510e4579de97702e97ac41e6aa412fd1ed Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 19 Sep 2024 16:20:43 -0700
Subject: [PATCH] fix azure gpt-4o test

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

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 42cd625ee..13bc92156 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -935,6 +935,9 @@ def test_completion_base64(model):
         print(f"\nResponse: {resp}")
 
         prompt_tokens = resp.usage.prompt_tokens
+    except litellm.ServiceUnavailableError as e:
+        print("got service unavailable error: ", e)
+        pass
     except Exception as e:
         if "500 Internal error encountered.'" in str(e):
             pass