(ci/cd) fix timeout error

This commit is contained in:
ishaan-jaff 2023-11-17 17:46:49 -08:00
parent e006cbbc73
commit a6a862dc5a

View file

@ -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