diff --git a/src/proxy/google-ai.ts b/src/proxy/google-ai.ts index 52fe145..f99f05e 100644 --- a/src/proxy/google-ai.ts +++ b/src/proxy/google-ai.ts @@ -41,12 +41,7 @@ const getModelsResponse = () => { // Get all model IDs from keys, including flash and pro models const modelIds = Array.from( new Set(keys.map((k) => k.modelIds).flat()) - ).filter((id) => id.startsWith("models/gemini") || - id.startsWith("models/") && ( - id.includes("flash") || - id.includes("pro") || - id.includes("vision") - )); + ).filter((id) => id.startsWith("models/")); const models = modelIds.map((id) => ({ id,