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