Fix test

Author: chrarnoldus

src/core/webview/__tests__/ClineProvider.spec.ts

@@ -2705,7 +2705,6 @@ describe("ClineProvider - Router Models", () => {
 				litellmApiKey: "litellm-key",
 				litellmBaseUrl: "http://localhost:4000",
 				// kilocode_change start
-				chutesApiKey: "chutes-key",
 				geminiApiKey: "gemini-key",
 				googleGeminiBaseUrl: "https://gemini.example.com",
 				ovhCloudAiEndpointsApiKey: "ovhcloud-key",
@@ -2744,7 +2743,6 @@ describe("ClineProvider - Router Models", () => {
 			baseUrl: "https://gemini.example.com",
 		})
 		expect(getModels).toHaveBeenCalledWith({ provider: "ovhcloud", apiKey: "ovhcloud-key" })
-		expect(getModels).toHaveBeenCalledWith({ provider: "chutes", apiKey: "chutes-key" })
 		expect(getModels).toHaveBeenCalledWith({
 			provider: "inception",
 			apiKey: "inception-key",
@@ -2830,7 +2828,6 @@ describe("ClineProvider - Router Models", () => {
 			.mockRejectedValueOnce(new Error("Requesty API error")) // requesty fail
 			.mockResolvedValueOnce(mockModels) // glama success
 			.mockRejectedValueOnce(new Error("Unbound API error")) // unbound fail
-			.mockRejectedValueOnce(new Error("Chutes API error")) // kilocode_change: chutes fail
 			.mockRejectedValueOnce(new Error("Kilocode-OpenRouter API error")) // kilocode-openrouter fail
 			.mockRejectedValueOnce(new Error("Ollama API error")) // kilocode_change
 			.mockResolvedValueOnce(mockModels) // vercel-ai-gateway success

src/core/webview/__tests__/webviewMessageHandler.routerModels.spec.ts

@@ -8,6 +8,11 @@ vi.mock("vscode", () => ({
 		showErrorMessage: vi.fn(),
 		showWarningMessage: vi.fn(),
 		showInformationMessage: vi.fn(),
+		// kilocode_change start
+		createTextEditorDecorationType: vi.fn(() => ({
+			dispose: vi.fn(),
+		})),
+		// kilocode_change end
 	},
 	workspace: {
 		workspaceFolders: undefined,

src/core/webview/__tests__/webviewMessageHandler.spec.ts

@@ -274,7 +274,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 				"kilocode-openrouter": mockModels,
 				roo: mockModels,
 				chutes: mockModels,
-				ollama: {},
+				ollama: mockModels, // kilocode_change
 				lmstudio: {},
 				"vercel-ai-gateway": mockModels,
 				huggingface: {},
@@ -406,7 +406,6 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 			.mockRejectedValueOnce(new Error("Requesty API error")) // requesty
 			.mockResolvedValueOnce(mockModels) // glama
 			.mockRejectedValueOnce(new Error("Unbound API error")) // unbound
-			.mockRejectedValueOnce(new Error("Chutes API error")) // chutes // kilocode_change
 			.mockResolvedValueOnce(mockModels) // kilocode-openrouter
 			.mockRejectedValueOnce(new Error("Ollama API error")) // kilocode_change
 			.mockResolvedValueOnce(mockModels) // vercel-ai-gateway
@@ -437,12 +436,6 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 		})
 
 		// kilocode_change start
-		expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
-			type: "singleRouterModelFetchResponse",
-			success: false,
-			error: "Chutes API error",
-			values: { provider: "chutes" },
-		})
 		expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
 			type: "singleRouterModelFetchResponse",
 			success: false,
@@ -482,6 +475,12 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 				"vercel-ai-gateway": mockModels,
 				huggingface: {},
 				"io-intelligence": {},
+				// kilocode_change start
+				"kilocode-openrouter": mockModels,
+				inception: {},
+				gemini: mockModels,
+				ovhcloud: mockModels,
+				// kilocode_change end
 			},
 			values: undefined,
 		})
@@ -551,22 +550,22 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 		expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
 			type: "singleRouterModelFetchResponse",
 			success: false,
-			error: "Chutes API error",
-			values: { provider: "chutes" },
+			error: "Ollama API error",
+			values: { provider: "ollama" },
 		})
 
 		expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
 			type: "singleRouterModelFetchResponse",
 			success: false,
-			error: "Ollama API error",
-			values: { provider: "ollama" },
+			error: "Vercel AI Gateway error",
+			values: { provider: "vercel-ai-gateway" },
 		})
 
 		expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
 			type: "singleRouterModelFetchResponse",
 			success: false,
-			error: "Vercel AI Gateway error",
-			values: { provider: "vercel-ai-gateway" },
+			error: "Chutes API error",
+			values: { provider: "chutes" },
 		})
 		// kilocode_change end
 

src/core/webview/webviewMessageHandler.ts

@@ -869,7 +869,6 @@ export const webviewMessageHandler = async (
 				},
 				{ key: "glama", options: { provider: "glama" } },
 				{ key: "unbound", options: { provider: "unbound", apiKey: apiConfiguration.unboundApiKey } },
-				{ key: "chutes", options: { provider: "chutes", apiKey: apiConfiguration.chutesApiKey } }, // kilocode_change
 				{
 					key: "kilocode-openrouter",
 					options: {