bibibi12345 commited on
Commit
af7a11e
·
verified ·
1 Parent(s): 24110a7

Update app/routes/models_api.py

Browse files
Files changed (1) hide show
  1. app/routes/models_api.py +1 -1
app/routes/models_api.py CHANGED
@@ -90,7 +90,7 @@ async def list_models(fastapi_request: Request, api_key: str = Depends(get_api_k
90
  })
91
 
92
  # Apply special suffixes for models starting with "gemini-2.5-flash"
93
- if original_model_id.startswith("gemini-2.5-flash"): # Suffix rules based on original_model_id
94
  special_flash_suffixes = ["-nothinking", "-max"]
95
  for special_suffix in special_flash_suffixes:
96
  suffixed_model_part = f"{original_model_id}{special_suffix}"
 
90
  })
91
 
92
  # Apply special suffixes for models starting with "gemini-2.5-flash"
93
+ if "gemini-2.5-flash" in original_model_id: # Suffix rules based on original_model_id
94
  special_flash_suffixes = ["-nothinking", "-max"]
95
  for special_suffix in special_flash_suffixes:
96
  suffixed_model_part = f"{original_model_id}{special_suffix}"