tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-06 08:23:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix vertex ai exceptions

Browse Source
This commit is contained in:
Ishaan Jaff
2024-06-07 17:13:32 -07:00
parent a4589e33dd
commit d5e97861ee
1 changed files with 13 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/utils.py
+13 -17
View File
@@ -9180,16 +9180,12 @@ def exception_type(
exception_mapping_worked = True
raise litellm.InternalServerError(
message=f"litellm.InternalServerError: VertexAIException - {error_str}",
status_code=500,
model=model,
llm_provider="vertex_ai",
request=(
original_exception.request
if hasattr(original_exception, "request")
else httpx.Request(
method="POST",
url=" https://cloud.google.com/vertex-ai/",
)
response=httpx.Response(
status_code=500,
content=str(original_exception),
request=httpx.Request(method="completion", url="https://github.com/BerriAI/litellm"), # type: ignore
),
litellm_debug_info=extra_information,
)
@@ -9264,17 +9260,13 @@ def exception_type(
exception_mapping_worked = True
raise litellm.InternalServerError(
message=f"VertexAIException InternalServerError - {error_str}",
status_code=500,
model=model,
llm_provider="vertex_ai",
litellm_debug_info=extra_information,
request=getattr(
original_exception,
"request",
httpx.Request(
method="POST",
url=" https://cloud.google.com/vertex-ai/",
),
response=httpx.Response(
status_code=500,
content=str(original_exception),
request=httpx.Request(method="completion", url="https://github.com/BerriAI/litellm"), # type: ignore
),
)
elif custom_llm_provider == "palm" or custom_llm_provider == "gemini":
@@ -9878,7 +9870,11 @@ def exception_type(
llm_provider="azure",
model=model,
litellm_debug_info=extra_information,
request=httpx.Request(method="POST", url="https://openai.com/"),
response=httpx.Response(
status_code=400,
content=str(original_exception),
request=httpx.Request(method="completion", url="https://github.com/BerriAI/litellm"), # type: ignore
),
)
elif "This model's maximum context length is" in error_str:
exception_mapping_worked = True