shenxiang-ai-cli 0.1.0 → 0.2.1

package/dist/index.js

@@ -4,7 +4,7 @@
 import { Command } from "commander";
 
 // src/commands/chat.ts
-import readline from "readline";
+import inquirer from "inquirer";
 
 // src/core/providers/openai-compatible.ts
 import OpenAI from "openai";
@@ -21,7 +21,9 @@ var OpenAICompatibleProvider = class {
     }
     const client = new OpenAI({
       apiKey: config3.apiKey,
-      baseURL: config3.baseUrl
+      baseURL: config3.baseUrl,
+      timeout: 12e4
+      // 2min timeout
     });
     const openaiTools = tools.map((t2) => ({
       type: "function",
@@ -70,6 +72,7 @@ var OpenAICompatibleProvider = class {
       });
       const toolCallAccumulator = /* @__PURE__ */ new Map();
       let tokenUsage;
+      let toolCallStartEmitted = false;
       for await (const chunk of stream) {
         if (chunk.usage) {
           tokenUsage = {
@@ -85,6 +88,10 @@ var OpenAICompatibleProvider = class {
           yield { type: "text", content: delta.content };
         }
         if (delta.tool_calls) {
+          if (!toolCallStartEmitted) {
+            toolCallStartEmitted = true;
+            yield { type: "tool_call_start" };
+          }
           for (const tc of delta.tool_calls) {
             const idx = tc.index;
             if (!toolCallAccumulator.has(idx)) {
@@ -176,14 +183,18 @@ var OpenAICompatibleProvider = class {
       body.tools = kimiTools;
     }
     try {
+      const controller = new AbortController();
+      const connectTimeout = setTimeout(() => controller.abort(), 3e4);
       const response = await fetch(`${baseUrl}/chat/completions`, {
         method: "POST",
         headers: {
           "Content-Type": "application/json",
           "Authorization": `Bearer ${config3.apiKey}`
         },
-        body: JSON.stringify(body)
+        body: JSON.stringify(body),
+        signal: controller.signal
       });
+      clearTimeout(connectTimeout);
       if (!response.ok) {
         const errText = await response.text();
         yield { type: "error", error: `${response.status} ${errText}` };
@@ -199,71 +210,83 @@ var OpenAICompatibleProvider = class {
       const toolCallAccumulator = /* @__PURE__ */ new Map();
       let tokenUsage;
       let reasoningContent = "";
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done) break;
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split("\n");
-        buffer = lines.pop() || "";
-        for (const line of lines) {
-          const trimmed = line.trim();
-          if (!trimmed.startsWith("data: ")) continue;
-          const data = trimmed.slice(6);
-          if (data === "[DONE]") continue;
-          let parsed;
-          try {
-            parsed = JSON.parse(data);
-          } catch {
-            continue;
-          }
-          if (parsed.usage) {
-            tokenUsage = {
-              promptTokens: parsed.usage.prompt_tokens || 0,
-              completionTokens: parsed.usage.completion_tokens || 0,
-              totalTokens: parsed.usage.total_tokens || 0
-            };
-          }
-          const choice = parsed.choices?.[0];
-          if (!choice) continue;
-          const delta = choice.delta || {};
-          if (delta.reasoning_content) {
-            reasoningContent += delta.reasoning_content;
-          }
-          if (delta.content) {
-            yield { type: "text", content: delta.content };
-          }
-          if (delta.tool_calls) {
-            for (const tc of delta.tool_calls) {
-              const idx = tc.index ?? 0;
-              if (!toolCallAccumulator.has(idx)) {
-                toolCallAccumulator.set(idx, { id: tc.id || "", functionName: tc.function?.name || "", arguments: "" });
-              }
-              const acc = toolCallAccumulator.get(idx);
-              if (tc.id) acc.id = tc.id;
-              if (tc.function?.name) acc.functionName = tc.function.name;
-              if (tc.function?.arguments) acc.arguments += tc.function.arguments;
+      let finished = false;
+      let toolCallStartEmitted = false;
+      const READ_TIMEOUT_MS = 12e4;
+      try {
+        while (true) {
+          const readPromise = reader.read();
+          const timeoutPromise = new Promise(
+            (resolve) => setTimeout(() => resolve({ done: true, value: void 0 }), READ_TIMEOUT_MS)
+          );
+          const { done, value } = await Promise.race([readPromise, timeoutPromise]);
+          if (done) break;
+          buffer += decoder.decode(value, { stream: true });
+          const lines = buffer.split("\n");
+          buffer = lines.pop() || "";
+          for (const line of lines) {
+            const trimmed = line.trim();
+            if (!trimmed.startsWith("data: ")) continue;
+            const data = trimmed.slice(6);
+            if (data === "[DONE]") continue;
+            let parsed;
+            try {
+              parsed = JSON.parse(data);
+            } catch {
+              continue;
             }
-          }
-          if (choice.finish_reason === "tool_calls" || choice.finish_reason === "stop") {
-            for (const [_, acc] of toolCallAccumulator) {
-              const toolCall = {
-                id: acc.id,
-                type: "function",
-                function: { name: acc.functionName, arguments: acc.arguments }
+            if (parsed.usage) {
+              tokenUsage = {
+                promptTokens: parsed.usage.prompt_tokens || 0,
+                completionTokens: parsed.usage.completion_tokens || 0,
+                totalTokens: parsed.usage.total_tokens || 0
               };
-              yield { type: "tool_call", toolCall };
             }
-            yield {
-              type: "done",
-              usage: tokenUsage,
-              content: reasoningContent || void 0
-            };
-            return;
+            const choice = parsed.choices?.[0];
+            if (!choice) continue;
+            const delta = choice.delta || {};
+            if (delta.reasoning_content) {
+              reasoningContent += delta.reasoning_content;
+            }
+            if (delta.content) {
+              yield { type: "text", content: delta.content };
+            }
+            if (delta.tool_calls) {
+              if (!toolCallStartEmitted) {
+                toolCallStartEmitted = true;
+                yield { type: "tool_call_start" };
+              }
+              for (const tc of delta.tool_calls) {
+                const idx = tc.index ?? 0;
+                if (!toolCallAccumulator.has(idx)) {
+                  toolCallAccumulator.set(idx, { id: tc.id || "", functionName: tc.function?.name || "", arguments: "" });
+                }
+                const acc = toolCallAccumulator.get(idx);
+                if (tc.id) acc.id = tc.id;
+                if (tc.function?.name) acc.functionName = tc.function.name;
+                if (tc.function?.arguments) acc.arguments += tc.function.arguments;
+              }
+            }
+            if (choice.finish_reason === "tool_calls" || choice.finish_reason === "stop") {
+              finished = true;
+              break;
+            }
           }
+          if (finished) break;
+        }
+      } finally {
+        try {
+          reader.cancel();
+        } catch {
         }
       }
       for (const [_, acc] of toolCallAccumulator) {
-        yield { type: "tool_call", toolCall: { id: acc.id, type: "function", function: { name: acc.functionName, arguments: acc.arguments } } };
+        const toolCall = {
+          id: acc.id,
+          type: "function",
+          function: { name: acc.functionName, arguments: acc.arguments }
+        };
+        yield { type: "tool_call", toolCall };
       }
       yield { type: "done", usage: tokenUsage, content: reasoningContent || void 0 };
     } catch (err) {
@@ -403,22 +426,49 @@ var AnthropicProvider = class {
 
 // src/core/providers/types.ts
 var MODEL_PROVIDERS = {
-  // Kimi (月之暗面 Moonshot)
-  "kimi-k2.5": { provider: "kimi", displayName: "Kimi K2.5 (\u6700\u65B0\u591A\u6A21\u6001)" },
-  "kimi-k2-0905-preview": { provider: "kimi", displayName: "Kimi K2" },
-  "moonshot-v1-auto": { provider: "kimi", displayName: "Moonshot v1 Auto" },
-  // OpenAI
-  "gpt-4o": { provider: "openai", displayName: "GPT-4o" },
-  "gpt-4o-mini": { provider: "openai", displayName: "GPT-4o Mini" },
-  // Anthropic
-  "claude-sonnet-4-20250514": { provider: "anthropic", displayName: "Claude Sonnet 4" },
-  "claude-3-5-sonnet-20241022": { provider: "anthropic", displayName: "Claude 3.5 Sonnet" },
-  // DeepSeek
-  "deepseek-chat": { provider: "deepseek", displayName: "DeepSeek Chat" },
-  "deepseek-coder": { provider: "deepseek", displayName: "DeepSeek Coder" },
-  // Qwen (通义千问)
-  "qwen-turbo": { provider: "qwen", displayName: "\u901A\u4E49\u5343\u95EE Turbo" },
-  "qwen-plus": { provider: "qwen", displayName: "\u901A\u4E49\u5343\u95EE Plus" }
+  // ── Kimi (月之暗面 Moonshot) ──────────────────
+  "kimi-k2.5": { provider: "kimi", displayName: "Kimi K2.5", group: "Kimi \u6708\u4E4B\u6697\u9762" },
+  "kimi-k2-0905-preview": { provider: "kimi", displayName: "Kimi K2", group: "Kimi \u6708\u4E4B\u6697\u9762" },
+  "moonshot-v1-auto": { provider: "kimi", displayName: "Moonshot v1 Auto", group: "Kimi \u6708\u4E4B\u6697\u9762" },
+  // ── OpenAI ────────────────────────────────────
+  "gpt-4o": { provider: "openai", displayName: "GPT-4o", group: "OpenAI" },
+  "gpt-4o-mini": { provider: "openai", displayName: "GPT-4o Mini", group: "OpenAI" },
+  "gpt-4.1": { provider: "openai", displayName: "GPT-4.1", group: "OpenAI" },
+  "gpt-4.1-mini": { provider: "openai", displayName: "GPT-4.1 Mini", group: "OpenAI" },
+  "o3-mini": { provider: "openai", displayName: "o3-mini (\u63A8\u7406)", group: "OpenAI" },
+  // ── Anthropic Claude ──────────────────────────
+  "claude-sonnet-4-20250514": { provider: "anthropic", displayName: "Claude Sonnet 4", group: "Anthropic Claude" },
+  "claude-3-5-sonnet-20241022": { provider: "anthropic", displayName: "Claude 3.5 Sonnet", group: "Anthropic Claude" },
+  "claude-3-5-haiku-20241022": { provider: "anthropic", displayName: "Claude 3.5 Haiku", group: "Anthropic Claude" },
+  // ── DeepSeek ──────────────────────────────────
+  "deepseek-chat": { provider: "deepseek", displayName: "DeepSeek V3", group: "DeepSeek \u6DF1\u5EA6\u6C42\u7D22" },
+  "deepseek-reasoner": { provider: "deepseek", displayName: "DeepSeek R1 (\u63A8\u7406)", group: "DeepSeek \u6DF1\u5EA6\u6C42\u7D22" },
+  // ── 通义千问 Qwen ────────────────────────────
+  "qwen-max": { provider: "qwen", displayName: "\u901A\u4E49\u5343\u95EE Max", group: "\u901A\u4E49\u5343\u95EE Qwen" },
+  "qwen-plus": { provider: "qwen", displayName: "\u901A\u4E49\u5343\u95EE Plus", group: "\u901A\u4E49\u5343\u95EE Qwen" },
+  "qwen-turbo": { provider: "qwen", displayName: "\u901A\u4E49\u5343\u95EE Turbo", group: "\u901A\u4E49\u5343\u95EE Qwen" },
+  "qwen-coder-plus": { provider: "qwen", displayName: "\u901A\u4E49\u5343\u95EE Coder Plus", group: "\u901A\u4E49\u5343\u95EE Qwen" },
+  // ── 智谱 GLM ──────────────────────────────────
+  "glm-4-plus": { provider: "zhipu", displayName: "GLM-4 Plus", group: "\u667A\u8C31 GLM" },
+  "glm-4-flash": { provider: "zhipu", displayName: "GLM-4 Flash (\u514D\u8D39)", group: "\u667A\u8C31 GLM" },
+  "glm-4-long": { provider: "zhipu", displayName: "GLM-4 Long (\u957F\u6587\u672C)", group: "\u667A\u8C31 GLM" },
+  "codegeex-4": { provider: "zhipu", displayName: "CodeGeeX 4 (\u7F16\u7A0B)", group: "\u667A\u8C31 GLM" },
+  // ── 豆包 Doubao (字节跳动) ────────────────────
+  "doubao-1.5-pro-256k": { provider: "doubao", displayName: "\u8C46\u5305 1.5 Pro 256K", group: "\u8C46\u5305 Doubao" },
+  "doubao-1.5-lite-32k": { provider: "doubao", displayName: "\u8C46\u5305 1.5 Lite 32K", group: "\u8C46\u5305 Doubao" },
+  // ── 百川 Baichuan ────────────────────────────
+  "Baichuan4-Turbo": { provider: "baichuan", displayName: "\u767E\u5DDD4 Turbo", group: "\u767E\u5DDD Baichuan" },
+  // ── 零一万物 Yi ───────────────────────────────
+  "yi-lightning": { provider: "lingyiwanwu", displayName: "Yi Lightning", group: "\u96F6\u4E00\u4E07\u7269 Yi" },
+  // ── Gemini (Google) ───────────────────────────
+  "gemini-2.5-flash": { provider: "gemini", displayName: "Gemini 2.5 Flash", group: "Google Gemini" },
+  "gemini-2.5-pro": { provider: "gemini", displayName: "Gemini 2.5 Pro", group: "Google Gemini" },
+  // ── Mistral ───────────────────────────────────
+  "mistral-large-latest": { provider: "mistral", displayName: "Mistral Large", group: "Mistral" },
+  "codestral-latest": { provider: "mistral", displayName: "Codestral (\u7F16\u7A0B)", group: "Mistral" },
+  // ── Groq (超快推理) ──────────────────────────
+  "llama-3.3-70b-versatile": { provider: "groq", displayName: "Llama 3.3 70B", group: "Groq" },
+  "llama-3.1-8b-instant": { provider: "groq", displayName: "Llama 3.1 8B (\u6781\u901F)", group: "Groq" }
 };
 
 // src/core/providers/index.ts
@@ -427,6 +477,13 @@ var providers = {
   deepseek: new OpenAICompatibleProvider("deepseek"),
   qwen: new OpenAICompatibleProvider("qwen"),
   kimi: new OpenAICompatibleProvider("kimi"),
+  zhipu: new OpenAICompatibleProvider("zhipu"),
+  doubao: new OpenAICompatibleProvider("doubao"),
+  baichuan: new OpenAICompatibleProvider("baichuan"),
+  lingyiwanwu: new OpenAICompatibleProvider("lingyiwanwu"),
+  gemini: new OpenAICompatibleProvider("gemini"),
+  mistral: new OpenAICompatibleProvider("mistral"),
+  groq: new OpenAICompatibleProvider("groq"),
   anthropic: new AnthropicProvider()
 };
 var DEFAULT_BASE_URLS = {
@@ -434,7 +491,14 @@ var DEFAULT_BASE_URLS = {
   deepseek: "https://api.deepseek.com",
   qwen: "https://dashscope.aliyuncs.com/compatible-mode/v1",
   kimi: "https://api.moonshot.cn/v1",
-  anthropic: "https://api.anthropic.com"
+  anthropic: "https://api.anthropic.com",
+  zhipu: "https://open.bigmodel.cn/api/paas/v4",
+  doubao: "https://ark.cn-beijing.volces.com/api/v3",
+  baichuan: "https://api.baichuan-ai.com/v1",
+  lingyiwanwu: "https://api.lingyiwanwu.com/v1",
+  gemini: "https://generativelanguage.googleapis.com/v1beta/openai",
+  mistral: "https://api.mistral.ai/v1",
+  groq: "https://api.groq.com/openai/v1"
 };
 function getProvider(model) {
   const info = MODEL_PROVIDERS[model];
@@ -453,7 +517,8 @@ function listModels() {
   return Object.entries(MODEL_PROVIDERS).map(([id, info]) => ({
     id,
     name: info.displayName,
-    provider: info.provider
+    provider: info.provider,
+    group: info.group
   }));
 }
 
@@ -1090,22 +1155,25 @@ var theme = {
   tokenValue: chalk.hex("#A78BFA")
 };
 function banner() {
-  const line = theme.brandDim("\u2501".repeat(46));
-  const border = theme.brandDim("\u2503");
-  const topLeft = theme.brandDim("\u250F");
-  const topRight = theme.brandDim("\u2513");
-  const botLeft = theme.brandDim("\u2517");
-  const botRight = theme.brandDim("\u251B");
-  return `
-${topLeft}${line}${topRight}
-${border}                                              ${border}
-${border}    ${theme.brandBold("\u6C88\u7FD4\u7684AI\u52A9\u624B")}  ${theme.muted("v0.1.0")}                   ${border}
-${border}    ${theme.dim("\u7EC8\u7AEF\u91CC\u7684AI\u5168\u6808\u5F00\u53D1\u642D\u6863")}               ${border}
-${border}                                              ${border}
-${botLeft}${line}${botRight}`;
+  const p = theme.brand;
+  const b = theme.brandBold;
+  const d = theme.dim;
+  const m = theme.muted;
+  const y = chalk.hex("#FBBF24");
+  const r = chalk.hex("#F87171");
+  const w = chalk.white.bold;
+  const lines = [
+    ``,
+    `    ${p("/\\_/\\")}         ${b("\u6C88\u7FD4\u7684AI\u52A9\u624B")}  ${m("v0.2.1")}`,
+    `   ${p("(")} ${y("o")}${p(".")}${y("o")} ${p(")")}        ${d("\u7EC8\u7AEF\u91CC\u7684AI\u5168\u6808\u5F00\u53D1\u642D\u6863")}`,
+    `    ${p("> ")}${r("^")}${p(" <")}`,
+    `   ${p("/|")}${p("   |")}${p("\\")}        ${d("\u4F60\u7684\u7F16\u7A0B\u597D\u4F19\u4F34")} ${chalk.hex("#A78BFA")("\u{1F43E}")}`,
+    `  ${p("(_|   |_)")}`
+  ];
+  return lines.join("\n");
 }
 function separator() {
-  return theme.dim("\u2500".repeat(46));
+  return theme.dim("\u2500".repeat(40));
 }
 
 // src/ui/renderer.ts
@@ -1636,13 +1704,41 @@ function getFrameworkGuide(framework) {
 
 // src/utils/config.ts
 import Conf from "conf";
+var PROVIDER_KEY_MAP = {
+  openai: "openaiApiKey",
+  anthropic: "anthropicApiKey",
+  deepseek: "deepseekApiKey",
+  kimi: "kimiApiKey",
+  qwen: "qwenApiKey",
+  zhipu: "zhipuApiKey",
+  doubao: "doubaoApiKey",
+  baichuan: "baichuanApiKey",
+  lingyiwanwu: "lingyiwanwuApiKey",
+  gemini: "geminiApiKey",
+  mistral: "mistralApiKey",
+  groq: "groqApiKey"
+};
+var PROVIDER_NAMES = {
+  openai: "OpenAI",
+  anthropic: "Anthropic Claude",
+  deepseek: "DeepSeek \u6DF1\u5EA6\u6C42\u7D22",
+  kimi: "Kimi \u6708\u4E4B\u6697\u9762",
+  qwen: "\u901A\u4E49\u5343\u95EE Qwen",
+  zhipu: "\u667A\u8C31 GLM",
+  doubao: "\u8C46\u5305 Doubao",
+  baichuan: "\u767E\u5DDD Baichuan",
+  lingyiwanwu: "\u96F6\u4E00\u4E07\u7269 Yi",
+  gemini: "Google Gemini",
+  mistral: "Mistral",
+  groq: "Groq"
+};
 var defaults = {
   locale: "zh",
   model: "kimi-k2.5",
   apiBaseUrl: "http://localhost:3210"
 };
 var config = new Conf({
-  projectName: "devpilot",
+  projectName: "sxai",
   defaults
 });
 function getConfig() {
@@ -1655,11 +1751,29 @@ function getConfig() {
     openaiApiKey: config.get("openaiApiKey"),
     anthropicApiKey: config.get("anthropicApiKey"),
     deepseekApiKey: config.get("deepseekApiKey"),
-    kimiApiKey: config.get("kimiApiKey")
+    kimiApiKey: config.get("kimiApiKey"),
+    qwenApiKey: config.get("qwenApiKey"),
+    zhipuApiKey: config.get("zhipuApiKey"),
+    doubaoApiKey: config.get("doubaoApiKey"),
+    baichuanApiKey: config.get("baichuanApiKey"),
+    lingyiwanwuApiKey: config.get("lingyiwanwuApiKey"),
+    geminiApiKey: config.get("geminiApiKey"),
+    mistralApiKey: config.get("mistralApiKey"),
+    groqApiKey: config.get("groqApiKey")
   };
 }
 function setConfig(key, value) {
-  config.set(key, value);
+  if (value === void 0) {
+    config.delete(key);
+  } else {
+    config.set(key, value);
+  }
+}
+function getProviderApiKey(provider) {
+  const cfg = getConfig();
+  const keyName = PROVIDER_KEY_MAP[provider];
+  if (!keyName) return void 0;
+  return cfg[keyName];
 }
 function getConfigPath() {
   return config.path;
@@ -1733,6 +1847,8 @@ var Agent = class {
       let reasoningContent = "";
       const spinner = ora({ text: t("thinking"), spinner: "dots" }).start();
       let spinnerStopped = false;
+      let hasTextOutput = false;
+      let toolSpinnerStarted = false;
       try {
         for await (const chunk of provider.chatStream(this.messages, this.tools, config3)) {
           if (!spinnerStopped && (chunk.type === "text" || chunk.type === "tool_call")) {
@@ -1741,12 +1857,26 @@ var Agent = class {
             if (chunk.type === "text") {
               process.stdout.write("\n");
             }
+            if (chunk.type === "tool_call" && !hasTextOutput) {
+              spinner.text = "\u6B63\u5728\u51C6\u5907\u64CD\u4F5C...";
+              spinner.start();
+              toolSpinnerStarted = true;
+            }
           }
           switch (chunk.type) {
             case "text":
               responseText += chunk.content || "";
+              hasTextOutput = true;
               streamChunk(chunk.content || "");
               break;
+            case "tool_call_start":
+              if (!toolSpinnerStarted) {
+                if (hasTextOutput) process.stdout.write("\n");
+                spinner.text = "\u6B63\u5728\u51C6\u5907\u64CD\u4F5C...";
+                spinner.start();
+                toolSpinnerStarted = true;
+              }
+              break;
             case "tool_call":
               if (chunk.toolCall) {
                 toolCalls.push(chunk.toolCall);
@@ -1770,11 +1900,13 @@ var Agent = class {
         }
       } catch (err) {
         if (!spinnerStopped) spinner.stop();
+        if (toolSpinnerStarted) spinner.stop();
         const msg = err instanceof Error ? err.message : String(err);
         printError(t("errors.apiError", { message: msg }));
         return "";
       }
       if (!spinnerStopped) spinner.stop();
+      if (toolSpinnerStarted) spinner.stop();
       if (responseText) {
         process.stdout.write("\n");
       }
@@ -1816,43 +1948,29 @@ var Agent = class {
     return finalResponse;
   }
   /**
-   * Get provider configuration based on current settings
+   * Get provider configuration based on current settings.
+   * Priority: user's own API key (BYOK, direct) > server proxy (via authToken)
    */
   getProviderConfig() {
     const cfg = getConfig();
     const model = cfg.model;
     const providerName = getProviderName(model);
-    let apiKey = "";
-    if (cfg.authToken) {
-      apiKey = cfg.authToken;
-    } else {
-      switch (providerName) {
-        case "openai":
-          apiKey = cfg.openaiApiKey || "";
-          break;
-        case "anthropic":
-          apiKey = cfg.anthropicApiKey || "";
-          break;
-        case "deepseek":
-          apiKey = cfg.deepseekApiKey || "";
-          break;
-        case "kimi":
-          apiKey = cfg.kimiApiKey || "";
-          break;
-        case "qwen":
-          apiKey = cfg.openaiApiKey || "";
-          break;
-        default:
-          apiKey = cfg.openaiApiKey || "";
-      }
+    const ownKey = getProviderApiKey(providerName);
+    if (ownKey) {
+      return {
+        apiKey: ownKey,
+        baseUrl: getDefaultBaseUrl(providerName),
+        model
+      };
     }
-    let baseUrl;
     if (cfg.authToken) {
-      baseUrl = `${cfg.apiBaseUrl}/api/ai/proxy`;
-    } else {
-      baseUrl = getDefaultBaseUrl(providerName);
+      return {
+        apiKey: cfg.authToken,
+        baseUrl: `${cfg.apiBaseUrl}/api/ai/proxy`,
+        model
+      };
     }
-    return { apiKey, baseUrl, model };
+    return { apiKey: "", baseUrl: getDefaultBaseUrl(providerName), model };
   }
   /**
    * Trim old messages to prevent context overflow
@@ -1887,31 +2005,44 @@ var Agent = class {
 };
 
 // src/commands/chat.ts
-function question(rl, prompt) {
-  return new Promise((resolve, reject) => {
-    const onClose = () => reject(new Error("readline closed"));
-    rl.once("close", onClose);
-    rl.question(prompt, (answer) => {
-      rl.removeListener("close", onClose);
-      resolve(answer);
-    });
-  });
-}
 async function chatCommand() {
   const config3 = getConfig();
   setLocale(config3.locale);
+  process.on("unhandledRejection", (reason) => {
+    const msg = reason instanceof Error ? reason.message : String(reason);
+    if (msg.includes("abort") || msg.includes("cancel") || msg.includes("ABORT")) return;
+    console.error(theme.error(`
+  \u26A0 \u5F02\u6B65\u9519\u8BEF: ${msg}`));
+  });
+  process.on("uncaughtException", (err) => {
+    console.error(theme.error(`
+  \u26A0 \u672A\u6355\u83B7\u9519\u8BEF: ${err.message}`));
+  });
   console.log(banner());
-  const currentModel = config3.model;
-  console.log(theme.dim(`  \u6A21\u578B ${theme.brandBold(currentModel)}  \xB7  ${t("exitHint")}`));
-  const hasAnyKey = config3.authToken || config3.kimiApiKey || config3.openaiApiKey || config3.anthropicApiKey || config3.deepseekApiKey;
-  if (!hasAnyKey) {
+  console.log(theme.dim(`  ${t("exitHint")}`));
+  const providerName = getProviderName(config3.model);
+  const hasOwnKey = !!getProviderApiKey(providerName);
+  if (hasOwnKey) {
+  } else if (!config3.authToken || !config3.userEmail) {
+    console.log();
+    printInfo("\u9996\u6B21\u4F7F\u7528\uFF0C\u8BF7\u6CE8\u518C\u8D26\u6237\uFF08\u914D\u7F6E\u81EA\u5DF1\u7684API Key\u53EF\u8DF3\u8FC7\u6CE8\u518C\uFF0C\u672C\u5730\u4F7F\u7528\uFF09");
+    console.log(theme.dim("  \u63D0\u793A: sxai set-key kimi <your-key>  \u5373\u53EF\u514D\u767B\u5F55\u4F7F\u7528"));
+    console.log();
+    const loggedIn = await doInlineRegister();
+    if (!loggedIn) {
+      return;
+    }
     console.log();
-    printWarning("\u5C1A\u672A\u914D\u7F6EAPI\u5BC6\u94A5\u3002\u8BF7\u8FD0\u884C\u4EE5\u4E0B\u547D\u4EE4\u4E4B\u4E00:");
-    console.log(theme.dim("    sxai set-key kimi <your-key>     # Kimi (\u6708\u4E4B\u6697\u9762)"));
-    console.log(theme.dim("    sxai set-key deepseek <your-key>"));
-    console.log(theme.dim("    sxai set-key openai <your-key>"));
-    console.log(theme.dim("    sxai config                      # \u4EA4\u4E92\u5F0F\u914D\u7F6E"));
   }
+  registerQuestionFn(async (prompt) => {
+    const { answer } = await inquirer.prompt([{
+      type: "input",
+      name: "answer",
+      message: prompt,
+      prefix: ""
+    }]);
+    return answer;
+  });
   const agent = new Agent();
   try {
     await agent.init();
@@ -1929,36 +2060,20 @@ async function chatCommand() {
     printError(`\u521D\u59CB\u5316\u5931\u8D25: ${err instanceof Error ? err.message : err}`);
     process.exit(1);
   }
-  const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-    terminal: true
-  });
-  registerQuestionFn((prompt2) => question(rl, prompt2));
-  let ctrlCCount = 0;
-  process.on("SIGINT", () => {
-    ctrlCCount++;
-    if (ctrlCCount >= 2) {
-      console.log();
-      printSuccess(t("exit"));
-      process.exit(0);
-    }
-    console.log(theme.dim("\n(\u518D\u6309\u4E00\u6B21 Ctrl+C \u9000\u51FA)"));
-    setTimeout(() => {
-      ctrlCCount = 0;
-    }, 2e3);
-  });
-  const prompt = theme.user(t("prompt"));
   while (true) {
     let input;
     try {
-      input = await question(rl, prompt);
+      const result = await inquirer.prompt([{
+        type: "input",
+        name: "input",
+        message: theme.user("\u4F60:"),
+        prefix: ""
+      }]);
+      input = result.input.trim();
     } catch {
       break;
     }
-    input = input.trim();
     if (!input) continue;
-    ctrlCCount = 0;
     if (input.startsWith("/")) {
       const result = await handleSlashCommand(input, agent);
       if (result === "exit") break;
@@ -1977,11 +2092,75 @@ async function chatCommand() {
     }
     console.log();
   }
-  rl.close();
   console.log();
   printSuccess(t("exit"));
   process.exit(0);
 }
+async function doInlineRegister() {
+  const config3 = getConfig();
+  const { email } = await inquirer.prompt([{
+    type: "input",
+    name: "email",
+    message: "\u90AE\u7BB1:",
+    validate: (v) => v.includes("@") ? true : "\u8BF7\u8F93\u5165\u6709\u6548\u7684\u90AE\u7BB1"
+  }]);
+  const { password } = await inquirer.prompt([{
+    type: "password",
+    name: "password",
+    message: "\u8BBE\u7F6E\u5BC6\u7801 (\u81F3\u5C116\u4F4D):",
+    mask: "*",
+    validate: (v) => v.length >= 6 ? true : "\u5BC6\u7801\u81F3\u5C116\u4F4D"
+  }]);
+  const { name } = await inquirer.prompt([{
+    type: "input",
+    name: "name",
+    message: "\u6635\u79F0:",
+    default: email.split("@")[0]
+  }]);
+  try {
+    const regResp = await fetch(`${config3.apiBaseUrl}/api/auth/register`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ email, password, name })
+    });
+    if (regResp.ok) {
+      const data = await regResp.json();
+      setConfig("authToken", data.token);
+      setConfig("userEmail", data.user.email);
+      console.log();
+      printSuccess(`\u6CE8\u518C\u6210\u529F\uFF01\u6B22\u8FCE ${data.user.name}`);
+      console.log(theme.dim(`  \u8BA1\u5212: ${data.user.plan}  \xB7  \u6BCF\u65E5 ${data.user.dailyLimit} \u6B21\u8BF7\u6C42`));
+      return true;
+    }
+    const regErr = await regResp.json();
+    if (regErr.error.includes("\u5DF2\u6CE8\u518C")) {
+      printInfo("\u8BE5\u90AE\u7BB1\u5DF2\u6CE8\u518C\uFF0C\u6B63\u5728\u5C1D\u8BD5\u767B\u5F55...");
+      const loginResp = await fetch(`${config3.apiBaseUrl}/api/auth/login`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ email, password })
+      });
+      if (loginResp.ok) {
+        const data = await loginResp.json();
+        setConfig("authToken", data.token);
+        setConfig("userEmail", data.user.email);
+        const roleTip = data.user.role === "admin" ? theme.error(" [\u7BA1\u7406\u5458]") : "";
+        printSuccess(`\u767B\u5F55\u6210\u529F\uFF01\u6B22\u8FCE\u56DE\u6765 ${data.user.name || data.user.email}${roleTip}`);
+        return true;
+      } else {
+        printError("\u5BC6\u7801\u9519\u8BEF\u3002\u8BF7\u91CD\u65B0\u8FD0\u884C sxai \u518D\u8BD5\u3002");
+        return false;
+      }
+    }
+    printError(`\u6CE8\u518C\u5931\u8D25: ${regErr.error}`);
+    return false;
+  } catch {
+    printError("\u7F51\u7EDC\u8FDE\u63A5\u5931\u8D25\u3002\u8BF7\u786E\u8BA4\u540E\u7AEF\u670D\u52A1\u6B63\u5728\u8FD0\u884C\u3002");
+    console.log(theme.dim(`  \u540E\u7AEF\u5730\u5740: ${config3.apiBaseUrl}`));
+    console.log(theme.dim("  \u542F\u52A8\u540E\u7AEF: cd packages/server && pnpm dev"));
+    return false;
+  }
+}
 async function handleSlashCommand(input, agent) {
   const [cmd, ...args] = input.split(" ");
   switch (cmd) {
@@ -1991,30 +2170,43 @@ async function handleSlashCommand(input, agent) {
       return "exit";
     case "/help":
     case "/h":
-      console.log(t("help"));
+      console.log(`
+  \u53EF\u7528\u547D\u4EE4:
+    /help      \u663E\u793A\u5E2E\u52A9
+    /model     \u67E5\u770B/\u5207\u6362\u6A21\u578B
+    /clear     \u6E05\u9664\u5BF9\u8BDD\u5386\u53F2
+    /account   \u67E5\u770B\u8D26\u6237\u4FE1\u606F
+    /status    \u67E5\u770B\u9879\u76EE\u4FE1\u606F
+    /config    \u67E5\u770B\u914D\u7F6E
+    /logout    \u9000\u51FA\u767B\u5F55
+    /exit      \u9000\u51FA\u7A0B\u5E8F
+`);
       break;
     case "/clear":
       agent.clearHistory();
       printSuccess("\u5BF9\u8BDD\u5386\u53F2\u5DF2\u6E05\u9664\u3002");
       break;
     case "/model": {
-      const models = listModels();
-      const currentModel = agent.getCurrentModel();
-      console.log("\n\u53EF\u7528\u6A21\u578B:");
-      for (const m of models) {
-        const marker = m.id === currentModel ? theme.success(" \u25C9") : "  \u25CB";
-        console.log(`${marker} ${theme.bold(m.name)} ${theme.dim(`(${m.id})`)}`);
-      }
       if (args[0]) {
-        const target = args[0];
-        if (models.find((m) => m.id === target)) {
-          setConfig("model", target);
-          printSuccess(`\u6A21\u578B\u5DF2\u5207\u6362\u4E3A: ${target}`);
-        } else {
-          printError(`\u672A\u77E5\u6A21\u578B: ${target}`);
-        }
+        setConfig("model", args[0]);
+        printSuccess(`\u6A21\u578B\u5DF2\u5207\u6362\u4E3A: ${args[0]}`);
       } else {
-        console.log(theme.dim("\n\u4F7F\u7528 /model <model-id> \u5207\u6362\u6A21\u578B"));
+        const models = listModels();
+        const currentModel = agent.getCurrentModel();
+        const groups = /* @__PURE__ */ new Map();
+        for (const m of models) {
+          if (!groups.has(m.group)) groups.set(m.group, []);
+          groups.get(m.group).push(m);
+        }
+        for (const [group, groupModels] of groups) {
+          console.log(theme.dim(`
+  \u2500\u2500 ${group} \u2500\u2500`));
+          for (const m of groupModels) {
+            const marker = m.id === currentModel ? theme.success(" \u25C9") : "  \u25CB";
+            console.log(`  ${marker} ${m.name} ${theme.dim(`(${m.id})`)}`);
+          }
+        }
+        console.log(theme.dim("\n  \u4F7F\u7528 /model <model-id> \u5207\u6362\u6A21\u578B"));
       }
       break;
     }
@@ -2031,126 +2223,139 @@ async function handleSlashCommand(input, agent) {
       }
       break;
     }
+    case "/account": {
+      const cfg = getConfig();
+      if (cfg.authToken) {
+        try {
+          const resp = await fetch(`${cfg.apiBaseUrl}/api/auth/me`, {
+            headers: { "Authorization": `Bearer ${cfg.authToken}` }
+          });
+          if (resp.ok) {
+            const data = await resp.json();
+            console.log(`
+  \u90AE\u7BB1: ${data.user.email}`);
+            console.log(`  \u8BA1\u5212: ${data.user.plan}`);
+            if (data.quota) {
+              console.log(`  \u4ECA\u65E5: ${data.quota.daily.used}/${data.quota.daily.limit}\u6B21`);
+              console.log(`  \u603B\u989D: ${fmtTokens(data.quota.lifetimeTokens.used)}/${fmtTokens(data.quota.lifetimeTokens.limit)} tokens`);
+            }
+          }
+        } catch {
+        }
+      } else {
+        printWarning("\u672A\u767B\u5F55");
+      }
+      break;
+    }
     case "/config": {
-      const config3 = getConfig();
+      const cfg = getConfig();
       console.log("\n\u5F53\u524D\u914D\u7F6E:");
-      console.log(`  \u6A21\u578B: ${config3.model}`);
-      console.log(`  \u8BED\u8A00: ${config3.locale}`);
-      console.log(`  \u540E\u7AEF: ${config3.apiBaseUrl}`);
-      console.log(`  \u767B\u5F55: ${config3.userEmail || "\u672A\u767B\u5F55"}`);
-      console.log(`  Kimi Key: ${config3.kimiApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
-      console.log(`  OpenAI Key: ${config3.openaiApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
-      console.log(`  Anthropic Key: ${config3.anthropicApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
-      console.log(`  DeepSeek Key: ${config3.deepseekApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
+      console.log(`  \u6A21\u578B: ${cfg.model}`);
+      console.log(`  \u8BED\u8A00: ${cfg.locale}`);
+      console.log(`  \u540E\u7AEF: ${cfg.apiBaseUrl}`);
+      console.log(`  \u767B\u5F55: ${cfg.userEmail || "\u672A\u767B\u5F55"}`);
       break;
     }
-    case "/login":
-      printInfo("\u8BF7\u4F7F\u7528 sxai login \u547D\u4EE4\u767B\u5F55\u3002");
-      break;
     case "/logout":
       setConfig("authToken", void 0);
       setConfig("userEmail", void 0);
-      printSuccess(t("auth.logoutSuccess"));
-      break;
+      printSuccess("\u5DF2\u9000\u51FA\u767B\u5F55\u3002");
+      return "exit";
     default:
-      printError(t("errors.unknownCommand", { command: cmd }));
+      printError(`\u672A\u77E5\u547D\u4EE4: ${cmd}\u3002\u8F93\u5165 /help \u67E5\u770B\u5E2E\u52A9\u3002`);
   }
 }
+function fmtTokens(n) {
+  if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
+  if (n >= 1e3) return `${(n / 1e3).toFixed(0)}K`;
+  return String(n);
+}
 
 // src/commands/config.ts
-import inquirer from "inquirer";
+import inquirer2 from "inquirer";
 async function configCommand() {
   const config3 = getConfig();
   console.log(`
-\u914D\u7F6E\u6587\u4EF6\u8DEF\u5F84: ${getConfigPath()}
+\u914D\u7F6E\u6587\u4EF6: ${theme.dim(getConfigPath())}
 `);
-  const { action } = await inquirer.prompt([{
+  const { action } = await inquirer2.prompt([{
     type: "list",
     name: "action",
     message: "\u9009\u62E9\u8981\u914D\u7F6E\u7684\u9879\u76EE:",
     choices: [
-      { name: "\u8BBE\u7F6EAI\u6A21\u578B", value: "model" },
-      { name: "\u8BBE\u7F6EKimi API Key (\u6708\u4E4B\u6697\u9762)", value: "kimi" },
-      { name: "\u8BBE\u7F6EOpenAI API Key", value: "openai" },
-      { name: "\u8BBE\u7F6EAnthropic API Key", value: "anthropic" },
-      { name: "\u8BBE\u7F6EDeepSeek API Key", value: "deepseek" },
-      { name: "\u8BBE\u7F6E\u540E\u7AEF\u670D\u52A1\u5730\u5740", value: "apiBaseUrl" },
-      { name: "\u8BBE\u7F6E\u754C\u9762\u8BED\u8A00", value: "locale" },
-      { name: "\u67E5\u770B\u5F53\u524D\u914D\u7F6E", value: "show" },
+      { name: "\u{1F916} \u8BBE\u7F6EAI\u6A21\u578B", value: "model" },
+      new inquirer2.Separator("\u2500\u2500 API Key \u8BBE\u7F6E \u2500\u2500"),
+      { name: "\u{1F319} Kimi \u6708\u4E4B\u6697\u9762", value: "key_kimi" },
+      { name: "\u{1F9E0} OpenAI", value: "key_openai" },
+      { name: "\u{1F3AD} Anthropic Claude", value: "key_anthropic" },
+      { name: "\u{1F50D} DeepSeek \u6DF1\u5EA6\u6C42\u7D22", value: "key_deepseek" },
+      { name: "\u2601\uFE0F  \u901A\u4E49\u5343\u95EE Qwen", value: "key_qwen" },
+      { name: "\u{1F52E} \u667A\u8C31 GLM", value: "key_zhipu" },
+      { name: "\u{1FAD8} \u8C46\u5305 Doubao", value: "key_doubao" },
+      { name: "\u{1F48E} Google Gemini", value: "key_gemini" },
+      { name: "\u{1F32C}\uFE0F  Mistral", value: "key_mistral" },
+      { name: "\u26A1 Groq", value: "key_groq" },
+      { name: "\u{1F3D4}\uFE0F  \u767E\u5DDD Baichuan", value: "key_baichuan" },
+      { name: "\u{1F31F} \u96F6\u4E00\u4E07\u7269 Yi", value: "key_lingyiwanwu" },
+      new inquirer2.Separator("\u2500\u2500 \u5176\u4ED6\u8BBE\u7F6E \u2500\u2500"),
+      { name: "\u{1F310} \u8BBE\u7F6E\u540E\u7AEF\u670D\u52A1\u5730\u5740", value: "apiBaseUrl" },
+      { name: "\u{1F30D} \u8BBE\u7F6E\u754C\u9762\u8BED\u8A00", value: "locale" },
+      { name: "\u{1F4CB} \u67E5\u770B\u5F53\u524D\u914D\u7F6E", value: "show" },
       { name: "\u9000\u51FA", value: "exit" }
     ]
   }]);
+  if (action.startsWith("key_")) {
+    const provider = action.replace("key_", "");
+    const displayName = PROVIDER_NAMES[provider] || provider;
+    const configKey = PROVIDER_KEY_MAP[provider];
+    const currentKey = configKey ? config3[configKey] : void 0;
+    if (currentKey) {
+      printInfo(`\u5F53\u524D\u5DF2\u914D\u7F6E ${displayName} Key`);
+    }
+    const { key } = await inquirer2.prompt([{
+      type: "password",
+      name: "key",
+      message: `\u8F93\u5165 ${displayName} API Key:`,
+      mask: "*"
+    }]);
+    if (key && configKey) {
+      setConfig(configKey, key);
+      printSuccess(`${displayName} API Key \u5DF2\u4FDD\u5B58\u3002\u4F7F\u7528\u8BE5\u6A21\u578B\u4E0D\u6D88\u8017\u5E73\u53F0\u989D\u5EA6\u3002`);
+    }
+    return;
+  }
   switch (action) {
     case "model": {
       const models = listModels();
-      const { model } = await inquirer.prompt([{
+      const groups = /* @__PURE__ */ new Map();
+      for (const m of models) {
+        if (!groups.has(m.group)) groups.set(m.group, []);
+        groups.get(m.group).push(m);
+      }
+      const choices = [];
+      for (const [group, groupModels] of groups) {
+        choices.push(new inquirer2.Separator(`\u2500\u2500 ${group} \u2500\u2500`));
+        for (const m of groupModels) {
+          choices.push({
+            name: `${m.name} (${m.id})`,
+            value: m.id
+          });
+        }
+      }
+      const { model } = await inquirer2.prompt([{
         type: "list",
         name: "model",
         message: "\u9009\u62E9AI\u6A21\u578B:",
-        choices: models.map((m) => ({
-          name: `${m.name} (${m.provider})`,
-          value: m.id
-        })),
-        default: config3.model
+        choices,
+        default: config3.model,
+        pageSize: 20
       }]);
       setConfig("model", model);
       printSuccess(`\u6A21\u578B\u5DF2\u8BBE\u7F6E\u4E3A: ${model}`);
       break;
     }
-    case "kimi": {
-      const { key } = await inquirer.prompt([{
-        type: "password",
-        name: "key",
-        message: "\u8F93\u5165Kimi API Key (\u6708\u4E4B\u6697\u9762):",
-        mask: "*"
-      }]);
-      if (key) {
-        setConfig("kimiApiKey", key);
-        printSuccess("Kimi API Key \u5DF2\u4FDD\u5B58\u3002");
-      }
-      break;
-    }
-    case "openai": {
-      const { key } = await inquirer.prompt([{
-        type: "password",
-        name: "key",
-        message: "\u8F93\u5165OpenAI API Key:",
-        mask: "*"
-      }]);
-      if (key) {
-        setConfig("openaiApiKey", key);
-        printSuccess("OpenAI API Key \u5DF2\u4FDD\u5B58\u3002");
-      }
-      break;
-    }
-    case "anthropic": {
-      const { key } = await inquirer.prompt([{
-        type: "password",
-        name: "key",
-        message: "\u8F93\u5165Anthropic API Key:",
-        mask: "*"
-      }]);
-      if (key) {
-        setConfig("anthropicApiKey", key);
-        printSuccess("Anthropic API Key \u5DF2\u4FDD\u5B58\u3002");
-      }
-      break;
-    }
-    case "deepseek": {
-      const { key } = await inquirer.prompt([{
-        type: "password",
-        name: "key",
-        message: "\u8F93\u5165DeepSeek API Key:",
-        mask: "*"
-      }]);
-      if (key) {
-        setConfig("deepseekApiKey", key);
-        printSuccess("DeepSeek API Key \u5DF2\u4FDD\u5B58\u3002");
-      }
-      break;
-    }
     case "apiBaseUrl": {
-      const { url } = await inquirer.prompt([{
+      const { url } = await inquirer2.prompt([{
         type: "input",
         name: "url",
         message: "\u8F93\u5165\u540E\u7AEF\u670D\u52A1\u5730\u5740:",
@@ -2161,7 +2366,7 @@ async function configCommand() {
       break;
     }
     case "locale": {
-      const { locale } = await inquirer.prompt([{
+      const { locale } = await inquirer2.prompt([{
         type: "list",
         name: "locale",
         message: "\u9009\u62E9\u754C\u9762\u8BED\u8A00:",
@@ -2177,14 +2382,18 @@ async function configCommand() {
     }
     case "show": {
       console.log("\n\u5F53\u524D\u914D\u7F6E:");
-      console.log(`  \u6A21\u578B: ${config3.model}`);
+      console.log(`  \u6A21\u578B: ${theme.brandBold(config3.model)}`);
       console.log(`  \u8BED\u8A00: ${config3.locale}`);
       console.log(`  \u540E\u7AEF: ${config3.apiBaseUrl}`);
-      console.log(`  \u767B\u5F55: ${config3.userEmail || "\u672A\u767B\u5F55"}`);
-      console.log(`  Kimi Key: ${config3.kimiApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
-      console.log(`  OpenAI Key: ${config3.openaiApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
-      console.log(`  Anthropic Key: ${config3.anthropicApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
-      console.log(`  DeepSeek Key: ${config3.deepseekApiKey ? "\u5DF2\u914D\u7F6E" : "\u672A\u914D\u7F6E"}`);
+      console.log(`  \u767B\u5F55: ${config3.userEmail || theme.dim("\u672A\u767B\u5F55")}`);
+      console.log();
+      console.log("  API Key \u914D\u7F6E:");
+      for (const [provider, keyName] of Object.entries(PROVIDER_KEY_MAP)) {
+        const displayName = (PROVIDER_NAMES[provider] || provider).padEnd(20);
+        const hasKey = !!config3[keyName];
+        const status = hasKey ? theme.success("\u2713 \u5DF2\u914D\u7F6E") : theme.dim("\u672A\u914D\u7F6E");
+        console.log(`    ${displayName} ${status}`);
+      }
       break;
     }
     case "exit":
@@ -2193,12 +2402,12 @@ async function configCommand() {
 }
 
 // src/commands/login.ts
-import inquirer2 from "inquirer";
+import inquirer3 from "inquirer";
 async function loginCommand() {
   const config3 = getConfig();
   if (config3.authToken && config3.userEmail) {
     printInfo(`\u5F53\u524D\u5DF2\u767B\u5F55: ${config3.userEmail}`);
-    const { relogin } = await inquirer2.prompt([{
+    const { relogin } = await inquirer3.prompt([{
       type: "confirm",
       name: "relogin",
       message: "\u662F\u5426\u91CD\u65B0\u767B\u5F55\uFF1F",
@@ -2207,7 +2416,7 @@ async function loginCommand() {
     if (!relogin) return;
   }
   console.log(theme.brandBold("\n\u{1F511} \u767B\u5F55\u6C88\u7FD4\u7684AI\u52A9\u624B\n"));
-  const { email } = await inquirer2.prompt([{
+  const { email } = await inquirer3.prompt([{
     type: "input",
     name: "email",
     message: "\u90AE\u7BB1:",
@@ -2216,7 +2425,7 @@ async function loginCommand() {
       return true;
     }
   }]);
-  const { password } = await inquirer2.prompt([{
+  const { password } = await inquirer3.prompt([{
     type: "password",
     name: "password",
     message: "\u5BC6\u7801:",
@@ -2242,7 +2451,7 @@ async function loginCommand() {
       console.log();
     } else if (response.status === 404) {
       printInfo("\u8BE5\u90AE\u7BB1\u5C1A\u672A\u6CE8\u518C");
-      const { register } = await inquirer2.prompt([{
+      const { register } = await inquirer3.prompt([{
         type: "confirm",
         name: "register",
         message: "\u662F\u5426\u7ACB\u5373\u6CE8\u518C\uFF1F",
@@ -2276,12 +2485,12 @@ async function loginCommand() {
 }
 
 // src/commands/register.ts
-import inquirer3 from "inquirer";
+import inquirer4 from "inquirer";
 async function registerCommand() {
   const config3 = getConfig();
   if (config3.authToken && config3.userEmail) {
     printInfo(`\u5F53\u524D\u5DF2\u767B\u5F55: ${config3.userEmail}`);
-    const { proceed } = await inquirer3.prompt([{
+    const { proceed } = await inquirer4.prompt([{
       type: "confirm",
       name: "proceed",
       message: "\u662F\u5426\u6CE8\u518C\u65B0\u8D26\u6237\uFF1F\uFF08\u5C06\u8986\u76D6\u5F53\u524D\u767B\u5F55\u72B6\u6001\uFF09",
@@ -2290,7 +2499,7 @@ async function registerCommand() {
     if (!proceed) return;
   }
   console.log(theme.brandBold("\n\u{1F4DD} \u6CE8\u518C\u6C88\u7FD4\u7684AI\u52A9\u624B\u8D26\u6237\n"));
-  const { email } = await inquirer3.prompt([{
+  const { email } = await inquirer4.prompt([{
     type: "input",
     name: "email",
     message: "\u90AE\u7BB1:",
@@ -2299,13 +2508,13 @@ async function registerCommand() {
       return true;
     }
   }]);
-  const { name } = await inquirer3.prompt([{
+  const { name } = await inquirer4.prompt([{
     type: "input",
     name: "name",
     message: "\u6635\u79F0 (\u53EF\u9009):",
     default: email.split("@")[0]
   }]);
-  const { password } = await inquirer3.prompt([{
+  const { password } = await inquirer4.prompt([{
     type: "password",
     name: "password",
     message: "\u5BC6\u7801 (\u81F3\u5C116\u4F4D):",
@@ -2315,7 +2524,7 @@ async function registerCommand() {
       return true;
     }
   }]);
-  const { confirmPassword } = await inquirer3.prompt([{
+  const { confirmPassword } = await inquirer4.prompt([{
     type: "password",
     name: "confirmPassword",
     message: "\u786E\u8BA4\u5BC6\u7801:",
@@ -2399,9 +2608,9 @@ async function accountCommand() {
       const dailyBar = progressBar(quota.daily.used, quota.daily.limit);
       console.log(`  \u4ECA\u65E5\u8BF7\u6C42  ${dailyBar}  ${quota.daily.used}/${quota.daily.limit}`);
       const monthBar = progressBar(quota.monthlyTokens.used, quota.monthlyTokens.limit);
-      console.log(`  \u672C\u6708Token ${monthBar}  ${fmtTokens(quota.monthlyTokens.used)}/${fmtTokens(quota.monthlyTokens.limit)}`);
+      console.log(`  \u672C\u6708Token ${monthBar}  ${fmtTokens2(quota.monthlyTokens.used)}/${fmtTokens2(quota.monthlyTokens.limit)}`);
       const lifeBar = progressBar(quota.lifetimeTokens.used, quota.lifetimeTokens.limit);
-      console.log(`  \u514D\u8D39\u989D\u5EA6  ${lifeBar}  ${fmtTokens(quota.lifetimeTokens.used)}/${fmtTokens(quota.lifetimeTokens.limit)}`);
+      console.log(`  \u514D\u8D39\u989D\u5EA6  ${lifeBar}  ${fmtTokens2(quota.lifetimeTokens.used)}/${fmtTokens2(quota.lifetimeTokens.limit)}`);
       if (!quota.allowed) {
         console.log();
         printWarning(quota.reason || "\u989D\u5EA6\u5DF2\u7528\u5B8C");
@@ -2411,9 +2620,9 @@ async function accountCommand() {
       console.log();
       console.log(theme.brandBold("  \u{1F4C8} \u7528\u91CF\u7EDF\u8BA1"));
       console.log(theme.dim("  \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500"));
-      console.log(`  \u4ECA\u65E5    ${usage.today.requests}\u6B21\u8BF7\u6C42  \xB7  ${fmtTokens(usage.today.tokens)} tokens`);
-      console.log(`  \u672C\u6708    ${usage.thisMonth.requests}\u6B21\u8BF7\u6C42  \xB7  ${fmtTokens(usage.thisMonth.tokens)} tokens`);
-      console.log(`  \u7D2F\u8BA1    ${usage.lifetime.requests}\u6B21\u8BF7\u6C42  \xB7  ${fmtTokens(usage.lifetime.tokens)} tokens`);
+      console.log(`  \u4ECA\u65E5    ${usage.today.requests}\u6B21\u8BF7\u6C42  \xB7  ${fmtTokens2(usage.today.tokens)} tokens`);
+      console.log(`  \u672C\u6708    ${usage.thisMonth.requests}\u6B21\u8BF7\u6C42  \xB7  ${fmtTokens2(usage.thisMonth.tokens)} tokens`);
+      console.log(`  \u7D2F\u8BA1    ${usage.lifetime.requests}\u6B21\u8BF7\u6C42  \xB7  ${fmtTokens2(usage.lifetime.tokens)} tokens`);
     }
     console.log();
   } catch (err) {
@@ -2433,7 +2642,7 @@ function planLabel(plan) {
       return plan;
   }
 }
-function fmtTokens(n) {
+function fmtTokens2(n) {
   if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
   if (n >= 1e3) return `${(n / 1e3).toFixed(0)}K`;
   return String(n);
@@ -2448,7 +2657,7 @@ function progressBar(used, limit, width = 16) {
 }
 
 // src/commands/admin.ts
-import inquirer4 from "inquirer";
+import inquirer5 from "inquirer";
 async function adminCommand() {
   const config3 = getConfig();
   if (!config3.authToken) {
@@ -2456,7 +2665,7 @@ async function adminCommand() {
     return;
   }
   console.log(theme.brandBold("\n\u2699\uFE0F  \u7BA1\u7406\u5458\u63A7\u5236\u53F0\n"));
-  const { action } = await inquirer4.prompt([{
+  const { action } = await inquirer5.prompt([{
     type: "list",
     name: "action",
     message: "\u9009\u62E9\u64CD\u4F5C:",
@@ -2532,7 +2741,7 @@ async function listUsers(config3) {
       const roleTag = u.role === "admin" ? theme.error("\u7BA1\u7406\u5458") : "\u7528\u6237  ";
       const planTag = u.plan === "free" ? theme.dim("\u514D\u8D39  ") : u.plan === "pro" ? theme.success("\u4E13\u4E1A  ") : theme.info("\u4F01\u4E1A  ");
       console.log(
-        `  ${(u.email + status).padEnd(28)} ${planTag} ${roleTag} ${String(u.dailyLimit).padEnd(6)} ${fmtTokens2(u.totalFreeTokens).padEnd(12)} ${fmtTokens2(u.totalTokensUsed)}`
+        `  ${(u.email + status).padEnd(28)} ${planTag} ${roleTag} ${String(u.dailyLimit).padEnd(6)} ${fmtTokens3(u.totalFreeTokens).padEnd(12)} ${fmtTokens3(u.totalTokensUsed)}`
       );
     }
     console.log();
@@ -2543,7 +2752,7 @@ async function listUsers(config3) {
 async function setUserQuota(config3) {
   const users = await fetchUserList(config3);
   if (!users) return;
-  const { userId } = await inquirer4.prompt([{
+  const { userId } = await inquirer5.prompt([{
     type: "list",
     name: "userId",
     message: "\u9009\u62E9\u7528\u6237:",
@@ -2551,21 +2760,21 @@ async function setUserQuota(config3) {
   }]);
   const selectedUser = users.find((u) => u.id === userId);
   console.log(theme.dim(`
-  \u5F53\u524D\u989D\u5EA6: \u65E5\u9650=${selectedUser.dailyLimit}\u6B21  \u6708Token=${fmtTokens2(selectedUser.monthlyTokenLimit)}  \u514D\u8D39\u603B\u989D=${fmtTokens2(selectedUser.totalFreeTokens)}
+  \u5F53\u524D\u989D\u5EA6: \u65E5\u9650=${selectedUser.dailyLimit}\u6B21  \u6708Token=${fmtTokens3(selectedUser.monthlyTokenLimit)}  \u514D\u8D39\u603B\u989D=${fmtTokens3(selectedUser.totalFreeTokens)}
 `));
-  const { dailyLimit } = await inquirer4.prompt([{
+  const { dailyLimit } = await inquirer5.prompt([{
     type: "number",
     name: "dailyLimit",
     message: "\u6BCF\u65E5\u8BF7\u6C42\u4E0A\u9650:",
     default: selectedUser.dailyLimit
   }]);
-  const { monthlyTokenLimit } = await inquirer4.prompt([{
+  const { monthlyTokenLimit } = await inquirer5.prompt([{
     type: "number",
     name: "monthlyTokenLimit",
     message: "\u6BCF\u6708Token\u4E0A\u9650:",
     default: selectedUser.monthlyTokenLimit
   }]);
-  const { totalFreeTokens } = await inquirer4.prompt([{
+  const { totalFreeTokens } = await inquirer5.prompt([{
     type: "number",
     name: "totalFreeTokens",
     message: "\u514D\u8D39Token\u603B\u989D:",
@@ -2591,13 +2800,13 @@ async function setUserQuota(config3) {
 async function setUserPlan(config3) {
   const users = await fetchUserList(config3);
   if (!users) return;
-  const { userId } = await inquirer4.prompt([{
+  const { userId } = await inquirer5.prompt([{
     type: "list",
     name: "userId",
     message: "\u9009\u62E9\u7528\u6237:",
     choices: users.map((u) => ({ name: `${u.email} (\u5F53\u524D: ${u.plan})`, value: u.id }))
   }]);
-  const { plan } = await inquirer4.prompt([{
+  const { plan } = await inquirer5.prompt([{
     type: "list",
     name: "plan",
     message: "\u9009\u62E9\u8BA1\u5212:",
@@ -2627,7 +2836,7 @@ async function setUserPlan(config3) {
 async function setUserRole(config3) {
   const users = await fetchUserList(config3);
   if (!users) return;
-  const { userId } = await inquirer4.prompt([{
+  const { userId } = await inquirer5.prompt([{
     type: "list",
     name: "userId",
     message: "\u9009\u62E9\u7528\u6237:",
@@ -2636,7 +2845,7 @@ async function setUserRole(config3) {
       value: u.id
     }))
   }]);
-  const { role } = await inquirer4.prompt([{
+  const { role } = await inquirer5.prompt([{
     type: "list",
     name: "role",
     message: "\u8BBE\u7F6E\u89D2\u8272:",
@@ -2665,16 +2874,16 @@ async function setUserRole(config3) {
 async function resetUsage(config3) {
   const users = await fetchUserList(config3);
   if (!users) return;
-  const { userId } = await inquirer4.prompt([{
+  const { userId } = await inquirer5.prompt([{
     type: "list",
     name: "userId",
     message: "\u9009\u62E9\u7528\u6237:",
     choices: users.map((u) => ({
-      name: `${u.email} (\u5DF2\u7528 ${fmtTokens2(u.totalTokensUsed)} tokens)`,
+      name: `${u.email} (\u5DF2\u7528 ${fmtTokens3(u.totalTokensUsed)} tokens)`,
       value: u.id
     }))
   }]);
-  const { confirm } = await inquirer4.prompt([{
+  const { confirm } = await inquirer5.prompt([{
     type: "confirm",
     name: "confirm",
     message: "\u786E\u5B9A\u8981\u91CD\u7F6E\u8BE5\u7528\u6237\u7684\u7D2F\u8BA1\u7528\u91CF\u5417\uFF1F",
@@ -2699,7 +2908,7 @@ async function resetUsage(config3) {
 async function toggleUserStatus(config3) {
   const users = await fetchUserList(config3);
   if (!users) return;
-  const { userId } = await inquirer4.prompt([{
+  const { userId } = await inquirer5.prompt([{
     type: "list",
     name: "userId",
     message: "\u9009\u62E9\u7528\u6237:",
@@ -2737,7 +2946,7 @@ async function showStats(config3) {
     console.log(`  \u7528\u6237\u603B\u6570    ${data.users.total}`);
     console.log(`  \u6D3B\u8DC3\u7528\u6237    ${data.users.active}`);
     console.log(`  \u603B\u8BF7\u6C42\u6570    ${data.usage.totalRequests}`);
-    console.log(`  \u603BToken\u7528\u91CF  ${fmtTokens2(Number(data.usage.totalTokens))}`);
+    console.log(`  \u603BToken\u7528\u91CF  ${fmtTokens3(Number(data.usage.totalTokens))}`);
     console.log();
   } catch {
     printError("\u83B7\u53D6\u7EDF\u8BA1\u5931\u8D25\uFF0C\u8BF7\u68C0\u67E5\u7F51\u7EDC\u8FDE\u63A5\u3002");
@@ -2758,7 +2967,7 @@ async function fetchUserList(config3) {
     return null;
   }
 }
-function fmtTokens2(n) {
+function fmtTokens3(n) {
   if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
   if (n >= 1e3) return `${(n / 1e3).toFixed(0)}K`;
   return String(n);
@@ -2795,19 +3004,20 @@ program.command("admin").description("\u7BA1\u7406\u5458\u63A7\u5236\u53F0\uFF08
 program.command("config").description("\u914D\u7F6E\u6C88\u7FD4\u7684AI\u52A9\u624B").action(async () => {
   await configCommand();
 });
-program.command("set-key <provider> <key>").description("\u8BBE\u7F6EAPI\u5BC6\u94A5 (provider: openai, anthropic, deepseek, kimi)").action((provider, key) => {
-  const keyMap = {
-    openai: "openaiApiKey",
-    anthropic: "anthropicApiKey",
-    deepseek: "deepseekApiKey",
-    kimi: "kimiApiKey"
-  };
-  const configKey = keyMap[provider];
+var providerList = Object.keys(PROVIDER_KEY_MAP).join(", ");
+program.command("set-key <provider> <key>").description(`\u8BBE\u7F6EAPI\u5BC6\u94A5 (${providerList})`).action((provider, key) => {
+  const configKey = PROVIDER_KEY_MAP[provider];
   if (!configKey) {
-    console.error(`\u672A\u77E5\u7684\u63D0\u4F9B\u5546: ${provider}\u3002\u652F\u6301: openai, anthropic, deepseek, kimi`);
+    printError(`\u672A\u77E5\u7684\u63D0\u4F9B\u5546: ${provider}`);
+    console.log(`\u652F\u6301\u7684\u63D0\u4F9B\u5546: ${providerList}`);
     process.exit(1);
   }
   setConfig(configKey, key);
-  printSuccess(`${provider} API Key \u5DF2\u4FDD\u5B58\u3002`);
+  const displayName = PROVIDER_NAMES[provider] || provider;
+  printSuccess(`${displayName} API Key \u5DF2\u4FDD\u5B58\u3002\u914D\u7F6E\u81EA\u5DF1\u7684Key\u540E\uFF0C\u4F7F\u7528\u8BE5\u6A21\u578B\u4E0D\u6D88\u8017\u5E73\u53F0\u989D\u5EA6\u3002`);
+});
+program.command("use <model>").description("\u5207\u6362AI\u6A21\u578B (\u5982: sxai use gpt-4o)").action((model) => {
+  setConfig("model", model);
+  printSuccess(`\u6A21\u578B\u5DF2\u5207\u6362\u4E3A: ${model}`);
 });
 program.parse();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shenxiang-ai-cli",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "description": "沈翔的AI助手 - 终端里的AI全栈开发搭档",
   "type": "module",
   "bin": {