Commit
·
d9d3a0e
1
Parent(s):
db96d95
added nothinking and max for express 0605
Browse files- 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
|
94 |
-
|
|
|
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}"
|