fix(timeout.py): import errors

This commit is contained in:
Krrish Dholakia 2023-11-04 16:05:14 -07:00
parent 5bd6d6ff4c
commit ab54262d37
2 changed files with 2 additions and 1 deletions

View file

@ -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
)

View file

@ -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):