osai-agent 4.0.0

package/src/commands/provider.js

@@ -0,0 +1,633 @@
+import chalk from 'chalk';
+import Conf from 'conf';
+import inquirer from 'inquirer';
+import ora from 'ora';
+import { printError, printSuccess, printInfo, printNotLoggedIn } from '../ui/terminal.js';
+import { toHttpUrl } from '../services/server-url.js';
+import { encrypt, decrypt, deriveKey } from '../services/crypto.js';
+import pkg from 'node-machine-id';
+const { machineIdSync } = pkg;
+
+const getConfig = () => new Conf({ projectName: 'osai-agent' });
+const getServer = () => toHttpUrl(getConfig().get('server'));
+const getToken = () => getConfig().get('token');
+const getHeaders = () => ({ 'Authorization': `Bearer ${getToken()}`, 'Content-Type': 'application/json' });
+
+const isLocalMode = () => {
+  const argv = process.argv.slice(2);
+  return argv.includes('--local');
+};
+
+const maskKey = (key) => {
+  if (!key) return null;
+  if (key.length <= 4) return key;
+  return key.slice(0, 3) + '...' + key.slice(-4);
+};
+
+export function migrateLocalProvider() {
+  const config = getConfig();
+  const old = config.get('localProvider');
+  if (!old || config.get('localProviders')) return;
+  const parsed = typeof old === 'string' ? JSON.parse(old) : old;
+  const { type, apiKeyEnc, model, baseUrl } = parsed;
+  if (type) {
+    config.set('localProviders', { [type]: { apiKeyEnc: apiKeyEnc || null, model: model || null, baseUrl: baseUrl || null } });
+    config.set('currentLocalProvider', type);
+  }
+  config.delete('localProvider');
+}
+
+export function saveLocalProvider(type, apiKey, model, baseUrl) {
+  migrateLocalProvider();
+  const config = getConfig();
+  const key = deriveKey(machineIdSync());
+  const providers = config.get('localProviders', {});
+  const existing = providers[type] || {};
+  providers[type] = {
+    apiKeyEnc: apiKey ? encrypt(apiKey, key) : (existing.apiKeyEnc || null),
+    model: model || existing.model || null,
+    baseUrl: baseUrl || existing.baseUrl || null,
+  };
+  config.set('localProviders', providers);
+  config.set('currentLocalProvider', type);
+}
+
+export function getLocalProvider() {
+  migrateLocalProvider();
+  const config = getConfig();
+  const current = config.get('currentLocalProvider');
+  if (!current) return null;
+  const providers = config.get('localProviders', {});
+  const entry = providers[current];
+  if (!entry) return null;
+  const key = deriveKey(machineIdSync());
+  const apiKey = entry.apiKeyEnc ? decrypt(entry.apiKeyEnc, key) : null;
+  return { type: current, apiKey, model: entry.model || null, baseUrl: entry.baseUrl || null };
+}
+
+export function getAllLocalProviders() {
+  migrateLocalProvider();
+  const config = getConfig();
+  const current = config.get('currentLocalProvider');
+  const providers = config.get('localProviders', {});
+  return Object.entries(providers).map(([type, data]) => {
+    const key = deriveKey(machineIdSync());
+    const apiKey = data.apiKeyEnc ? decrypt(data.apiKeyEnc, key) : null;
+    return { type, apiKey, model: data.model || null, baseUrl: data.baseUrl || null, active: type === current };
+  });
+}
+
+export function removeLocalProvider(type) {
+  migrateLocalProvider();
+  const config = getConfig();
+  if (type) {
+    const providers = config.get('localProviders', {});
+    delete providers[type];
+    config.set('localProviders', providers);
+    if (config.get('currentLocalProvider') === type) {
+      const remaining = Object.keys(providers);
+      config.set('currentLocalProvider', remaining.length > 0 ? remaining[0] : null);
+    }
+  } else {
+    config.delete('localProviders');
+    config.delete('currentLocalProvider');
+  }
+}
+
+export const listProviders = async () => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+  const spinner = ora('Fetching provider catalog...').start();
+  try {
+    const response = await fetch(`${getServer()}/api/provider/catalog`, { headers: getHeaders() });
+    if (!response.ok) { spinner.fail('Failed to fetch catalog'); const e = await response.json(); printError(e.error || 'Unknown error'); return; }
+    const catalog = await response.json();
+    spinner.stop();
+
+    const active = catalog.find(p => p.active);
+    console.log();
+    console.log(chalk.hex('#7aa2f7').bold('  Provider Catalog'));
+    console.log(chalk.hex('#565f89')('  ─'.repeat(48)));
+    console.log(chalk.hex('#565f89')(`  ${'Provider'.padEnd(20)} SDK Type   Free   Models`));
+    console.log(chalk.hex('#565f89')('  ─'.repeat(48)));
+
+    for (const p of catalog) {
+      const activeMark = p.active ? chalk.green(' ◀') : '';
+      const freeMark = p.free_tier ? chalk.green('Yes') : chalk.red('No ');
+      const name = p.id === active?.id ? chalk.white.bold(p.name) : chalk.white(p.name);
+      const sdk = p.sdk_type === 'anthropic' ? 'native  ' : 'OpenAI  ';
+      console.log(`  ${(name + activeMark).padEnd(22)} ${sdk}   ${freeMark}   ${chalk.gray(p.models_count + ' models')}`);
+    }
+    console.log(chalk.hex('#565f89')('  ─'.repeat(48)));
+    console.log(chalk.gray(`  ${active?.name || 'OS AI Agent'} is currently active\n`));
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+const manualModelSelection = async (providerType) => {
+  printInfo(`Fetching available models for ${providerType}...`);
+  try {
+    const response = await fetch(`${getServer()}/api/provider/models/${providerType}`, { headers: getHeaders() });
+    if (!response.ok) return null;
+    const data = await response.json();
+    if (!data.models || data.models.length === 0) return null;
+
+    const modelNames = data.models.map(m => (typeof m === 'string' ? m : (m.id || m.name))).filter(Boolean);
+    if (modelNames.length === 0) return null;
+
+    console.log(chalk.hex('#7aa2f7').bold(`\n  Select a model for ${providerType}:`));
+    console.log();
+
+    const choices = modelNames.slice(0, 50).map((name, i) => ({
+      name: `${String(i + 1).padStart(2)}. ${name}`,
+      value: name
+    }));
+    choices.push({ name: `  0. Enter model ID manually`, value: '__manual__' });
+
+    for (const c of choices) {
+      console.log(chalk.hex('#565f89')(`  ${c.name}`));
+    }
+    console.log();
+
+    const { selection } = await inquirer.prompt({
+      type: 'input',
+      name: 'selection',
+      message: 'Enter number (0 for manual)',
+      validate: (v) => {
+        const num = parseInt(v);
+        if (isNaN(num) || num < 0 || num > choices.length - 1) return `Enter a number between 0 and ${choices.length - 1}`;
+        return true;
+      }
+    });
+
+    const num = parseInt(selection);
+    if (num === 0) {
+      const { manual } = await inquirer.prompt({
+        type: 'input',
+        name: 'manual',
+        message: 'Enter model ID',
+        validate: (v) => v.trim().length > 0 || 'Model ID required'
+      });
+      return manual.trim();
+    }
+
+    return choices[num].value;
+  } catch {
+    return null;
+  }
+};
+
+const showOllamaTip = () => {
+  console.log(chalk.hex('#565f89')('  Tip: Make sure Ollama is running: ollama serve'));
+};
+
+export const setProvider = async (provider, options) => {
+  if (!provider) { printError('Usage: osai-agent provider set <provider> [--key <key>] [--model <model>] [--url <url>]'); return; }
+
+  const keylessProviders = ['ollama', 'lmstudio', 'vllm'];
+
+  if (isLocalMode()) {
+    const key = options.key;
+    const model = options.model;
+    const url = options.url;
+    saveLocalProvider(provider, key, model, url);
+    const keyDisplay = key ? ` | Key: ${maskKey(key)}` : '';
+    const modelDisplay = model ? ` | Model: ${model}` : '';
+    const urlDisplay = url ? ` | URL: ${url}` : '';
+    console.log();
+    printSuccess(`Local provider set to ${provider}${modelDisplay}${keyDisplay}${urlDisplay}`);
+    if (keylessProviders.includes(provider)) {
+      printInfo(`Tip: Make sure ${provider} is running locally`);
+    }
+    console.log();
+    return;
+  }
+
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  if (keylessProviders.includes(provider) && !options.key && !options.model) {
+    printInfo(`No API key required for ${provider} (local)`);
+    const spinner = ora(`Fetching installed models from local server...`).start();
+    try {
+      const catResp = await fetch(`${getServer()}/api/provider/catalog`, { headers: getHeaders() });
+      const catalog = catResp.ok ? await catResp.json() : [];
+      const entry = catalog.find(p => p.id === provider);
+      const defaultUrl = entry?.base_url || 'http://localhost:11434/v1';
+
+      if (!options.url) {
+        options.url = defaultUrl;
+      }
+
+      // Save provider first with no key
+      const saveResp = await fetch(`${getServer()}/api/provider`, {
+        method: 'PUT',
+        headers: getHeaders(),
+        body: JSON.stringify({ type: provider, base_url: options.url })
+      });
+      if (!saveResp.ok) { spinner.fail('Failed to set provider'); const e = await saveResp.json(); printError(e.error || 'Unknown error'); return; }
+
+      spinner.succeed(`Provider set to ${provider}`);
+
+      const model = await manualModelSelection(provider);
+      if (model) {
+        const spinner2 = ora('Saving model selection...').start();
+        await fetch(`${getServer()}/api/provider`, {
+          method: 'PUT',
+          headers: getHeaders(),
+          body: JSON.stringify({ type: provider, model, base_url: options.url })
+        });
+        spinner2.succeed('Model updated');
+        printSuccess(`Provider set to ${provider} | Model: ${model}`);
+      } else {
+        if (provider === 'ollama') showOllamaTip();
+        printSuccess(`Provider set to ${provider}`);
+      }
+    } catch (err) {
+      spinner.fail('Could not connect to local provider');
+      if (provider === 'ollama') {
+        printError(`Could not connect to Ollama at ${options.url || 'http://localhost:11434'}`);
+        showOllamaTip();
+        printInfo('Showing empty model list — run \'ollama pull llama3.3\' to install models');
+      } else {
+        printError(err.message);
+      }
+      // Still save the provider selection even if models can't be fetched
+      const saveResp = await fetch(`${getServer()}/api/provider`, {
+        method: 'PUT',
+        headers: getHeaders(),
+        body: JSON.stringify({ type: provider, base_url: options.url })
+      });
+      if (saveResp.ok) printSuccess(`Provider set to ${provider}`);
+    }
+    return;
+  }
+
+  // Providers that need an API key
+  const spinner = ora(`Setting provider to ${provider}...`).start();
+  try {
+    const body = { type: provider };
+    if (options.key) body.api_key = options.key;
+    if (options.model) body.model = options.model;
+    if (options.url) body.base_url = options.url;
+
+    const response = await fetch(`${getServer()}/api/provider`, {
+      method: 'PUT',
+      headers: getHeaders(),
+      body: JSON.stringify(body)
+    });
+    const data = await response.json();
+    if (!response.ok) { spinner.fail('Failed to set provider'); printError(data.error || 'Unknown error'); return; }
+    spinner.succeed('Provider updated');
+
+    const keyDisplay = data.key ? ` | Key: ${data.key}` : '';
+    printSuccess(`API key saved for ${data.type}${keyDisplay}`);
+
+    // Auto-discover models if no model was specified
+    if (!options.model) {
+      const model = await manualModelSelection(provider);
+      if (model) {
+        const spinner2 = ora('Saving model selection...').start();
+        const body2 = { type: provider, model };
+        if (options.key) body2.api_key = options.key;
+        if (options.url) body2.base_url = options.url;
+        await fetch(`${getServer()}/api/provider`, {
+          method: 'PUT',
+          headers: getHeaders(),
+          body: JSON.stringify(body2)
+        });
+        spinner2.succeed('Model updated');
+        printSuccess(`Provider set to ${data.type} | Model: ${model}`);
+      } else {
+        printSuccess(`Provider set to ${data.type}`);
+      }
+    } else {
+      const modelDisplay = data.model ? ` | Model: ${data.model}` : '';
+      printSuccess(`Provider set to ${data.type}${modelDisplay}${keyDisplay}`);
+    }
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+export const showProviderModels = async (provider) => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+  if (!provider) { printError('Usage: osai-agent provider models <provider>'); return; }
+
+  const spinner = ora(`Fetching models for ${provider}...`).start();
+  try {
+    const response = await fetch(`${getServer()}/api/provider/models/${provider}`, { headers: getHeaders() });
+    if (!response.ok) { spinner.fail('Failed to fetch models'); const e = await response.json(); printError(e.error || 'Unknown error'); return; }
+    const data = await response.json();
+    spinner.stop();
+
+    const sourceColors = {
+      'live (local)': chalk.green,
+      'live (public)': chalk.cyan,
+      'live (authenticated)': chalk.blue,
+      cached: chalk.magenta,
+      static: chalk.yellow
+    };
+    const colorFn = sourceColors[data.source] || chalk.gray;
+    const sourceLabel = colorFn(data.source);
+
+    console.log();
+    console.log(chalk.hex('#7aa2f7').bold(`  ${data.provider} — Available Models`));
+    console.log(chalk.hex('#565f89')(`  Source: ${sourceLabel}`));
+    if (data.warning) {
+      console.log(chalk.hex('#ff9e64')(`  ⚠ ${data.warning}`));
+    }
+
+    if (provider === 'ollama') {
+      const hasSize = data.models.some(m => typeof m === 'object' && m.size);
+      if (hasSize) {
+        const totalGb = data.models.reduce((sum, m) => sum + (m.size ? m.size / 1e9 : 0), 0);
+        console.log(chalk.hex('#565f89')(`  Installed models (${data.models.length}):`));
+        console.log();
+        for (const m of data.models) {
+          const sizeGb = m.size ? (m.size / 1e9).toFixed(1) : '?';
+          console.log(`  ${chalk.white(m.id.padEnd(30))} ${chalk.gray(`(${sizeGb} GB)`)}`);
+        }
+        console.log(chalk.gray(`  Total: ${totalGb.toFixed(1)} GB used`));
+      } else {
+        data.models.forEach((m, i) => {
+          const name = typeof m === 'string' ? m : (m.id || m.name || m);
+          console.log(chalk.hex('#565f89')(`  ${String(i + 1).padStart(2)}.`) + ` ${chalk.white(name)}`);
+        });
+      }
+      console.log(chalk.gray(`  Tip: Install more: ollama pull <model>`));
+    } else if (provider === 'openrouter') {
+      const totalModels = data.models?.length || 0;
+      console.log(chalk.hex('#565f89')(`  ${totalModels} models available`));
+      console.log(chalk.hex('#565f89')('  Tip: Use vendor/model:free for free models (e.g. meta-llama/llama-4-maverick:free)'));
+      console.log(chalk.hex('#565f89')('  Browse all: https://openrouter.ai/models'));
+      console.log();
+      const displayLimit = 50;
+      const showAll = process.argv.includes('--all');
+      const toShow = showAll ? data.models : data.models.slice(0, displayLimit);
+      toShow.forEach((m, i) => {
+        console.log(chalk.hex('#565f89')(`  ${String(i + 1).padStart(2)}.`) + ` ${chalk.white(m)}`);
+      });
+      if (!showAll && data.models.length > displayLimit) {
+        console.log(chalk.gray(`  ... and ${data.models.length - displayLimit} more (use --all to see all)`));
+      }
+    } else {
+      console.log();
+      data.models.forEach((m, i) => {
+        const name = typeof m === 'string' ? m : (m.id || m.name || m);
+        console.log(chalk.hex('#565f89')(`  ${String(i + 1).padStart(2)}.`) + ` ${chalk.white(name)}`);
+      });
+    }
+    console.log();
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+export const showProvider = async () => {
+  if (isLocalMode()) {
+    const all = getAllLocalProviders();
+    console.log();
+    console.log(chalk.hex('#7aa2f7').bold('  Local Provider Configuration'));
+    console.log(chalk.hex('#565f89')('  ─'.repeat(40)));
+    if (all.length === 0) {
+      console.log(`  ${chalk.gray('No providers configured.')}`);
+      console.log(`  ${chalk.gray('  Use: osai-agent provider set <type> --key <key> --local')}`);
+    } else {
+      for (const p of all) {
+        const activeMark = p.active ? chalk.green(' ◀ active') : '';
+        console.log(`  ${chalk.white(p.type.padEnd(16))}${activeMark}`);
+        console.log(`  ${chalk.gray(`    Model  : ${p.model || 'auto'}`)}`);
+        console.log(`  ${chalk.gray(`    Key    : ${p.apiKey ? '(encrypted, stored locally)' : 'N/A'}`)}`);
+        console.log(`  ${chalk.gray(`    Base   : ${p.baseUrl || 'default'}`)}`);
+      }
+    }
+    console.log();
+    return;
+  }
+
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  const spinner = ora('Fetching provider config...').start();
+  try {
+    const response = await fetch(`${getServer()}/api/provider`, { headers: getHeaders() });
+    if (!response.ok) { spinner.fail('Failed to fetch provider'); const e = await response.json(); printError(e.error || 'Unknown error'); return; }
+    const data = await response.json();
+    spinner.stop();
+
+    console.log();
+    console.log(chalk.hex('#7aa2f7').bold('  Provider Configuration'));
+    console.log(chalk.hex('#565f89')('  ─'.repeat(32)));
+
+    if (data.type === 'osai') {
+      console.log(`  ${chalk.white('Current provider :')} ${chalk.green('OS AI Agent')}`);
+      console.log(`  ${chalk.white('Model           :')} ${chalk.gray('auto')}`);
+      console.log(`  ${chalk.white('API Key         :')} ${chalk.gray('N/A (OS AI managed)')}`);
+      console.log(`  ${chalk.white('Free tier       :')} ${chalk.green('Yes')}`);
+      console.log(`  ${chalk.white('Quota           :')} ${chalk.yellow('50 commands/day (free plan)')}`);
+    } else {
+      const catalogResponse = await fetch(`${getServer()}/api/provider/catalog`, { headers: getHeaders() });
+      const catalog = catalogResponse.ok ? await catalogResponse.json() : [];
+      const entry = catalog.find(p => p.id === data.type);
+
+      console.log(`  ${chalk.white('Current provider :')} ${chalk.green(data.type)}`);
+      console.log(`  ${chalk.white('Model           :')} ${chalk.white(data.model || 'N/A')}`);
+      console.log(`  ${chalk.white('API Key         :')} ${chalk.gray(maskKey(data.key) || 'N/A')}`);
+      console.log(`  ${chalk.white('Base URL        :')} ${chalk.gray(data.base_url || 'N/A')}`);
+      console.log(`  ${chalk.white('Custom name     :')} ${chalk.gray(data.custom_name || 'N/A')}`);
+      console.log(`  ${chalk.white('Free tier       :')} ${entry?.free_tier ? chalk.green('Yes') : chalk.red('No')}`);
+      console.log(`  ${chalk.white('Quota           :')} ${chalk.green('N/A (BYOK)')}`);
+    }
+    console.log();
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+export const resetProvider = async () => {
+  if (isLocalMode()) {
+    removeLocalProvider();
+    printSuccess('All local providers reset');
+    return;
+  }
+
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  const { confirm } = await inquirer.prompt({
+    type: 'confirm',
+    name: 'confirm',
+    message: 'Reset to OS AI Agent provider?',
+    default: false
+  });
+
+  if (!confirm) { printInfo('Reset cancelled'); return; }
+
+  const spinner = ora('Resetting provider...').start();
+  try {
+    const response = await fetch(`${getServer()}/api/provider`, {
+      method: 'DELETE',
+      headers: getHeaders()
+    });
+    if (!response.ok) { spinner.fail('Failed to reset'); const e = await response.json(); printError(e.error || 'Unknown error'); return; }
+    spinner.succeed('Provider reset');
+    printSuccess('Provider reset to OS AI Agent');
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+export const addCustomProvider = async (options) => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  const name = options.name;
+  const url = options.url;
+  const key = options.key;
+  const model = options.model;
+
+  if (!name || !url || !key) {
+    printError('Usage: osai-agent provider add custom --name <name> --url <base-url> --key <api-key> [--model <model>]');
+    return;
+  }
+
+  const spinner = ora('Adding custom provider...').start();
+  try {
+    const response = await fetch(`${getServer()}/api/provider`, {
+      method: 'PUT',
+      headers: getHeaders(),
+      body: JSON.stringify({ type: 'custom', api_key: key, base_url: url, model: model || null, custom_name: name })
+    });
+    const data = await response.json();
+    if (!response.ok) { spinner.fail('Failed to add custom provider'); printError(data.error || 'Unknown error'); return; }
+    spinner.succeed('Custom provider added');
+    printSuccess(`Custom provider "${name}" configured | Key: ${maskKey(key)}`);
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+// ── listConfiguredProviders — Show all saved providers ──────────────
+export const listConfiguredProviders = async () => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  const spinner = ora('Fetching configured providers...').start();
+  try {
+    const res = await fetch(`${getServer()}/api/provider/keys`, { headers: getHeaders() });
+    if (!res.ok) { spinner.fail('Failed to fetch providers'); const e = await res.json(); printError(e.error || 'Unknown error'); return; }
+    const providers = await res.json();
+    spinner.stop();
+
+    if (providers.length === 0) {
+      console.log();
+      console.log(chalk.hex('#565f89')('  No providers configured yet.'));
+      console.log(chalk.hex('#565f89')('  Use: osai-agent provider set <type> --key <key>'));
+      console.log();
+      return;
+    }
+
+    console.log();
+    console.log(chalk.hex('#7aa2f7').bold('  Configured Providers'));
+    console.log(chalk.hex('#565f89')('  ---'));
+    for (const p of providers) {
+      const activeMark = p.active ? chalk.green(' (active)') : '';
+      const modelStr = p.model ? ` | Model: ${chalk.white(p.model)}` : '';
+      const keyStr = p.key_masked ? ` | Key: ${chalk.gray(p.key_masked)}` : '';
+      console.log(`  ${chalk.white(p.name.padEnd(22))}${activeMark}${modelStr}${keyStr}`);
+    }
+    console.log(chalk.hex('#565f89')('  ---'));
+    console.log(chalk.gray('  Use: osai-agent provider switch <type> to activate'));
+    console.log();
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+// ── changeModel — Change model of a configured provider ────────────
+export const changeModel = async (type, model) => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  const spinner = ora(`Updating model for ${type}...`).start();
+  try {
+    const res = await fetch(`${getServer()}/api/provider/model`, {
+      method: 'PATCH',
+      headers: getHeaders(),
+      body: JSON.stringify({ type, model })
+    });
+    const data = await res.json();
+    if (!res.ok) { spinner.fail('Failed to update model'); printError(data.error || 'Unknown error'); return; }
+    spinner.succeed('Model updated');
+    printSuccess(`Model set to ${data.model} for ${data.type}`);
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+// ── changeKey — Change API key of a configured provider ────────────
+export const changeKey = async (type, newKey) => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  if (!newKey) {
+    const { key } = await inquirer.prompt({
+      type: 'password',
+      name: 'key',
+      message: `Enter new API key for ${type}:`,
+      validate: (v) => v.trim().length > 0 || 'API key required'
+    });
+    newKey = key;
+  }
+
+  const spinner = ora(`Updating API key for ${type}...`).start();
+  try {
+    const res = await fetch(`${getServer()}/api/provider/key`, {
+      method: 'PATCH',
+      headers: getHeaders(),
+      body: JSON.stringify({ type, api_key: newKey })
+    });
+    const data = await res.json();
+    if (!res.ok) { spinner.fail('Failed to update API key'); printError(data.error || 'Unknown error'); return; }
+    spinner.succeed('API key updated');
+    printSuccess(`API key updated for ${data.type} | Key: ${data.key}`);
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};
+
+// ── switchProvider — Switch to a configured provider ───────────────
+export const switchProvider = async (type) => {
+  const token = getToken();
+  if (!token) { printNotLoggedIn(); return; }
+
+  const spinner = ora(`Switching to ${type}...`).start();
+  try {
+    const res = await fetch(`${getServer()}/api/provider/switch`, {
+      method: 'POST',
+      headers: getHeaders(),
+      body: JSON.stringify({ type })
+    });
+    const data = await res.json();
+    if (!res.ok) { spinner.fail('Failed to switch provider'); printError(data.error || 'Unknown error'); return; }
+    spinner.succeed('Provider switched');
+    const modelDisplay = data.model ? ` | Model: ${data.model}` : '';
+    const keyDisplay = data.key ? ` | Key: ${data.key}` : '';
+    printSuccess(`Switched to ${data.type}${modelDisplay}${keyDisplay}`);
+  } catch (err) {
+    spinner.fail('Network error');
+    printError(err.message);
+  }
+};