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

refactor: add black formatting

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-25 14:10:38 +05:30
parent b87d630b0a
commit 4905929de3
156 changed files with 19723 additions and 10869 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/timeout.py
View file

@ -56,8 +56,8 @@ def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout):
model = args[0] if len(args) > 0 else kwargs["model"]
raise exception_to_raise(
f"A timeout error occurred. The function call took longer than {local_timeout_duration} second(s).",
model=model, # [TODO]: replace with logic for parsing out llm provider from model name
llm_provider="openai"
model=model, # [TODO]: replace with logic for parsing out llm provider from model name
llm_provider="openai",
)
thread.stop_loop()
return result
@ -78,8 +78,8 @@ def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout):
model = args[0] if len(args) > 0 else kwargs["model"]
raise exception_to_raise(
f"A timeout error occurred. The function call took longer than {local_timeout_duration} second(s).",
model=model, # [TODO]: replace with logic for parsing out llm provider from model name
llm_provider="openai"
model=model, # [TODO]: replace with logic for parsing out llm provider from model name
llm_provider="openai",
)
if iscoroutinefunction(func):