refactor: remove requestHuggingFaceModels and use requestRouterModels instead

- Removed requestHuggingFaceModels message type from WebviewMessage.ts
- Removed requestHuggingFaceModels handler from webviewMessageHandler.ts
- Removed huggingFaceModels from ExtensionMessage.ts
- Updated HuggingFace.tsx component to use requestRouterModels and extract HuggingFace models from the routerModels response

This simplifies the codebase by using a single unified message type for all router-based model providers.

Author: roomote

src/core/webview/webviewMessageHandler.ts

@@ -522,7 +522,6 @@ export const webviewMessageHandler = async (
 				litellm: {},
 				ollama: {},
 				lmstudio: {},
-				huggingface: {},
 			}
 
 			const safeGetModels = async (options: GetModelsOptions): Promise<ModelRecord> => {
@@ -565,10 +564,9 @@ export const webviewMessageHandler = async (
 
 			const fetchedRouterModels: Partial<Record<RouterName, ModelRecord>> = {
 				...routerModels,
-				// Initialize ollama, lmstudio, and huggingface with empty objects since they use separate handlers
+				// Initialize ollama and lmstudio with empty objects since they use separate handlers
 				ollama: {},
 				lmstudio: {},
-				huggingface: {},
 			}
 
 			results.forEach((result, index) => {
@@ -676,59 +674,6 @@ export const webviewMessageHandler = async (
 			// TODO: Cache like we do for OpenRouter, etc?
 			provider.postMessageToWebview({ type: "vsCodeLmModels", vsCodeLmModels })
 			break
-		case "requestHuggingFaceModels":
-			try {
-				// Flush cache first to ensure fresh models
-				await flushModels("huggingface")
-
-				const huggingFaceModels = await getModels({
-					provider: "huggingface",
-				})
-
-				// Convert the model record to an array format expected by the webview
-				const modelArray = Object.entries(huggingFaceModels).map(([id, info]) => ({
-					id,
-					_id: id,
-					inferenceProviderMapping: [
-						{
-							provider: "huggingface",
-							providerId: id,
-							status: "live" as const,
-							task: "conversational" as const,
-						},
-					],
-					trendingScore: 0,
-					config: {
-						architectures: [],
-						model_type:
-							info.description
-								?.split(", ")
-								.find((part) => part.startsWith("Type: "))
-								?.replace("Type: ", "") || "",
-						tokenizer_config: {
-							model_max_length: info.contextWindow,
-						},
-					},
-					tags: [],
-					pipeline_tag: info.supportsImages ? ("image-text-to-text" as const) : ("text-generation" as const),
-					library_name: info.description
-						?.split(", ")
-						.find((part) => part.startsWith("Library: "))
-						?.replace("Library: ", ""),
-				}))
-
-				provider.postMessageToWebview({
-					type: "huggingFaceModels",
-					huggingFaceModels: modelArray,
-				})
-			} catch (error) {
-				console.error("Failed to fetch Hugging Face models:", error)
-				provider.postMessageToWebview({
-					type: "huggingFaceModels",
-					huggingFaceModels: [],
-				})
-			}
-			break
 		case "openImage":
 			openImage(message.text!, { values: message.values })
 			break

src/shared/ExtensionMessage.ts

@@ -67,7 +67,6 @@ export interface ExtensionMessage {
 		| "ollamaModels"
 		| "lmStudioModels"
 		| "vsCodeLmModels"
-		| "huggingFaceModels"
 		| "vsCodeLmApiAvailable"
 		| "updatePrompt"
 		| "systemPrompt"
@@ -137,28 +136,6 @@ export interface ExtensionMessage {
 	ollamaModels?: string[]
 	lmStudioModels?: string[]
 	vsCodeLmModels?: { vendor?: string; family?: string; version?: string; id?: string }[]
-	huggingFaceModels?: Array<{
-		_id: string
-		id: string
-		inferenceProviderMapping: Array<{
-			provider: string
-			providerId: string
-			status: "live" | "staging" | "error"
-			task: "conversational"
-		}>
-		trendingScore: number
-		config: {
-			architectures: string[]
-			model_type: string
-			tokenizer_config?: {
-				chat_template?: string | Array<{ name: string; template: string }>
-				model_max_length?: number
-			}
-		}
-		tags: string[]
-		pipeline_tag: "text-generation" | "image-text-to-text"
-		library_name?: string
-	}>
 	mcpServers?: McpServer[]
 	commits?: GitCommit[]
 	listApiConfig?: ProviderSettingsEntry[]

src/shared/WebviewMessage.ts

@@ -67,7 +67,6 @@ export interface WebviewMessage {
 		| "requestOllamaModels"
 		| "requestLmStudioModels"
 		| "requestVsCodeLmModels"
-		| "requestHuggingFaceModels"
 		| "openImage"
 		| "saveImage"
 		| "openFile"

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

@@ -62,16 +62,54 @@ export const HuggingFace = ({ apiConfiguration, setApiConfigurationField }: Hugg
 	// Fetch models when component mounts
 	useEffect(() => {
 		setLoading(true)
-		vscode.postMessage({ type: "requestHuggingFaceModels" })
+		vscode.postMessage({ type: "requestRouterModels" })
 	}, [])
 
 	// Handle messages from extension
 	const onMessage = useCallback((event: MessageEvent) => {
 		const message: ExtensionMessage = event.data
 
 		switch (message.type) {
-			case "huggingFaceModels":
-				setModels(message.huggingFaceModels || [])
+			case "routerModels":
+				// Extract HuggingFace models from routerModels
+				if (message.routerModels?.huggingface) {
+					// Convert from ModelRecord format to HuggingFaceModel array format
+					const modelArray = Object.entries(message.routerModels.huggingface).map(([id, info]) => ({
+						id,
+						_id: id,
+						inferenceProviderMapping: [
+							{
+								provider: "huggingface",
+								providerId: id,
+								status: "live" as const,
+								task: "conversational" as const,
+							},
+						],
+						trendingScore: 0,
+						config: {
+							architectures: [],
+							model_type:
+								info.description
+									?.split(", ")
+									.find((part: string) => part.startsWith("Type: "))
+									?.replace("Type: ", "") || "",
+							tokenizer_config: {
+								model_max_length: info.contextWindow,
+							},
+						},
+						tags: [],
+						pipeline_tag: info.supportsImages
+							? ("image-text-to-text" as const)
+							: ("text-generation" as const),
+						library_name: info.description
+							?.split(", ")
+							.find((part: string) => part.startsWith("Library: "))
+							?.replace("Library: ", ""),
+					}))
+					setModels(modelArray)
+				} else {
+					setModels([])
+				}
 				setLoading(false)
 				break
 		}