phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix timeouts + tests + bump v

Browse source
This commit is contained in:
ishaan-jaff 2023-09-05 17:17:58 -07:00
parent 0bbcba269d
commit fb3983d058
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: