@shi_zhen/code-helper 0.1.5 → 0.1.6

package/dist/cli/index.mjs

@@ -36,7 +36,7 @@ var require_package = __commonJS({
   "package.json"(exports, module) {
     module.exports = {
       name: "@shi_zhen/code-helper",
-      version: "0.1.5",
+      version: "0.1.6",
       description: "Code Helper - A unified tool to manage your Claude Code",
       repository: {
         type: "git",
@@ -694,6 +694,15 @@ function switchProfile(id) {
   if (profile.timeout) {
     settings.env.API_TIMEOUT_MS = profile.timeout;
   }
+  if (profile.haikuModel) {
+    settings.env.ANTHROPIC_DEFAULT_HAIKU_MODEL = profile.haikuModel;
+  }
+  if (profile.sonnetModel) {
+    settings.env.ANTHROPIC_DEFAULT_SONNET_MODEL = profile.sonnetModel;
+  }
+  if (profile.opusModel) {
+    settings.env.ANTHROPIC_DEFAULT_OPUS_MODEL = profile.opusModel;
+  }
   writeClaudeSettings(settings);
   config.currentProfileId = id;
   writeProfiles(config);
@@ -714,14 +723,14 @@ import ora from "ora";
 // src/cli/utils/api.ts
 import http from "http";
 import https from "https";
-async function verifyApiKey(baseUrl, apiKey) {
+async function verifyApiKey(baseUrl, apiKey, model) {
   return new Promise((resolve) => {
     try {
       const url = new URL(baseUrl);
       const isHttps = url.protocol === "https:";
       const client = isHttps ? https : http;
       const testPayload = JSON.stringify({
-        model: "claude-3-5-sonnet-20241022",
+        model: model || "claude-3-5-sonnet-20241022",
         max_tokens: 1,
         messages: [{ role: "user", content: "test" }]
       });
@@ -811,6 +820,15 @@ async function showAuthMenu() {
       console.log(
         `${theme.dim("  API Key:")} ${maskApiKey(currentProfile.apiKey)}`
       );
+      if (currentProfile.haikuModel) {
+        console.log(`${theme.dim("  Haiku Model:")} ${currentProfile.haikuModel}`);
+      }
+      if (currentProfile.sonnetModel) {
+        console.log(`${theme.dim("  Sonnet Model:")} ${currentProfile.sonnetModel}`);
+      }
+      if (currentProfile.opusModel) {
+        console.log(`${theme.dim("  Opus Model:")} ${currentProfile.opusModel}`);
+      }
       console.log();
     }
     const savedProfiles = getAllProfiles();
@@ -1114,9 +1132,49 @@ async function handleManualAuth() {
       }
     }
   ]);
+  let haikuModel;
+  let sonnetModel;
+  let opusModel;
+  const { configModels } = await inquirer.prompt([
+    {
+      type: "confirm",
+      name: "configModels",
+      message: "\u662F\u5426\u9700\u8981\u914D\u7F6E\u6A21\u578B\u6620\u5C04\uFF1F\uFF08\u7528\u4E8E ZenMux \u7B49\u9700\u8981\u81EA\u5B9A\u4E49\u6A21\u578B\u7684\u5E73\u53F0\uFF09",
+      default: false
+    }
+  ]);
+  if (configModels) {
+    console.log();
+    console.log(theme.dim("\u8BF7\u8F93\u5165\u5404\u6A21\u578B\u7684\u6620\u5C04\u540D\u79F0\uFF08\u76F4\u63A5\u56DE\u8F66\u8DF3\u8FC7\uFF09\uFF1A"));
+    console.log();
+    const modelAnswers = await inquirer.prompt([
+      {
+        type: "input",
+        name: "haikuModel",
+        message: "ANTHROPIC_DEFAULT_HAIKU_MODEL\uFF08\u5FEB\u901F\u6A21\u578B\uFF09:",
+        default: ""
+      },
+      {
+        type: "input",
+        name: "sonnetModel",
+        message: "ANTHROPIC_DEFAULT_SONNET_MODEL\uFF08\u5747\u8861\u6A21\u578B\uFF09:",
+        default: ""
+      },
+      {
+        type: "input",
+        name: "opusModel",
+        message: "ANTHROPIC_DEFAULT_OPUS_MODEL\uFF08\u6700\u5F3A\u6A21\u578B\uFF09:",
+        default: ""
+      }
+    ]);
+    if (modelAnswers.haikuModel?.trim()) haikuModel = modelAnswers.haikuModel.trim();
+    if (modelAnswers.sonnetModel?.trim()) sonnetModel = modelAnswers.sonnetModel.trim();
+    if (modelAnswers.opusModel?.trim()) opusModel = modelAnswers.opusModel.trim();
+  }
+  const verifyModel = sonnetModel || haikuModel || opusModel;
   console.log();
   const spinner = ora("\u6B63\u5728\u9A8C\u8BC1 API Key...").start();
-  const verification = await verifyApiKey(baseUrl.trim(), apiKey.trim());
+  const verification = await verifyApiKey(baseUrl.trim(), apiKey.trim(), verifyModel);
   if (!verification.valid) {
     spinner.fail("API Key \u9A8C\u8BC1\u5931\u8D25");
     console.log();
@@ -1130,6 +1188,9 @@ async function handleManualAuth() {
     if (timeout && timeout.trim()) {
       options.timeout = timeout.trim();
     }
+    if (haikuModel) options.haikuModel = haikuModel;
+    if (sonnetModel) options.sonnetModel = sonnetModel;
+    if (opusModel) options.opusModel = opusModel;
     await saveApiConfig(baseUrl.trim(), apiKey.trim(), options);
     console.log();
     showSuccess("\u914D\u7F6E\u5DF2\u751F\u6548");
@@ -1139,7 +1200,10 @@ async function handleManualAuth() {
       baseUrl.trim(),
       apiKey.trim(),
       void 0,
-      timeout && timeout.trim() ? timeout.trim() : void 0
+      timeout && timeout.trim() ? timeout.trim() : void 0,
+      haikuModel,
+      sonnetModel,
+      opusModel
     );
   } catch (error) {
     console.log();
@@ -1147,7 +1211,7 @@ async function handleManualAuth() {
     await waitForEnter();
   }
 }
-async function askToSaveProfile(defaultName, baseUrl, apiKey, model, timeout) {
+async function askToSaveProfile(defaultName, baseUrl, apiKey, model, timeout, haikuModel, sonnetModel, opusModel) {
   console.log();
   const { save } = await inquirer.prompt([
     {
@@ -1176,7 +1240,10 @@ async function askToSaveProfile(defaultName, baseUrl, apiKey, model, timeout) {
       baseUrl,
       apiKey,
       model,
-      timeout
+      timeout,
+      haikuModel,
+      sonnetModel,
+      opusModel
     });
     switchProfile(newProfile.id);
     showSuccess(`\u914D\u7F6E "${name.trim()}" \u5DF2\u4FDD\u5B58`);