From fabaf07a3fa7d6e3f2d7afc0c16fc94e42985a64 Mon Sep 17 00:00:00 2001
From: sam hoang <samhv.ict@gmail.com>
Date: Tue, 11 Feb 2025 01:51:50 +0700
Subject: [PATCH] pr comment

---
 .../src/components/settings/ModelPicker.tsx   | 22 ++++++++++---------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/webview-ui/src/components/settings/ModelPicker.tsx b/webview-ui/src/components/settings/ModelPicker.tsx
index f6b6197d82..deb1062fdc 100644
--- a/webview-ui/src/components/settings/ModelPicker.tsx
+++ b/webview-ui/src/components/settings/ModelPicker.tsx
@@ -23,17 +23,19 @@ import { vscode } from "../../utils/vscode"
 import { normalizeApiConfiguration } from "./ApiOptions"
 import { ModelInfoView } from "./ModelInfoView"
 
-interface ModelPickerProps {
+type ModelProvider = "glama" | "openRouter" | "unbound" | "requesty" | "openAi"
+
+type ModelKeys<T extends ModelProvider> = `${T}Models`
+type ConfigKeys<T extends ModelProvider> = `${T}ModelId`
+type InfoKeys<T extends ModelProvider> = `${T}ModelInfo`
+type RefreshMessageType<T extends ModelProvider> = `refresh${Capitalize<T>}Models`
+
+interface ModelPickerProps<T extends ModelProvider = ModelProvider> {
 	defaultModelId: string
-	modelsKey: "glamaModels" | "openRouterModels" | "unboundModels" | "requestyModels" | "openAiModels"
-	configKey: "glamaModelId" | "openRouterModelId" | "unboundModelId" | "requestyModelId" | "openAiModelId"
-	infoKey: "glamaModelInfo" | "openRouterModelInfo" | "unboundModelInfo" | "requestyModelInfo" | "openAiModelInfo"
-	refreshMessageType:
-		| "refreshGlamaModels"
-		| "refreshOpenRouterModels"
-		| "refreshUnboundModels"
-		| "refreshRequestyModels"
-		| "refreshOpenAiModels"
+	modelsKey: ModelKeys<T>
+	configKey: ConfigKeys<T>
+	infoKey: InfoKeys<T>
+	refreshMessageType: RefreshMessageType<T>
 	refreshValues?: Record<string, any>
 	serviceName: string
 	serviceUrl: string