From a6a862dc5ab0bb17178af72d6f4d0dc3cf277bf5 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 17 Nov 2023 17:46:49 -0800
Subject: [PATCH] (ci/cd) fix timeout error

---
 litellm/tests/test_supabase_integration.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_supabase_integration.py b/litellm/tests/test_supabase_integration.py
index d117c5265d..b07505c238 100644
--- a/litellm/tests/test_supabase_integration.py
+++ b/litellm/tests/test_supabase_integration.py
@@ -39,7 +39,8 @@ def test_acompletion_sync():
                 messages=[{"role": "user", "content": "write a poem"}], 
                 max_tokens=10,
                 stream=True,
-                user="ishaanStreamingUser"
+                user="ishaanStreamingUser",
+                timeout=5
             )
             complete_response = ""
             start_time = time.time()
@@ -54,6 +55,8 @@ def test_acompletion_sync():
                     print("🤗🤗🤗 DONE")
                     return
 
+        except litellm.Timeout as e: 
+            pass
         except:
             print(f"error occurred: {traceback.format_exc()}")
             pass