feat: derive and cache model metadata from provider state for custom providers (#747)

packages/ui/src/stores/useConfigStore.ts

@@ -233,6 +233,38 @@ const buildModelMetadataKey = (providerId: string, modelId: string) => {
     return `${normalizedProvider}/${modelId}`;
 };
 
+const mapModalities = (cap: { text: boolean; audio: boolean; image: boolean; video: boolean; pdf: boolean }): string[] => {
+    const result: string[] = [];
+    if (cap.text) result.push('text');
+    if (cap.audio) result.push('audio');
+    if (cap.image) result.push('image');
+    if (cap.video) result.push('video');
+    if (cap.pdf) result.push('pdf');
+    return result;
+};
+
+const deriveModelMetadata = (providerId: string, model: ProviderModel): ModelMetadata => ({
+    id: model.id,
+    providerId,
+    name: model.name,
+    tool_call: model.capabilities.toolcall,
+    reasoning: model.capabilities.reasoning,
+    temperature: model.capabilities.temperature,
+    attachment: model.capabilities.attachment,
+    modalities: {
+        input: mapModalities(model.capabilities.input),
+        output: mapModalities(model.capabilities.output),
+    },
+    cost: {
+        input: model.cost.input,
+        output: model.cost.output,
+        cache_read: model.cost.cache.read,
+        cache_write: model.cost.cache.write,
+    },
+    limit: model.limit,
+    release_date: model.release_date,
+});
+
 const transformModelsDevResponse = (payload: unknown): Map<string, ModelMetadata> => {
     const metadataMap = new Map<string, ModelMetadata>();
 
@@ -1744,8 +1776,25 @@ export const useConfigStore = create<ConfigStore>()(
                     if (!key) {
                         return undefined;
                     }
-                    const { modelsMetadata } = get();
+                    const { modelsMetadata, providers } = get();
-                    return modelsMetadata.get(key);
+                    const cached = modelsMetadata.get(key);
+                    if (cached) {
+                        return cached;
+                    }
+
+                    // Fallback: derive metadata from provider model data (covers custom providers not in models.dev)
+                    const provider = providers.find((p) => p.id === providerId);
+                    if (!provider) {
+                        return undefined;
+                    }
+                    const model = provider.models.find((m) => m.id === modelId);
+                    if (!model) {
+                        return undefined;
+                    }
+
+                    const derived = deriveModelMetadata(providerId, model);
+                    set({ modelsMetadata: new Map(modelsMetadata).set(key, derived) });
+                    return derived;
                 },
                 getVisibleAgents: () => {
                     const { agents } = get();