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

fix(proxy_server.py): fix embedding model exception mapping

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-06 11:14:41 -07:00
parent 1ba61b908b
commit 47ce6ccac0
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/proxy_server.py
View file

@ -3372,8 +3372,9 @@ async def embeddings(
) )
verbose_proxy_logger.debug(traceback.format_exc()) verbose_proxy_logger.debug(traceback.format_exc())
if isinstance(e, HTTPException): if isinstance(e, HTTPException):
message = get_error_message_str(e)
raise ProxyException( raise ProxyException(
message=getattr(e, "message", str(e)), message=message,
type=getattr(e, "type", "None"), type=getattr(e, "type", "None"),
param=getattr(e, "param", "None"), param=getattr(e, "param", "None"),
code=getattr(e, "status_code", status.HTTP_400_BAD_REQUEST), code=getattr(e, "status_code", status.HTTP_400_BAD_REQUEST),

5
litellm/proxy/utils.py
View file

@ -2888,6 +2888,11 @@ def get_error_message_str(e: Exception) -> str:
error_message = e.detail error_message = e.detail
elif isinstance(e.detail, dict): elif isinstance(e.detail, dict):
error_message = json.dumps(e.detail) error_message = json.dumps(e.detail)
elif hasattr(e, "message"):
if isinstance(e.message, "str"):
error_message = e.message
elif isinstance(e.message, dict):
error_message = json.dumps(e.message)
else: else:
error_message = str(e) error_message = str(e)
else: else: