fix: auxiliary switch reads provider from cfg registry

Auxiliary model switching now looks up provider details from cfg
registry (base_url, api_key) and adds custom: prefix automatically,
matching the behavior of main provider switching.

— 小糯

src/commands/models.ts

@@ -1,10 +1,12 @@
 import { readFileSync, writeFileSync, existsSync } from "fs";
+import { execSync } from "child_process";
 import { homedir } from "os";
 import { join } from "path";
 import { parse, stringify } from "yaml";
 
 const CONFIG_PATH = join(homedir(), ".hermes", "config.yaml");
 const ENV_PATH = join(homedir(), ".hermes", ".env");
+const CFG_PROVIDERS_KEY = "HERMES_CUSTOM_PROVIDERS";
 
 // ── Types ──────────────────────────────────────────────────────────────
 
@@ -72,10 +74,43 @@ function saveConfig(config: Config) {
   writeFileSync(CONFIG_PATH, stringify(config, { lineWidth: 120 }), "utf-8");
 }
 
+/**
+ * Read all available providers from cfg (central registry).
+ * Falls back to config.yaml custom_providers if cfg unavailable.
+ */
+function getCfgProviders(): CustomProvider[] {
+  try {
+    const raw = execSync(`cfg get ${CFG_PROVIDERS_KEY}`, {
+      encoding: "utf-8",
+      timeout: 5000,
+    }).trim();
+    if (!raw) return [];
+    return JSON.parse(raw) as CustomProvider[];
+  } catch {
+    return [];
+  }
+}
+
+/**
+ * Get providers from cfg (authoritative source).
+ * The config.yaml custom_providers is now only a "currently active" slot.
+ */
+function getAllProviders(): CustomProvider[] {
+  const cfgProviders = getCfgProviders();
+  if (cfgProviders.length > 0) return cfgProviders;
+  // Fallback to config.yaml if cfg has nothing
+  const config = loadConfig();
+  return config.custom_providers || [];
+}
+
 function getCustomProviders(config: Config): CustomProvider[] {
   return config.custom_providers || [];
 }
 
+function findProviderFromRegistry(name: string): CustomProvider | undefined {
+  return getAllProviders().find((p) => p.name === name);
+}
+
 function findProvider(
   config: Config,
   name: string
@@ -123,42 +158,41 @@ function getAuthHeaders(provider: CustomProvider): Record<string, string> {
 
 function providers() {
   const config = loadConfig();
-  const customProviders = getCustomProviders(config);
+  const allProviders = getAllProviders();
   const mainProvider = config.model?.provider || "auto";
   const mainModel = config.model?.default || "(not set)";
 
   console.log(`Main: provider=${mainProvider}  model=${mainModel}\n`);
 
-  if (customProviders.length === 0) {
+  if (allProviders.length === 0) {
     console.log("No custom providers configured.");
     return;
   }
 
-  console.log("Custom providers:\n");
+  console.log("Custom providers (from cfg):\n");
-  for (const p of customProviders) {
+  for (const p of allProviders) {
     const active = p.name === mainProvider || `custom:${p.name}` === mainProvider ? " ◀ active" : "";
     console.log(`  ${p.name}${active}`);
     console.log(`    url: ${p.base_url}`);
     if (p.api_mode) console.log(`    mode: ${p.api_mode}`);
   }
-  console.log(`\nTotal: ${customProviders.length}`);
+  console.log(`\nTotal: ${allProviders.length}`);
 }
 
 // ── list ────────────────────────────────────────────────────────────────
 
 async function list(providerName?: string) {
-  const config = loadConfig();
+  const allProviders = getAllProviders();
-  const customProviders = getCustomProviders(config);
 
   const targets: CustomProvider[] = providerName
-    ? customProviders.filter((p) => p.name === providerName)
+    ? allProviders.filter((p) => p.name === providerName)
-    : customProviders;
+    : allProviders;
 
   if (targets.length === 0) {
     if (providerName) {
       console.error(`Provider not found: ${providerName}`);
       console.error(
-        `Available: ${customProviders.map((p) => p.name).join(", ")}`
+        `Available: ${allProviders.map((p) => p.name).join(", ")}`
       );
     } else {
       console.log("No custom providers configured.");
@@ -220,12 +254,11 @@ async function list(providerName?: string) {
 // ── test ────────────────────────────────────────────────────────────────
 
 async function test(providerName?: string) {
-  const config = loadConfig();
+  const allProviders = getAllProviders();
-  const customProviders = getCustomProviders(config);
 
   const targets: CustomProvider[] = providerName
-    ? customProviders.filter((p) => p.name === providerName)
+    ? allProviders.filter((p) => p.name === providerName)
-    : customProviders;
+    : allProviders;
 
   if (targets.length === 0) {
     if (providerName) {
@@ -311,32 +344,74 @@ function switchConfig(opts: SwitchOptions) {
     // Switch auxiliary tasks
     if (!config.auxiliary) config.auxiliary = {} as Record<string, AuxiliaryEntry>;
 
+    // Look up provider from cfg registry to get base_url & api_key
+    const registryProvider = findProviderFromRegistry(opts.provider);
+    const providerValue = registryProvider ? `custom:${registryProvider.name}` : opts.provider;
+
+    // Ensure custom_providers in config.yaml contains this provider
+    if (registryProvider) {
+      if (!config.custom_providers) config.custom_providers = [];
+      const existing = config.custom_providers.find((p) => p.name === registryProvider.name);
+      if (!existing) {
+        config.custom_providers.push(registryProvider);
+      }
+    }
+
     for (const task of opts.auxTasks) {
       if (!config.auxiliary[task]) {
         config.auxiliary[task] = { provider: "", model: "", timeout: 30 };
       }
-      config.auxiliary[task].provider = opts.provider;
+      config.auxiliary[task].provider = providerValue;
+      if (registryProvider) {
+        config.auxiliary[task].base_url = registryProvider.base_url;
+        const apiKey = getProviderApiKey(registryProvider);
+        if (apiKey) config.auxiliary[task].api_key = apiKey;
+      }
       if (opts.model) {
         config.auxiliary[task].model = opts.model;
       }
-      console.log(`  auxiliary.${task} → provider=${opts.provider}${opts.model ? ` model=${opts.model}` : ""}`);
+      console.log(`  auxiliary.${task} → provider=${providerValue}${opts.model ? ` model=${opts.model}` : ""}`);
     }
   } else {
-    // Switch main provider
+    // Switch main provider — look up from cfg registry
-    const providerValue = findProvider(config, opts.provider)
+    const registryProvider = findProviderFromRegistry(opts.provider);
-      ? `custom:${opts.provider}`
-      : opts.provider;
     
+    if (registryProvider) {
+      // Write this provider into config.yaml custom_providers (replace all)
+      config.custom_providers = [registryProvider];
+      const providerValue = `custom:${registryProvider.name}`;
       if (!config.model) config.model = {};
       config.model.provider = providerValue;
       if (opts.model) {
         config.model.default = opts.model;
       }
       console.log(`  main → provider=${providerValue}${opts.model ? ` model=${opts.model}` : ""}`);
+      console.log(`  config.yaml custom_providers updated with ${registryProvider.name}`);
+    } else {
+      // Standard provider (openrouter, anthropic, etc.)
+      if (!config.model) config.model = {};
+      config.model.provider = opts.provider;
+      if (opts.model) {
+        config.model.default = opts.model;
+      }
+      // Clear custom_providers since switching to standard
+      delete config.custom_providers;
+      console.log(`  main → provider=${opts.provider}${opts.model ? ` model=${opts.model}` : ""}`);
+    }
   }
 
   saveConfig(config);
-  console.log("\nConfig saved. Run 'hermes gateway restart' or start a new session to apply.");
+
+  // Restart gateway
+  try {
+    execSync("systemctl --user restart hermes-gateway 2>/dev/null || true", {
+      encoding: "utf-8",
+      timeout: 10000,
+    });
+    console.log("\nConfig saved & gateway restarted.");
+  } catch {
+    console.log("\nConfig saved. Restart gateway manually to apply.");
+  }
 }
 
 async function switchTelegram(opts: SwitchOptions) {