mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
(ci/cd) fix timeout error
This commit is contained in:
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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue