fix: custom handler for fetching requesty models

src/core/webview/webviewMessageHandler.ts

@@ -946,6 +946,23 @@ export const webviewMessageHandler = async (
 			}
 			break
 		}
+		case "requestRequestyModels": {
+			// Specific handler for Requesty models only.
+			const { apiConfiguration: requestyApiConfiguration } = await provider.getState()
+			const requestyOptions = {
+				provider: "requesty" as const,
+				baseUrl: requestyApiConfiguration.requestyBaseUrl,
+				apiKey: requestyApiConfiguration.requestyApiKey,
+			}
+
+			// Flush cache and refresh to ensure fresh models.
+			await flushModels(requestyOptions, true)
+
+			const requestyModels = await getModels(requestyOptions)
+
+			provider.postMessageToWebview({ type: "requestyModels", requestyModels: requestyModels })
+			break
+		}
 		case "requestLmStudioModels": {
 			// Specific handler for LM Studio models only.
 			const { apiConfiguration: lmStudioApiConfig } = await provider.getState()

src/shared/ExtensionMessage.ts

@@ -82,6 +82,7 @@ export interface ExtensionMessage {
 		| "lmStudioModels"
 		| "vsCodeLmModels"
 		| "huggingFaceModels"
+		| "requestyModels"
 		| "vsCodeLmApiAvailable"
 		| "updatePrompt"
 		| "systemPrompt"
@@ -184,6 +185,7 @@ export interface ExtensionMessage {
 			}
 		}>
 	}>
+	requestyModels?: ModelRecord
 	mcpServers?: McpServer[]
 	commits?: GitCommit[]
 	listApiConfig?: ProviderSettingsEntry[]

src/shared/WebviewMessage.ts

@@ -63,6 +63,7 @@ export interface WebviewMessage {
 		| "requestRooCreditBalance"
 		| "requestVsCodeLmModels"
 		| "requestHuggingFaceModels"
+		| "requestRequestyModels"
 		| "openImage"
 		| "saveImage"
 		| "openFile"

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

@@ -3,7 +3,7 @@ import { VSCodeCheckbox, VSCodeTextField } from "@vscode/webview-ui-toolkit/reac
 
 import { type ProviderSettings, type OrganizationAllowList, requestyDefaultModelId } from "@roo-code/types"
 
-import type { RouterModels } from "@roo/api"
+import type { ModelRecord, RouterModels } from "@roo/api"
 
 import { vscode } from "@src/utils/vscode"
 import { useAppTranslation } from "@src/i18n/TranslationContext"
@@ -14,6 +14,8 @@ import { ModelPicker } from "../ModelPicker"
 import { RequestyBalanceDisplay } from "./RequestyBalanceDisplay"
 import { getCallbackUrl } from "@/oauth/urls"
 import { toRequestyServiceUrl } from "@roo/utils/requesty"
+import { useEvent } from "react-use"
+import { ExtensionMessage } from "@roo/ExtensionMessage"
 
 type RequestyProps = {
 	apiConfiguration: ProviderSettings
@@ -29,7 +31,6 @@ type RequestyProps = {
 export const Requesty = ({
 	apiConfiguration,
 	setApiConfigurationField,
-	routerModels,
 	refetchRouterModels,
 	organizationAllowList,
 	modelValidationError,
@@ -38,6 +39,8 @@ export const Requesty = ({
 }: RequestyProps) => {
 	const { t } = useAppTranslation()
 
+	const [requestyModels, setRequestyModels] = useState<ModelRecord>({})
+
 	const [requestyEndpointSelected, setRequestyEndpointSelected] = useState(!!apiConfiguration.requestyBaseUrl)
 
 	// This ensures that the "Use custom URL" checkbox is hidden when the user deletes the URL.
@@ -65,6 +68,25 @@ export const Requesty = ({
 		return authUrl.toString()
 	}
 
+	const onMessage = useCallback((event: MessageEvent) => {
+		const message: ExtensionMessage = event.data
+
+		if (message.type !== "requestyModels") {
+			return
+		}
+
+		const newModels = message.requestyModels ?? {}
+		setRequestyModels(newModels)
+	}, [])
+
+	useEvent("message", onMessage)
+
+	// Refresh models on mount
+	useEffect(() => {
+		// Request fresh models - the handler now flushes cache automatically
+		vscode.postMessage({ type: "requestRequestyModels" })
+	}, [])
+
 	return (
 		<>
 			<VSCodeTextField
@@ -127,7 +149,7 @@ export const Requesty = ({
 			<Button
 				variant="outline"
 				onClick={() => {
-					vscode.postMessage({ type: "flushRouterModels", text: "requesty" })
+					vscode.postMessage({ type: "requestRequestyModels" })
 					refetchRouterModels()
 				}}>
 				<div className="flex items-center gap-2">
@@ -139,7 +161,7 @@ export const Requesty = ({
 				apiConfiguration={apiConfiguration}
 				setApiConfigurationField={setApiConfigurationField}
 				defaultModelId={requestyDefaultModelId}
-				models={routerModels?.requesty ?? {}}
+				models={requestyModels}
 				modelIdKey="requestyModelId"
 				serviceName="Requesty"
 				serviceUrl="https://requesty.ai"