From 21e9d24ba7ac8212ca621d5a54fcad6911a7760b Mon Sep 17 00:00:00 2001
From: NaccOll <wuyoubin0504@gmail.com>
Date: Wed, 20 Aug 2025 04:24:29 +0800
Subject: [PATCH] feat: add new models and update configurations for vscode-lm
 (#7201)

---
 packages/types/src/providers/vscode-llm.ts | 48 +++++++++++++++++++---
 1 file changed, 42 insertions(+), 6 deletions(-)

diff --git a/packages/types/src/providers/vscode-llm.ts b/packages/types/src/providers/vscode-llm.ts
index bf38cb814b..92e871a01a 100644
--- a/packages/types/src/providers/vscode-llm.ts
+++ b/packages/types/src/providers/vscode-llm.ts
@@ -101,6 +101,18 @@ export const vscodeLlmModels = {
 		supportsToolCalling: true,
 		maxInputTokens: 81638,
 	},
+	"claude-4-sonnet": {
+		contextWindow: 128000,
+		supportsImages: true,
+		supportsPromptCache: false,
+		inputPrice: 0,
+		outputPrice: 0,
+		family: "claude-sonnet-4",
+		version: "claude-sonnet-4",
+		name: "Claude Sonnet 4",
+		supportsToolCalling: true,
+		maxInputTokens: 111836,
+	},
 	"gemini-2.0-flash-001": {
 		contextWindow: 127827,
 		supportsImages: true,
@@ -114,7 +126,7 @@ export const vscodeLlmModels = {
 		maxInputTokens: 127827,
 	},
 	"gemini-2.5-pro": {
-		contextWindow: 63830,
+		contextWindow: 128000,
 		supportsImages: true,
 		supportsPromptCache: false,
 		inputPrice: 0,
@@ -123,10 +135,10 @@ export const vscodeLlmModels = {
 		version: "gemini-2.5-pro-preview-03-25",
 		name: "Gemini 2.5 Pro (Preview)",
 		supportsToolCalling: true,
-		maxInputTokens: 63830,
+		maxInputTokens: 108637,
 	},
 	"o4-mini": {
-		contextWindow: 111446,
+		contextWindow: 128000,
 		supportsImages: false,
 		supportsPromptCache: false,
 		inputPrice: 0,
@@ -135,10 +147,10 @@ export const vscodeLlmModels = {
 		version: "o4-mini-2025-04-16",
 		name: "o4-mini (Preview)",
 		supportsToolCalling: true,
-		maxInputTokens: 111446,
+		maxInputTokens: 111452,
 	},
 	"gpt-4.1": {
-		contextWindow: 111446,
+		contextWindow: 128000,
 		supportsImages: true,
 		supportsPromptCache: false,
 		inputPrice: 0,
@@ -147,7 +159,31 @@ export const vscodeLlmModels = {
 		version: "gpt-4.1-2025-04-14",
 		name: "GPT-4.1 (Preview)",
 		supportsToolCalling: true,
-		maxInputTokens: 111446,
+		maxInputTokens: 111452,
+	},
+	"gpt-5-mini": {
+		contextWindow: 128000,
+		supportsImages: true,
+		supportsPromptCache: false,
+		inputPrice: 0,
+		outputPrice: 0,
+		family: "gpt-5-mini",
+		version: "gpt-5-mini",
+		name: "GPT-5 mini (Preview)",
+		supportsToolCalling: true,
+		maxInputTokens: 108637,
+	},
+	"gpt-5": {
+		contextWindow: 128000,
+		supportsImages: true,
+		supportsPromptCache: false,
+		inputPrice: 0,
+		outputPrice: 0,
+		family: "gpt-5",
+		version: "gpt-5",
+		name: "GPT-5 (Preview)",
+		supportsToolCalling: true,
+		maxInputTokens: 108637,
 	},
 } as const satisfies Record<
 	string,