phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix timeouts + tests + bump v

Browse source
This commit is contained in:
ishaan-jaff 2023-09-05 17:17:58 -07:00
parent f76cfb33f5
commit 2691559e24
3 changed files with 18 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/timeout.py
View file

@ -38,6 +38,8 @@ def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout):
local_timeout_duration = timeout_duration
if "force_timeout" in kwargs and kwargs["force_timeout"] is not None:
local_timeout_duration = kwargs["force_timeout"]
elif "request_timeout" in kwargs and kwargs["request_timeout"] is not None:
local_timeout_duration = kwargs["request_timeout"]
try:
result = future.result(timeout=local_timeout_duration)
except futures.TimeoutError: