feat(api): implement auto model selection and update provider integration (#660)

Author: mini2s

src/api/providers/fetchers/modelCache.ts

@@ -28,7 +28,7 @@ import { ZgsmAuthConfig } from "../../../core/costrict/auth"
 import { IZgsmModelResponseData } from "@roo-code/types"
 import { getHuggingFaceModels } from "./huggingface"
 import { ClineProvider } from "../../../core/webview/ClineProvider"
-import { getRooModels } from "./roo"
+// import { getRooModels } from "./roo"
 import { getChutesModels } from "./chutes"
 
 const memoryCache = new NodeCache({ stdTTL: 5 * 60, checkperiod: 5 * 60 })
@@ -128,9 +128,10 @@ export const getModels = async (options: GetModelsOptions): Promise<ModelRecord>
 				break
 			case "roo": {
 				// Roo Code Cloud provider requires baseUrl and optional apiKey
-				const rooBaseUrl =
-					options.baseUrl ?? process.env.ROO_CODE_PROVIDER_URL ?? "https://api.roocode.com/proxy"
-				models = await getRooModels(rooBaseUrl, options.apiKey)
+				// const rooBaseUrl =
+				// 	options.baseUrl ?? process.env.ROO_CODE_PROVIDER_URL ?? "https://api.roocode.com/proxy"
+				// models = await getRooModels(rooBaseUrl, options.apiKey)
+				models = {}
 				break
 			}
 			case "chutes":

src/api/providers/zgsm.ts

@@ -440,10 +440,10 @@ export class ZgsmAiHandler extends BaseProvider implements SingleCompletionHandl
 		// Yield selected LLM info if available (for Auto model mode)
 		if (isAuto) {
 			yield {
-				type: "text",
-				text: `["${selectedLLM}" ${selectReason ? `, "(${selectReason})"` : ""}]`,
-				isAuto,
+				type: "automodel",
+				text: selectReason ? ` (${selectReason})` : "",
 				originModelId: this.options.zgsmModelId,
+				selectedLLM,
 			}
 		}
 

src/api/transform/stream.ts

@@ -2,6 +2,7 @@ export type ApiStream = AsyncGenerator<ApiStreamChunk>
 
 export type ApiStreamChunk =
 	| ApiStreamTextChunk
+	| ApiStreamAutoModelChunk
 	| ApiStreamUsageChunk
 	| ApiStreamReasoningChunk
 	| ApiStreamGroundingChunk
@@ -17,8 +18,14 @@ export interface ApiStreamError {
 export interface ApiStreamTextChunk {
 	type: "text"
 	text: string
+}
+
+export interface ApiStreamAutoModelChunk {
+	type: "automodel"
+	text: string
 	isAuto?: boolean
 	originModelId?: string
+	selectedLLM?: string
 }
 
 export interface ApiStreamReasoningChunk {

src/core/task/Task.ts

@@ -2190,30 +2190,29 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
 								presentAssistantMessage(this)
 								break
 							}
-							case "text": {
+							case "automodel": {
 								// Check if it is Selected LLM information (only in Auto model mode).
 								if (
-									chunk.isAuto &&
+									this.apiConfiguration.apiProvider === "zgsm" &&
 									lastApiReqIndex >= 0 &&
-									this.clineMessages[lastApiReqIndex] &&
-									this.apiConfiguration.apiProvider === "zgsm"
+									this.clineMessages[lastApiReqIndex]
 								) {
 									// Extract Selected LLM and Reason information and update the api_req_started message.
 									const existingData = JSON.parse(this.clineMessages[lastApiReqIndex].text || "{}")
-									const [selectedLLM, selectReason] = JSON.parse(chunk.text)
-
 									this.clineMessages[lastApiReqIndex].text = JSON.stringify({
 										...existingData,
-										selectedLLM,
-										selectReason,
-										isAuto: chunk.isAuto,
+										selectedLLM: chunk.selectedLLM,
+										selectReason: chunk.text,
+										isAuto: true,
 										originModelId: chunk.originModelId,
 									} satisfies ClineApiReqInfo)
 									// Save the selection information but do not add it to the assistant message to avoid it being processed by the parser.
-									console.log(`[Backend Model Route Detail] ${selectedLLM}${selectReason}`)
+									console.log(`[Backend Model Route Detail] ${chunk.selectedLLM}${chunk.text}`)
 									break
 								}
-
+								break
+							}
+							case "text": {
 								assistantMessage += chunk.text
 
 								if (this.assistantMessageParser) {

src/core/webview/webviewMessageHandler.ts

@@ -1067,38 +1067,38 @@ export const webviewMessageHandler = async (
 			}
 			break
 		}
-		case "requestRooModels": {
-			// Specific handler for Roo models only - flushes cache to ensure fresh auth token is used
-			try {
-				// Flush cache first to ensure fresh models with current auth state
-				await flushModels("roo")
-
-				const rooModels = await getModels({
-					provider: "roo",
-					baseUrl: process.env.ROO_CODE_PROVIDER_URL ?? "https://api.roocode.com/proxy",
-					apiKey: CloudService.hasInstance()
-						? CloudService.instance.authService?.getSessionToken()
-						: undefined,
-				})
-
-				// Always send a response, even if no models are returned
-				provider.postMessageToWebview({
-					type: "singleRouterModelFetchResponse",
-					success: true,
-					values: { provider: "roo", models: rooModels },
-				})
-			} catch (error) {
-				// Send error response
-				const errorMessage = error instanceof Error ? error.message : String(error)
-				provider.postMessageToWebview({
-					type: "singleRouterModelFetchResponse",
-					success: false,
-					error: errorMessage,
-					values: { provider: "roo" },
-				})
-			}
-			break
-		}
+		// case "requestRooModels": {
+		// 	// Specific handler for Roo models only - flushes cache to ensure fresh auth token is used
+		// 	try {
+		// 		// Flush cache first to ensure fresh models with current auth state
+		// 		await flushModels("roo")
+
+		// 		const rooModels = await getModels({
+		// 			provider: "roo",
+		// 			baseUrl: process.env.ROO_CODE_PROVIDER_URL ?? "https://api.roocode.com/proxy",
+		// 			apiKey: CloudService.hasInstance()
+		// 				? CloudService.instance.authService?.getSessionToken()
+		// 				: undefined,
+		// 		})
+
+		// 		// Always send a response, even if no models are returned
+		// 		provider.postMessageToWebview({
+		// 			type: "singleRouterModelFetchResponse",
+		// 			success: true,
+		// 			values: { provider: "roo", models: rooModels },
+		// 		})
+		// 	} catch (error) {
+		// 		// Send error response
+		// 		const errorMessage = error instanceof Error ? error.message : String(error)
+		// 		provider.postMessageToWebview({
+		// 			type: "singleRouterModelFetchResponse",
+		// 			success: false,
+		// 			error: errorMessage,
+		// 			values: { provider: "roo" },
+		// 		})
+		// 	}
+		// 	break
+		// }
 		case "requestOpenAiModels":
 			if (message?.values?.baseUrl && message?.values?.apiKey) {
 				const openAiModels = await getOpenAiModels(

src/integrations/terminal/TerminalRegistry.ts

@@ -62,7 +62,7 @@ export class TerminalRegistry {
 						terminal.busy = true // Mark terminal as busy when shell execution starts
 					} else {
 						console.error(
-							"[onDidStartTerminalShellExecution] Shell execution started, but not from a Roo-registered terminal:",
+							"[onDidStartTerminalShellExecution] Shell execution started, but not from a CoStrict-registered terminal:",
 							e,
 						)
 					}

src/utils/autoImportSettings.ts

@@ -9,7 +9,7 @@ import { t } from "../i18n"
 import { importSettingsFromPath, ImportOptions } from "../core/config/importExport"
 
 /**
- * Automatically imports RooCode settings from a specified path if it exists.
+ * Automatically imports CoStrict settings from a specified path if it exists.
  * This function is called during extension activation to allow users to pre-configure
  * their settings by placing a settings file at a predefined location.
  */

webview-ui/src/components/settings/About.tsx

@@ -63,8 +63,8 @@ export const About = ({ telemetrySetting, setTelemetrySetting, className, ...pro
 						components={{
 							zgsmGithubLink: <VSCodeLink href="https://github.com/zgsm-ai/costrict" />,
 							githubLink: <VSCodeLink href="https://github.com/zgsm-ai/costrict" />,
-							redditLink: <VSCodeLink href="https://reddit.com/r/RooCode" />,
-							discordLink: <VSCodeLink href="https://discord.gg/roocode" />,
+							// redditLink: <VSCodeLink href="https://reddit.com/r/RooCode" />,
+							// discordLink: <VSCodeLink href="https://discord.gg/roocode" />,
 						}}
 					/>
 				</div>

webview-ui/src/components/settings/providers/index.ts

@@ -18,7 +18,7 @@ export { OpenAI } from "./OpenAI"
 export { OpenAICompatible } from "./OpenAICompatible"
 export { OpenRouter } from "./OpenRouter"
 export { QwenCode } from "./QwenCode"
-export { Roo } from "./Roo"
+// export { Roo } from "./Roo"
 export { Requesty } from "./Requesty"
 export { SambaNova } from "./SambaNova"
 export { Unbound } from "./Unbound"

webview-ui/src/context/ExtensionStateContext.tsx

@@ -327,7 +327,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
 		isCodebaseReady: true,
 	})
 	const [includeTaskHistoryInEnhance, setIncludeTaskHistoryInEnhance] = useState(true)
-	const [prevCloudIsAuthenticated, setPrevCloudIsAuthenticated] = useState(false)
+	// const [prevCloudIsAuthenticated, setPrevCloudIsAuthenticated] = useState(false)
 	const [includeCurrentTime, setIncludeCurrentTime] = useState(true)
 	const [includeCurrentCost, setIncludeCurrentCost] = useState(true)
 	const [notices, setNotices] = useState<
@@ -513,15 +513,15 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
 	}, [])
 
 	// Watch for authentication state changes and refresh Roo models
-	useEffect(() => {
-		const currentAuth = state.cloudIsAuthenticated ?? false
-		const currentProvider = state.apiConfiguration?.apiProvider
-		if (!prevCloudIsAuthenticated && currentAuth && currentProvider === "roo") {
-			// User just authenticated and Roo is the active provider - refresh Roo models
-			vscode.postMessage({ type: "requestRooModels" })
-		}
-		setPrevCloudIsAuthenticated(currentAuth)
-	}, [state.cloudIsAuthenticated, prevCloudIsAuthenticated, state.apiConfiguration?.apiProvider])
+	// useEffect(() => {
+	// 	const currentAuth = state.cloudIsAuthenticated ?? false
+	// 	const currentProvider = state.apiConfiguration?.apiProvider
+	// 	if (!prevCloudIsAuthenticated && currentAuth && currentProvider === "roo") {
+	// 		// User just authenticated and Roo is the active provider - refresh Roo models
+	// 		vscode.postMessage({ type: "requestRooModels" })
+	// 	}
+	// 	setPrevCloudIsAuthenticated(currentAuth)
+	// }, [state.cloudIsAuthenticated, prevCloudIsAuthenticated, state.apiConfiguration?.apiProvider])
 
 	const contextValue: ExtensionStateContextType = {
 		...state,
@@ -541,7 +541,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
 		writeDelayMs: state.writeDelayMs,
 		screenshotQuality: state.screenshotQuality,
 		routerModels: extensionRouterModels,
-		cloudIsAuthenticated: state.cloudIsAuthenticated ?? false,
+		// cloudIsAuthenticated: state.cloudIsAuthenticated ?? false,
 		// cloudOrganizations: state.cloudOrganizations ?? [],
 		organizationSettingsVersion: state.organizationSettingsVersion ?? -1,
 		marketplaceItems,