(ci/cd) fix timeout error

2026-08-09 02:24:40 +00:00 · 2023-11-17 17:46:49 -08:00
parent e006cbbc73
commit a6a862dc5a
1 changed files with 4 additions and 1 deletions
@@ -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