From 194c823783f4468f6bbcfca79524fa5dbb9638d3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 24 Feb 2024 09:55:58 -0800
Subject: [PATCH] fix(huggingface_restapi.py): return initial hf error

---
 litellm/llms/huggingface_restapi.py | 22 ++++++++++++++++------
 litellm/proxy/proxy_server.py       |  2 +-
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py
index 123b8ecbc..205bad7ee 100644
--- a/litellm/llms/huggingface_restapi.py
+++ b/litellm/llms/huggingface_restapi.py
@@ -575,12 +575,20 @@ class Huggingface(BaseLLM):
                 response = await client.post(url=api_base, json=data, headers=headers)
                 response_json = response.json()
                 if response.status_code != 200:
-                    raise HuggingfaceError(
-                        status_code=response.status_code,
-                        message=response.text,
-                        request=response.request,
-                        response=response,
-                    )
+                    if "error" in response_json:
+                        raise HuggingfaceError(
+                            status_code=response.status_code,
+                            message=response_json["error"],
+                            request=response.request,
+                            response=response,
+                        )
+                    else:
+                        raise HuggingfaceError(
+                            status_code=response.status_code,
+                            message=response.text,
+                            request=response.request,
+                            response=response,
+                        )
 
                 ## RESPONSE OBJECT
                 return self.convert_to_model_response_object(
@@ -595,6 +603,8 @@ class Huggingface(BaseLLM):
         except Exception as e:
             if isinstance(e, httpx.TimeoutException):
                 raise HuggingfaceError(status_code=500, message="Request Timeout Error")
+            elif isinstance(e, HuggingfaceError):
+                raise e
             elif response is not None and hasattr(response, "text"):
                 raise HuggingfaceError(
                     status_code=500,
diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 4e425bd7f..14f8bc96c 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -2445,7 +2445,7 @@ async def completion(
         )
         traceback.print_exc()
         error_traceback = traceback.format_exc()
-        error_msg = f"{str(e)}\n\n{error_traceback}"
+        error_msg = f"{str(e)}"
         raise ProxyException(
             message=getattr(e, "message", error_msg),
             type=getattr(e, "type", "None"),