yymaxapi 1.0.54 → 1.0.56

package/bin/yymaxapi.js

@@ -3397,50 +3397,76 @@ async function switchModel(paths) {
     return;
   }
 
-  // 已知 provider 别名映射
   const PROVIDER_LABELS = {
     'claude-yunyi': '云翼 Claude (包月)',
     'yunyi': '云翼 Codex (包月)',
     'heibai': 'MAXAPI (按量)',
   };
 
-  // 收集所有 provider 的所有模型
+  const claudeProviderName = API_CONFIG.claude.providerName;
+  const codexProviderName = API_CONFIG.codex.providerName;
+
+  // 预设模型列表
   const choices = [];
-  for (const [providerName, providerConfig] of Object.entries(providers)) {
-    const label = PROVIDER_LABELS[providerName] || providerName;
-    const models = providerConfig.models || [];
-    if (models.length === 0) continue;
+  const presetKeys = new Set();
 
-    choices.push(new inquirer.Separator(`  -- ${label} --`));
-    for (const m of models) {
-      const modelKey = `${providerName}/${m.id}`;
+  if (CLAUDE_MODELS.length > 0) {
+    choices.push(new inquirer.Separator('  -- Claude --'));
+    for (const m of CLAUDE_MODELS) {
+      const pName = providers[claudeProviderName] ? claudeProviderName : Object.keys(providers)[0];
+      const modelKey = `${pName}/${m.id}`;
       const isCurrent = modelKey === primary;
       choices.push({
-        name: isCurrent ? `${m.name || m.id} (当前)` : (m.name || m.id),
+        name: isCurrent ? `${m.name} (当前)` : m.name,
         value: modelKey,
       });
+      presetKeys.add(modelKey);
     }
   }
 
-  // 也把 agents.defaults.models 里注册过但 provider.models 里没列出的补上
+  if (CODEX_MODELS.length > 0) {
+    choices.push(new inquirer.Separator('  -- GPT --'));
+    for (const m of CODEX_MODELS) {
+      const pName = providers[codexProviderName] ? codexProviderName : Object.keys(providers)[0];
+      const modelKey = `${pName}/${m.id}`;
+      const isCurrent = modelKey === primary;
+      choices.push({
+        name: isCurrent ? `${m.name} (当前)` : m.name,
+        value: modelKey,
+      });
+      presetKeys.add(modelKey);
+    }
+  }
+
+  // 收集用户配置中的其他模型（不在预设列表里的）
+  const otherModels = [];
+  for (const [providerName, providerConfig] of Object.entries(providers)) {
+    for (const m of (providerConfig.models || [])) {
+      const modelKey = `${providerName}/${m.id}`;
+      if (!presetKeys.has(modelKey)) {
+        otherModels.push({ modelKey, name: m.name || m.id, providerName });
+      }
+    }
+  }
   const registeredKeys = Object.keys(config.agents?.defaults?.models || {});
-  const listedKeys = new Set(choices.filter(c => c.value).map(c => c.value));
   for (const modelKey of registeredKeys) {
-    if (listedKeys.has(modelKey)) continue;
+    if (presetKeys.has(modelKey) || otherModels.some(o => o.modelKey === modelKey)) continue;
     const [pName, mId] = modelKey.split('/');
     if (!pName || !mId) continue;
-    const label = PROVIDER_LABELS[pName] || pName;
-    const isCurrent = modelKey === primary;
-    choices.push({
-      name: isCurrent ? `${label} / ${mId} (当前)` : `${label} / ${mId}`,
-      value: modelKey,
-    });
+    otherModels.push({ modelKey, name: `${PROVIDER_LABELS[pName] || pName} / ${mId}`, providerName: pName });
   }
 
-  if (choices.filter(c => c.value).length === 0) {
-    console.log(chalk.yellow('⚠️ 未找到任何可用模型'));
-    return;
-  }
+  // 当前主模型不在预设列表中时，标注在"其他"入口上
+  const primaryInPreset = presetKeys.has(primary);
+  const otherLabel = otherModels.length > 0
+    ? `切换其他模型 (${otherModels.length} 个)`
+    : '切换其他模型';
+
+  choices.push(new inquirer.Separator(''));
+  choices.push({
+    name: !primaryInPreset && primary ? `${otherLabel} ← 当前在此` : otherLabel,
+    value: '__other__',
+  });
 
   const currentProvider = primary.split('/')[0] || '';
   const currentProviderConfig = providers[currentProvider];
@@ -3454,25 +3480,51 @@ async function switchModel(paths) {
     type: 'list',
     name: 'selected',
     message: '选择模型:',
-    default: primary,
+    default: primaryInPreset ? primary : undefined,
     pageSize: 15,
     choices,
   }]);
 
