Update app/main.py

app/main.py

@@ -1369,8 +1369,10 @@ async def chat_completions(request: OpenAIRequest, api_key: str = Depends(get_ap
             return JSONResponse(status_code=400, content=error_response)
 
         # --- Handle specific OpenAI client model ---
-        if request.model == "gemini-2.5-pro-exp-03-25-openai":
+        if request.model.endswith("-openai"): # Generalized check for suffix
             print(f"INFO: Using OpenAI library path for model: {request.model}")
+            base_model_name = request.model.replace("-openai", "") # Extract base model name
+            UNDERLYING_MODEL_ID = f"google/{base_model_name}" # Add google/ prefix
 
             # --- Determine Credentials for OpenAI Client (Correct Priority) ---
             credentials_to_use = None
@@ -1459,7 +1461,7 @@ async def chat_completions(request: OpenAIRequest, api_key: str = Depends(get_ap
                 f"https://{LOCATION}-aiplatform.googleapis.com/v1beta1/"
                 f"projects/{PROJECT_ID}/locations/{LOCATION}/endpoints/openapi"
             )
-            UNDERLYING_MODEL_ID = "gemini-2.5-pro-exp-03-25" # As specified
+            # UNDERLYING_MODEL_ID is now set above based on the request
 
             # Initialize Async OpenAI Client
             openai_client = openai.AsyncOpenAI(