From ab54262d37d77bdde23a98ab18425cf11c1f40de Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 4 Nov 2023 16:05:14 -0700
Subject: [PATCH] fix(timeout.py): import errors

---
 litellm/llms/huggingface_restapi.py | 1 +
 litellm/timeout.py                  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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