-  if (selected === primary) {
+  let finalSelected = selected;
+
+  if (selected === '__other__') {
+    if (otherModels.length === 0) {
+      console.log(chalk.gray('\n当前配置中没有其他模型，仅有预设模型可用'));
+      return;
+    }
+    const otherChoices = otherModels.map(o => {
+      const label = PROVIDER_LABELS[o.providerName] || o.providerName;
+      const isCurrent = o.modelKey === primary;
+      return {
+        name: isCurrent ? `${o.name} [${label}] (当前)` : `${o.name} [${label}]`,
+        value: o.modelKey,
+      };
+    });
+    const { pickedOther } = await inquirer.prompt([{
+      type: 'list',
+      name: 'pickedOther',
+      message: '选择其他模型:',
+      default: !primaryInPreset ? primary : undefined,
+      pageSize: 15,
+      choices: otherChoices,
+    }]);
+    finalSelected = pickedOther;
+  }
+
+  if (finalSelected === primary) {
     console.log(chalk.gray('\n模型未变更'));
     return;
   }
 
-  const selectedProvider = selected.split('/')[0];
-  const selectedModelId = selected.split('/')[1];
-  config.agents.defaults.model.primary = selected;
-  if (!config.agents.defaults.models[selected]) {
-    config.agents.defaults.models[selected] = { alias: selectedProvider };
+  const selectedProviderKey = finalSelected.split('/')[0];
+  const selectedModelId = finalSelected.split('/')[1];
+  config.agents.defaults.model.primary = finalSelected;
+  if (!config.agents.defaults.models[finalSelected]) {
+    config.agents.defaults.models[finalSelected] = { alias: selectedProviderKey };
   }
 
-  // 同步更新 provider.models，确保 primary 指向的模型在列表中
-  const providerConfig = providers[selectedProvider];
+  // 同步更新 provider.models
+  const providerConfig = providers[selectedProviderKey];
   if (providerConfig && selectedModelId) {
     const existingModel = (providerConfig.models || []).find(m => m.id === selectedModelId);
     if (!existingModel) {
@@ -3493,16 +3545,16 @@ async function switchModel(paths) {
     }
   }
 
-  config.agents.defaults.model.fallbacks = (config.agents.defaults.model.fallbacks || []).filter(k => k !== selected);
-  if (primary && primary !== selected && !config.agents.defaults.model.fallbacks.includes(primary)) {
+  config.agents.defaults.model.fallbacks = (config.agents.defaults.model.fallbacks || []).filter(k => k !== finalSelected);
+  if (primary && primary !== finalSelected && !config.agents.defaults.model.fallbacks.includes(primary)) {
     config.agents.defaults.model.fallbacks.push(primary);
   }
   ensureGatewaySettings(config);
   writeConfigWithSync(paths, config);
 
-  const selectedProviderConfig = providers[selectedProvider];
-  const selectedLabel = PROVIDER_LABELS[selectedProvider] || selectedProvider;
-  console.log(chalk.green(`\n✅ 已切换到 ${selected}`));
+  const selectedProviderConfig = providers[selectedProviderKey];
+  const selectedLabel = PROVIDER_LABELS[selectedProviderKey] || selectedProviderKey;
+  console.log(chalk.green(`\n✅ 已切换到 ${finalSelected}`));
   console.log(chalk.gray(`   来源: ${selectedLabel}`));
   if (selectedProviderConfig) {
     console.log(chalk.gray(`   节点: ${selectedProviderConfig.baseUrl}`));
@@ -3511,7 +3563,6 @@ async function switchModel(paths) {
   const gwPort = config.gateway?.port || 18789;
   const gwToken = config.gateway?.auth?.token;
 
-  // 自动重启 Gateway 使切换立即生效
   if (await isPortOpen(gwPort)) {
     const gwSpinner = ora({ text: '重启 Gateway 使模型切换生效...', spinner: 'dots' }).start();
     const ok = await restartGateway({ silent: true });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yymaxapi",
-  "version": "1.0.54",
+  "version": "1.0.56",
   "description": "跨平台 OpenClaw/Clawdbot 配置管理工具 - 管理中转地址、模型切换、API Keys、测速优化",
   "main": "bin/yymaxapi.js",
   "bin": {