diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py index 4347a7311..a27b8acb3 100644 --- a/litellm/llms/huggingface_restapi.py +++ b/litellm/llms/huggingface_restapi.py @@ -285,6 +285,7 @@ def completion( try: completion_response = response.json() except: + import traceback raise HuggingfaceError( message=f"Original Response received: {response.text}; Stacktrace: {traceback.format_exc()}", status_code=response.status_code ) diff --git a/litellm/timeout.py b/litellm/timeout.py index babcc7bea..9a8cba241 100644 --- a/litellm/timeout.py +++ b/litellm/timeout.py @@ -17,7 +17,7 @@ from concurrent import futures from inspect import iscoroutinefunction from functools import wraps 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):