From e1cbb397fed844992e93aef3b58ea84dd26f923c Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 30 Jul 2024 15:32:36 -0700
Subject: [PATCH] test(test_streaming.py): handle predibase instability

---
 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 9aebc0f247..ed88f818cc 100644
--- a/litellm/tests/test_streaming.py
+++ b/litellm/tests/test_streaming.py
@@ -571,6 +571,8 @@ async def test_completion_predibase_streaming(sync_mode):
         pass
     except litellm.InternalServerError as e:
         pass
+    except litellm.ServiceUnavailableError as e:
+        pass
     except Exception as e:
         print("ERROR class", e.__class__)
         print("ERROR message", e)