Skip to content

Commit cdb9474

Browse files
committed
fix, gemini needs 'google/' prefix in model name
1 parent ae64e16 commit cdb9474

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/api/routers/vertex.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -135,8 +135,10 @@ def from_anthropic_to_openai_response(msg, model):
135135
})
136136

137137
def get_chat_completion_model_name(model_alias):
138-
# publishers/google/models/gemini-2.0-flash-lite-001 -> "gemini-2.0-flash-lite-001"
139-
return model_alias.split("/")[-1]
138+
if model_alias.startswith("publishers/google/"):
139+
return f"google/{model_alias.split('/')[-1]}"
140+
141+
return model_alias.split('/')[-1]
140142

141143
async def handle_proxy(request: Request, path: str):
142144
try:

0 commit comments

Comments
 (0)