bibibi12345 commited on
Commit
d9d3a0e
·
1 Parent(s): db96d95

added nothinking and max for express 0605

Browse files
Files changed (1) hide show
  1. app/routes/models_api.py +3 -2
app/routes/models_api.py CHANGED
@@ -90,8 +90,9 @@ async def list_models(fastapi_request: Request, api_key: str = Depends(get_api_k
90
  "permission": [], "root": original_model_id, "parent": None
91
  })
92
 
93
- # Apply special suffixes for models starting with "gemini-2.5-flash" or specifically "gemini-2.5-pro-preview-06-05"
94
- if "gemini-2.5-flash" in original_model_id or original_model_id == "gemini-2.5-pro-preview-06-05": # Suffix rules based on original_model_id
 
95
  special_thinking_suffixes = ["-nothinking", "-max"]
96
  for special_suffix in special_thinking_suffixes:
97
  suffixed_model_part = f"{original_model_id}{special_suffix}"
 
90
  "permission": [], "root": original_model_id, "parent": None
91
  })
92
 
93
+ # Apply special suffixes for models starting with "gemini-2.5-flash" or containing "gemini-2.5-pro"
94
+ # This includes both regular and EXPRESS versions
95
+ if "gemini-2.5-flash" in original_model_id or "gemini-2.5-pro" in original_model_id: # Suffix rules based on original_model_id
96
  special_thinking_suffixes = ["-nothinking", "-max"]
97
  for special_suffix in special_thinking_suffixes:
98
  suffixed_model_part = f"{original_model_id}{special_suffix}"