From 36bc57a9624cdfaa50ffb2c1ad7f9c518c5e6c55 Mon Sep 17 00:00:00 2001 From: jackwener Date: Fri, 20 Mar 2026 18:55:45 +0800 Subject: [PATCH] fix(serve): update model mappings to match actual Antigravity UI - Default to 'claude sonnet 4.6' - Map 'sonnet' -> 'claude sonnet 4.6' - Map 'opus' -> 'claude opus 4.6' - Map 'gemini.*pro' -> 'gemini 3.1 pro (high)' - Map 'gemini.*flash' -> 'gemini 3 flash' - Map 'gpt' -> 'gpt-oss 120b' --- src/clis/antigravity/serve.ts | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/src/clis/antigravity/serve.ts b/src/clis/antigravity/serve.ts index 4a41470..ad57a0e 100644 --- a/src/clis/antigravity/serve.ts +++ b/src/clis/antigravity/serve.ts @@ -100,18 +100,20 @@ async function startNewConversation(page: IPage): Promise { * Switch the active model in Antigravity UI. */ async function switchModel(page: IPage, anthropicModelId: string): Promise { - // Map standard Anthropic model IDs to Antigravity UI names - let targetName = 'claude 3.5 sonnet'; // Default fallback + // Map standard model IDs to Antigravity UI names based on actual UI + let targetName = 'claude sonnet 4.6'; // Default fallback const id = anthropicModelId.toLowerCase(); - if (id.includes('3-7-sonnet') || id.includes('3.7-sonnet')) { - targetName = 'claude 3.7 sonnet'; - } else if (id.includes('3-5-sonnet') || id.includes('3.5-sonnet')) { - targetName = 'claude 3.5 sonnet'; - } else if (id.includes('3-5-haiku') || id.includes('3.5-haiku')) { - targetName = 'claude 3.5 haiku'; + if (id.includes('sonnet')) { + targetName = 'claude sonnet 4.6'; } else if (id.includes('opus')) { - targetName = 'claude 3 opus'; + targetName = 'claude opus 4.6'; + } else if (id.includes('gemini') && id.includes('pro')) { + targetName = 'gemini 3.1 pro (high)'; + } else if (id.includes('gemini') && id.includes('flash')) { + targetName = 'gemini 3 flash'; + } else if (id.includes('gpt')) { + targetName = 'gpt-oss 120b'; } try {