tiennm99/ccs
1
0 Fork 0
mirror of https://github.com/tiennm99/ccs.git synced 2026-10-14 04:15:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: sanitize cliproxy provider model pickers

Browse Source
This commit is contained in:
Tam Nhu Tran committed 2026-04-10 18:07:22 -04:00
1 parent cfdae7be9f
commit 0e4677f558
6 files changed
+327 -45
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/cliproxy/model-catalog.ts
+37 -4
View File
@@ -14,6 +14,21 @@ import {
import { stripModelConfigurationSuffixes } from '../shared/extended-context-utils';
import { GEMINI_MINOR_VERSION_COMPATIBILITY_IDS } from '../shared/gemini-minor-version-compatibility';
const AGY_GEMINI_PRO_HIGH_ID = 'gemini-3.1-pro-high';
const AGY_GEMINI_PRO_LOW_ID = 'gemini-3.1-pro-low';
const AGY_GEMINI_PRO_COMPATIBILITY_IDS = Object.freeze({
'gemini-3-pro-high': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3.1-pro-high': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3-pro-low': AGY_GEMINI_PRO_LOW_ID,
'gemini-3.1-pro-low': AGY_GEMINI_PRO_LOW_ID,
'gemini-3-pro-preview': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3-pro-preview-customtools': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3.1-pro-preview': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3.1-pro-preview-customtools': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3-1-pro-preview': AGY_GEMINI_PRO_HIGH_ID,
'gemini-3-1-pro-preview-customtools': AGY_GEMINI_PRO_HIGH_ID,
} satisfies Record<string, string>);
/**
* Thinking support configuration for a model.
* Defines how thinking/reasoning budget can be controlled.
@@ -114,11 +129,19 @@ export const MODEL_CATALOG: Partial<Record<CLIProxyProvider, ProviderCatalog>> =
},
},
{
id: 'gemini-3.1-pro-preview',
name: 'Gemini 3.1 Pro',
description: 'Google latest Gemini Pro model via Antigravity',
id: AGY_GEMINI_PRO_HIGH_ID,
name: 'Gemini 3.1 Pro High',
description: 'Current Antigravity Gemini Pro route with higher reasoning budget',
nativeImageInput: true,
thinking: { type: 'levels', levels: ['low', 'high'], dynamicAllowed: true },
thinking: { type: 'none' },
extendedContext: true,
},
{
id: AGY_GEMINI_PRO_LOW_ID,
name: 'Gemini 3.1 Pro Low',
description: 'Current Antigravity Gemini Pro route with the lighter quota tier',
nativeImageInput: true,
thinking: { type: 'none' },
extendedContext: true,
},
{
@@ -465,6 +488,16 @@ export function findModel(provider: CLIProxyProvider, modelId: string): ModelEnt
if (compatibilityId) {
lookupCandidates.add(compatibilityId);
}
if (isAntigravityProvider(provider)) {
const agyCompatibilityId =
AGY_GEMINI_PRO_COMPATIBILITY_IDS[
candidate as keyof typeof AGY_GEMINI_PRO_COMPATIBILITY_IDS
];
if (agyCompatibilityId) {
lookupCandidates.add(agyCompatibilityId);
}
}
}
return catalog.models.find((m) => lookupCandidates.has(m.id.toLowerCase()));