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

fix(timeout.py): import errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-04 16:05:14 -07:00
parent 2310e0c6eb
commit b1dcd0f1b9
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/llms/huggingface_restapi.py
View file

@ -285,6 +285,7 @@ def completion(
try: try:
completion_response = response.json() completion_response = response.json()
except: except:
import traceback
raise HuggingfaceError( raise HuggingfaceError(
message=f"Original Response received: {response.text}; Stacktrace: {traceback.format_exc()}", status_code=response.status_code message=f"Original Response received: {response.text}; Stacktrace: {traceback.format_exc()}", status_code=response.status_code
) )

2
litellm/timeout.py
View file

@ -17,7 +17,7 @@ from concurrent import futures
from inspect import iscoroutinefunction from inspect import iscoroutinefunction
from functools import wraps from functools import wraps
from threading import Thread from threading import Thread
from litellm.error import Timeout from openai.error import Timeout
def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout): def timeout(timeout_duration: float = 0.0, exception_to_raise=Timeout):