ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(gemini)Handle HTTP 201 status code in Vertex AI response

Browse Source
This commit is contained in:
sven 2025-03-13 13:44:38 +09:00
parent 77c1a809f3
commit 8d053930e9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
View File

@ -846,7 +846,7 @@ async def make_call(
message=VertexGeminiConfig().translate_exception_str(exception_string),
headers=e.response.headers,
)
if response.status_code != 200:
if response.status_code != 200 and response.status_code != 201:
raise VertexAIError(
status_code=response.status_code,
message=response.text,
@ -884,7 +884,7 @@ def make_sync_call(
response = client.post(api_base, headers=headers, data=data, stream=True)
if response.status_code != 200:
if response.status_code != 200 and response.status_code != 201:
raise VertexAIError(
status_code=response.status_code,
message=str(response.read()),