luulam/9router
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix (#1361)

Browse Source 
Co-authored-by: Ansh7473 <Ansh7473@users.noreply.github.com>
This commit is contained in:
YourAnsh
2026-05-23 07:56:10 +05:30
committed by GitHub
parent d29b19bc27
commit 2604b46847
7 changed files with 47 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/mitm/config.js
View File

@@ -30,19 +30,25 @@ const URL_PATTERNS = {
// Synonym map: rawModel from request → canonical alias key in mitmAlias DB
const MODEL_SYNONYMS = {
antigravity: { "gemini-default": "gemini-3-flash" },
antigravity: {
"gemini-default": "gemini-3-flash-agent",
"gemini-3.1-pro-high": "gemini-pro-agent",
"gemini-3-pro-high": "gemini-pro-agent",
"gemini-3-pro-low": "gemini-3.1-pro-low",
},
};
// Pattern fallback: rawModel regex → canonical alias key (when exact + prefix match fail)
// Order matters: more specific patterns first. Catches AG renamed variants (e.g. gemini-pro-agent)
const MODEL_PATTERNS = {
antigravity: [
{ match: /flash/i, alias: "gemini-3-flash" },
{ match: /pro.*low|low.*pro/i, alias: "gemini-3.1-pro-low" },
{ match: /gemini.*pro|pro.*gemini/i, alias: "gemini-3.1-pro-high" },
{ match: /opus/i, alias: "claude-opus-4-6-thinking" },
{ match: /sonnet|claude/i, alias: "claude-sonnet-4-6" },
{ match: /gpt.*oss|oss/i, alias: "gpt-oss-120b-medium" },
{ match: /flash.*low|low.*flash|flash.*medium|medium.*flash/i, alias: "gemini-3.5-flash-low" },
{ match: /flash.*agent|agent.*flash|flash/i, alias: "gemini-3-flash-agent" },
{ match: /pro.*low|low.*pro/i, alias: "gemini-3.1-pro-low" },
{ match: /gemini.*pro|pro.*gemini/i, alias: "gemini-pro-agent" },
{ match: /opus/i, alias: "claude-opus-4-6-thinking" },
{ match: /sonnet|claude/i, alias: "claude-sonnet-4-6" },
{ match: /gpt.*oss|oss/i, alias: "gpt-oss-120b-medium" },
],
};

2
src/mitm/server.js
View File

@@ -112,7 +112,7 @@ function getMappedModel(tool, model) {
// Prefix match fallback
const prefixKey = Object.keys(aliases).find(k => k && aliases[k] && (lookup.startsWith(k) || k.startsWith(lookup)));
if (prefixKey) return aliases[prefixKey];
// Pattern fallback: catches AG renamed variants (e.g. gemini-pro-agent → gemini-3.1-pro-high)
// Pattern fallback: catches AG renamed variants (e.g. deprecated pro IDs → gemini-pro-agent)
const patterns = MODEL_PATTERNS?.[tool] || [];
for (const { match, alias } of patterns) {
if (match.test(lookup) && aliases[alias]) return aliases[alias];