feat(byok): fetch live models dynamically from provider api with user api key on input

memento-note/lib/ai/models-list.ts

@@ -1,3 +1,17 @@
+import { type AiGatewayProvider } from '@/lib/ai/router';
+
+// Base URLs mapping for providers
+const PROVIDER_URLS: Record<string, string> = {
+  openai: 'https://api.openai.com/v1',
+  deepseek: 'https://api.deepseek.com/v1',
+  openrouter: 'https://openrouter.ai/api/v1',
+  mistral: 'https://api.mistral.ai/v1',
+  zai: 'https://api.zukijourney.com/v1',
+  minimax: 'https://api.minimax.io/v1',
+  glm: 'https://open.bigmodel.ai/api/paas/v4',
+};
+
+// Fallback popular models when live fetching fails or for providers without /models endpoint (e.g. Anthropic, Google)
 export const PROVIDER_MODEL_SUGGESTIONS: Record<string, string[]> = {
   openai: ['gpt-4o-mini', 'gpt-4o', 'gpt-4-turbo', 'gpt-3.5-turbo'],
   anthropic: ['claude-3-5-sonnet-latest', 'claude-3-5-haiku-latest', 'claude-3-opus-latest'],
@@ -9,3 +23,61 @@ export const PROVIDER_MODEL_SUGGESTIONS: Record<string, string[]> = {
   openrouter: ['openai/gpt-4o-mini', 'anthropic/claude-3.5-sonnet', 'deepseek/deepseek-chat'],
   custom: [],
 };
+
+/**
+ * Dynamically queries the provider's /models endpoint using the user's API Key
+ * to fetch their actual available models list instead of relying on hardcoded choices.
+ */
+export async function fetchLiveModelsForProvider(
+  provider: AiGatewayProvider,
+  apiKey: string,
+  customBaseUrl?: string
+): Promise<string[]> {
+  try {
+    // Anthropic and Google do not expose a public list via a simple key GET /models (or need specific formats)
+    // We fall back to the popular defaults for those.
+    if (provider === 'anthropic' || provider === 'anthropic_custom' || provider === 'google') {
+      const standardProvider = provider === 'anthropic_custom' ? 'anthropic' : provider;
+      return PROVIDER_MODEL_SUGGESTIONS[standardProvider] ?? [];
+    }
+
+    const baseUrl = provider === 'custom'
+      ? customBaseUrl?.replace(/\/$/, '')
+      : PROVIDER_URLS[provider];
+
+    if (!baseUrl) {
+      return PROVIDER_MODEL_SUGGESTIONS[provider] ?? [];
+    }
+
+    const headers: Record<string, string> = { 'Content-Type': 'application/json' };
+    headers['Authorization'] = `Bearer ${apiKey}`;
+
+    if (provider === 'openrouter') {
+      headers['HTTP-Referer'] = 'https://localhost:3000';
+      headers['X-Title'] = 'Memento AI';
+    }
+
+    const response = await fetch(`${baseUrl}/models`, {
+      headers,
+      signal: AbortSignal.timeout(6000),
+    });
+
+    if (!response.ok) {
+      return PROVIDER_MODEL_SUGGESTIONS[provider] ?? [];
+    }
+
+    const data = await response.json();
+    const fetched: string[] = (data.data ?? [])
+      .map((m: any) => m.id || m.name)
+      .filter(Boolean)
+      .sort();
+
+    if (fetched.length > 0) {
+      return fetched;
+    }
+  } catch (err) {
+    console.warn(`[fetchLiveModelsForProvider] Failed to fetch live models for ${provider}, using fallbacks:`, err);
+  }
+
+  return PROVIDER_MODEL_SUGGESTIONS[provider] ?? [];
+}