phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(ci/cd) fix timeout error

Browse source
This commit is contained in:
ishaan-jaff 2023-11-17 17:46:49 -08:00
parent e006cbbc73
commit a6a862dc5a
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_supabase_integration.py
View file

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