fix: streamline model refresh by automatically flushing cache in handlers for LMStudio and Ollama components

Author: daniel-lxs

src/core/webview/webviewMessageHandler.ts

@@ -448,6 +448,9 @@ export const webviewMessageHandler = async (
 			// Specific handler for Ollama models only
 			const { apiConfiguration: ollamaApiConfig } = await provider.getState()
 			try {
+				// Flush cache first to ensure fresh models
+				await flushModels("ollama")
+
 				const ollamaModels = await getModels({
 					provider: "ollama",
 					baseUrl: ollamaApiConfig.ollamaBaseUrl,
@@ -469,6 +472,9 @@ export const webviewMessageHandler = async (
 			// Specific handler for LM Studio models only
 			const { apiConfiguration: lmStudioApiConfig } = await provider.getState()
 			try {
+				// Flush cache first to ensure fresh models
+				await flushModels("lmstudio")
+
 				const lmStudioModels = await getModels({
 					provider: "lmstudio",
 					baseUrl: lmStudioApiConfig.lmStudioBaseUrl,

webview-ui/src/components/settings/providers/LMStudio.tsx

@@ -52,15 +52,8 @@ export const LMStudio = ({ apiConfiguration, setApiConfigurationField }: LMStudi
 
 	// Refresh models on mount
 	useEffect(() => {
-		// Flush cache first to ensure we get fresh models
-		vscode.postMessage({ type: "flushRouterModels", text: "lmstudio" })
-
-		// Request fresh LM Studio models after a small delay to ensure cache is flushed
-		const timer = setTimeout(() => {
-			vscode.postMessage({ type: "requestLmStudioModels" })
-		}, 100)
-
-		return () => clearTimeout(timer)
+		// Request fresh models - the handler now flushes cache automatically
+		vscode.postMessage({ type: "requestLmStudioModels" })
 	}, [])
 
 	// Check if the selected model exists in the fetched models

webview-ui/src/components/settings/providers/Ollama.tsx

@@ -51,15 +51,8 @@ export const Ollama = ({ apiConfiguration, setApiConfigurationField }: OllamaPro
 
 	// Refresh models on mount
 	useEffect(() => {
-		// Flush cache first to ensure we get fresh models
-		vscode.postMessage({ type: "flushRouterModels", text: "ollama" })
-
-		// Request fresh Ollama models after a small delay to ensure cache is flushed
-		const timer = setTimeout(() => {
-			vscode.postMessage({ type: "requestOllamaModels" })
-		}, 100)
-
-		return () => clearTimeout(timer)
+		// Request fresh models - the handler now flushes cache automatically
+		vscode.postMessage({ type: "requestOllamaModels" })
 	}, [])
 
 	// Check if the selected model exists in the fetched models