update for handle mistral. added some more models

Author: nullfunc

src/api/routers/vertex.py

@@ -137,11 +137,8 @@ def from_anthropic_to_openai_response(msg, model):
     })
 
 def get_chat_completion_model_name(model_alias):
-    if model_alias in known_chat_models:
-        # publishers/google/models/gemini-2.0-flash-lite-001 -> "google/gemini-2.0-flash-lite-001"
-        model_alias = model_alias.replace("publishers/", "").replace("models/", "")
-
-    return model_alias
+    # publishers/google/models/gemini-2.0-flash-lite-001 -> "gemini-2.0-flash-lite-001"
+    return model_alias.split("/")[-1]
 
 async def handle_proxy(request: Request, path: str):
     try:

src/data/modelmap.json

@@ -38,10 +38,17 @@
     "ai/smollm2": "publishers/google/models/gemini-2.0-flash-lite-001",
     "ai/smollm2:latest": "publishers/google/models/gemini-2.0-flash-lite-001",
     "gemini-2.5-pro": "publishers/google/models/gemini-2.5-pro-preview-05-06",
+    "gemini-2.5-pro:latest": "publishers/google/models/gemini-2.5-pro-preview-05-06",
     "gemini-2.5-flash": "publishers/google/models/gemini-2.5-flash-preview-05-20",
+    "gemini-2.5-flash:latest": "publishers/google/models/gemini-2.5-flash-preview-05-20",
     "gemini-2.0-flash": "publishers/google/models/gemini-2.0-flash-001",
+    "gemini-2.0-flash:latest": "publishers/google/models/gemini-2.0-flash-001",
     "gemini-2.0-flash-lite": "publishers/google/models/gemini-2.0-flash-lite-001",
+    "gemini-2.0-flash-lite:latest": "publishers/google/models/gemini-2.0-flash-lite-001",
     "claude-3-haiku": "publishers/anthropic/models/claude-3-haiku@20240307",
+    "claude-3-haiku:latest": "publishers/anthropic/models/claude-3-haiku@20240307",
+    "ai/mistral": "publishers/mistralai/models/mistral-nemo",
+    "a/mistral:latest": "publishers/mistralai/models/mistral-nemo",
     "default": "publishers/google/models/gemini-2.0-flash-lite-001"
   }
 }