shenxiang-ai-cli 0.3.2 → 0.4.0

package/dist/index.js

@@ -255,6 +255,7 @@ var OpenAICompatibleProvider = class {
             const delta = choice.delta || {};
             if (delta.reasoning_content) {
               reasoningContent += delta.reasoning_content;
+              yield { type: "thinking", content: delta.reasoning_content };
             }
             if (delta.content) {
               yield { type: "text", content: delta.content };
@@ -695,6 +696,160 @@ ${sorted.map((f) => `  ${f}`).join("\n")}`;
 // src/core/tools/file-write.ts
 import fs2 from "fs/promises";
 import path2 from "path";
+
+// src/ui/renderer.ts
+import { Marked } from "marked";
+import { markedTerminal } from "marked-terminal";
+import chalk2 from "chalk";
+
+// src/ui/theme.ts
+import chalk from "chalk";
+var theme = {
+  // Brand colors
+  brand: chalk.hex("#A78BFA"),
+  brandBold: chalk.hex("#A78BFA").bold,
+  brandDim: chalk.hex("#7C3AED"),
+  // Status
+  success: chalk.hex("#34D399"),
+  error: chalk.hex("#F87171"),
+  warning: chalk.hex("#FBBF24"),
+  info: chalk.hex("#60A5FA"),
+  // Text
+  dim: chalk.gray,
+  bold: chalk.bold,
+  italic: chalk.italic,
+  muted: chalk.hex("#6B7280"),
+  // Code
+  code: chalk.bgHex("#1F2937").hex("#E5E7EB"),
+  filePath: chalk.underline.hex("#60A5FA"),
+  // Roles
+  user: chalk.hex("#34D399").bold,
+  assistant: chalk.hex("#A78BFA").bold,
+  system: chalk.gray,
+  tool: chalk.hex("#FBBF24"),
+  // Token stats
+  tokenLabel: chalk.hex("#6B7280"),
+  tokenValue: chalk.hex("#A78BFA")
+};
+function banner() {
+  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.4.0")}`,
+    `   ${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(40));
+}
+
+// src/ui/renderer.ts
+var marked = new Marked(markedTerminal());
+function streamChunk(text) {
+  process.stdout.write(text);
+}
+function streamThinking(text) {
+  process.stdout.write(theme.dim(text));
+}
+function printColoredDiff(oldContent, newContent, filepath) {
+  const oldLines = oldContent.split("\n");
+  const newLines = newContent.split("\n");
+  const maxLines = Math.max(oldLines.length, newLines.length);
+  const diffLines = [];
+  let additions = 0;
+  let deletions = 0;
+  let prefixLen = 0;
+  while (prefixLen < Math.min(oldLines.length, newLines.length) && oldLines[prefixLen] === newLines[prefixLen]) {
+    prefixLen++;
+  }
+  let suffixLen = 0;
+  while (suffixLen < Math.min(oldLines.length - prefixLen, newLines.length - prefixLen) && oldLines[oldLines.length - 1 - suffixLen] === newLines[newLines.length - 1 - suffixLen]) {
+    suffixLen++;
+  }
+  const oldStart = prefixLen;
+  const oldEnd = oldLines.length - suffixLen;
+  const newStart = prefixLen;
+  const newEnd = newLines.length - suffixLen;
+  const contextStart = Math.max(0, prefixLen - 2);
+  for (let i = contextStart; i < prefixLen; i++) {
+    diffLines.push(theme.dim(`  ${String(i + 1).padStart(4)}  ${oldLines[i]}`));
+  }
+  for (let i = oldStart; i < oldEnd; i++) {
+    diffLines.push(chalk2.red(`  ${String(i + 1).padStart(4)} - ${oldLines[i]}`));
+    deletions++;
+  }
+  for (let i = newStart; i < newEnd; i++) {
+    diffLines.push(chalk2.green(`  ${String(i + 1).padStart(4)} + ${newLines[i]}`));
+    additions++;
+  }
+  const contextEnd = Math.min(newLines.length, newEnd + 2);
+  for (let i = newEnd; i < contextEnd; i++) {
+    diffLines.push(theme.dim(`  ${String(i + 1).padStart(4)}  ${newLines[i]}`));
+  }
+  const maxDiffLines = 30;
+  if (diffLines.length > maxDiffLines) {
+    const shown = diffLines.slice(0, maxDiffLines);
+    shown.push(theme.dim(`  ... ${diffLines.length - maxDiffLines} more lines`));
+    console.log(shown.join("\n"));
+  } else if (diffLines.length > 0) {
+    console.log(diffLines.join("\n"));
+  }
+  console.log(
+    `  ${theme.filePath(filepath)} ` + chalk2.green(`+${additions}`) + " " + chalk2.red(`-${deletions}`) + " lines"
+  );
+}
+function printToolCall(name, description) {
+  console.log(theme.tool(`  \u25B8 ${description}`));
+}
+function printToolResult(result, truncate = 500) {
+  const display = result.length > truncate ? result.substring(0, truncate) + theme.dim(`
+    ... (${result.length} \u5B57\u7B26)`) : result;
+  const indented = display.split("\n").map((l) => "    " + l).join("\n");
+  console.log(theme.dim(indented));
+}
+function printTokenUsage(usage, durationMs) {
+  const parts = [];
+  parts.push(theme.tokenLabel("  tokens: "));
+  parts.push(theme.tokenValue(`\u2191${usage.promptTokens}`));
+  parts.push(theme.tokenLabel(" + "));
+  parts.push(theme.tokenValue(`\u2193${usage.completionTokens}`));
+  parts.push(theme.tokenLabel(" = "));
+  parts.push(theme.brandBold(`${usage.totalTokens}`));
+  if (durationMs && durationMs > 0) {
+    const seconds = (durationMs / 1e3).toFixed(1);
+    const tokPerSec = usage.completionTokens > 0 ? (usage.completionTokens / (durationMs / 1e3)).toFixed(0) : "0";
+    parts.push(theme.tokenLabel(`  \u23F1 ${seconds}s`));
+    parts.push(theme.tokenLabel(` (${tokPerSec} tok/s)`));
+  }
+  console.log(parts.join(""));
+}
+function printError(message) {
+  console.log(theme.error(`  \u2717 ${message}`));
+}
+function printSuccess(message) {
+  console.log(theme.success(`  \u2713 ${message}`));
+}
+function printInfo(message) {
+  console.log(theme.info(`  \u2139 ${message}`));
+}
+function printWarning(message) {
+  console.log(theme.warning(`  \u26A0 ${message}`));
+}
+function printSeparator() {
+  console.log(separator());
+}
+
+// src/core/tools/file-write.ts
 var writeFileTool = {
   definition: {
     name: "write_file",
@@ -717,12 +872,25 @@ var writeFileTool = {
   handler: async (args) => {
     try {
       const filePath = path2.resolve(process.cwd(), args.path);
+      let isNew = true;
+      let oldContent = "";
+      try {
+        oldContent = await fs2.readFile(filePath, "utf-8");
+        isNew = false;
+      } catch {
+      }
       await fs2.mkdir(path2.dirname(filePath), { recursive: true });
       await fs2.writeFile(filePath, args.content, "utf-8");
       const lines = args.content.split("\n").length;
+      if (!isNew && oldContent) {
+        try {
+          printColoredDiff(oldContent, args.content, args.path);
+        } catch {
+        }
+      }
       return {
         success: true,
-        output: `File written: ${args.path} (${lines} lines)`
+        output: isNew ? `File created: ${args.path} (${lines} lines)` : `File overwritten: ${args.path} (${lines} lines)`
       };
     } catch (err) {
       const msg = err instanceof Error ? err.message : String(err);
@@ -777,6 +945,10 @@ var editFileTool = {
       }
       const newContent = content.replace(oldStr, newStr);
       await fs2.writeFile(filePath, newContent, "utf-8");
+      try {
+        printColoredDiff(content, newContent, args.path);
+      } catch {
+      }
       const addedLines = newStr.split("\n").length;
       const removedLines = oldStr.split("\n").length;
       return {
@@ -1137,107 +1309,6 @@ var gitLogTool = {
   requiresConfirmation: false
 };
 
-// src/ui/renderer.ts
-import { Marked } from "marked";
-import { markedTerminal } from "marked-terminal";
-
-// src/ui/theme.ts
-import chalk from "chalk";
-var theme = {
-  // Brand colors
-  brand: chalk.hex("#A78BFA"),
-  brandBold: chalk.hex("#A78BFA").bold,
-  brandDim: chalk.hex("#7C3AED"),
-  // Status
-  success: chalk.hex("#34D399"),
-  error: chalk.hex("#F87171"),
-  warning: chalk.hex("#FBBF24"),
-  info: chalk.hex("#60A5FA"),
-  // Text
-  dim: chalk.gray,
-  bold: chalk.bold,
-  italic: chalk.italic,
-  muted: chalk.hex("#6B7280"),
-  // Code
-  code: chalk.bgHex("#1F2937").hex("#E5E7EB"),
-  filePath: chalk.underline.hex("#60A5FA"),
-  // Roles
-  user: chalk.hex("#34D399").bold,
-  assistant: chalk.hex("#A78BFA").bold,
-  system: chalk.gray,
-  tool: chalk.hex("#FBBF24"),
-  // Token stats
-  tokenLabel: chalk.hex("#6B7280"),
-  tokenValue: chalk.hex("#A78BFA")
-};
-function banner() {
-  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.3.2")}`,
-    `   ${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(40));
-}
-
-// src/ui/renderer.ts
-var marked = new Marked(markedTerminal());
-function streamChunk(text) {
-  process.stdout.write(text);
-}
-function printToolCall(name, description) {
-  console.log(theme.tool(`  \u25B8 ${description}`));
-}
-function printToolResult(result, truncate = 500) {
-  const display = result.length > truncate ? result.substring(0, truncate) + theme.dim(`
-    ... (${result.length} \u5B57\u7B26)`) : result;
-  const indented = display.split("\n").map((l) => "    " + l).join("\n");
-  console.log(theme.dim(indented));
-}
-function printTokenUsage(usage, durationMs) {
-  const parts = [];
-  parts.push(theme.tokenLabel("  tokens: "));
-  parts.push(theme.tokenValue(`\u2191${usage.promptTokens}`));
-  parts.push(theme.tokenLabel(" + "));
-  parts.push(theme.tokenValue(`\u2193${usage.completionTokens}`));
-  parts.push(theme.tokenLabel(" = "));
-  parts.push(theme.brandBold(`${usage.totalTokens}`));
-  if (durationMs && durationMs > 0) {
-    const seconds = (durationMs / 1e3).toFixed(1);
-    const tokPerSec = usage.completionTokens > 0 ? (usage.completionTokens / (durationMs / 1e3)).toFixed(0) : "0";
-    parts.push(theme.tokenLabel(`  \u23F1 ${seconds}s`));
-    parts.push(theme.tokenLabel(` (${tokPerSec} tok/s)`));
-  }
-  console.log(parts.join(""));
-}
-function printError(message) {
-  console.log(theme.error(`  \u2717 ${message}`));
-}
-function printSuccess(message) {
-  console.log(theme.success(`  \u2713 ${message}`));
-}
-function printInfo(message) {
-  console.log(theme.info(`  \u2139 ${message}`));
-}
-function printWarning(message) {
-  console.log(theme.warning(`  \u26A0 ${message}`));
-}
-function printSeparator() {
-  console.log(separator());
-}
-
 // src/i18n/zh.ts
 var zh = {
   welcome: "\u6B22\u8FCE\u4F7F\u7528\u6C88\u7FD4\u7684AI\u52A9\u624B \u2014 \u4F60\u7684AI\u5168\u6808\u5F00\u53D1\u642D\u6863",
@@ -1410,6 +1481,13 @@ async function askQuestion(prompt) {
 
 // src/core/tools/index.ts
 var toolRegistry = /* @__PURE__ */ new Map();
+var _yoloMode = false;
+function setYoloMode(enabled) {
+  _yoloMode = enabled;
+}
+function isYoloMode() {
+  return _yoloMode;
+}
 function registerAllTools() {
   const tools = [
     readFileTool,
@@ -1439,8 +1517,9 @@ async function confirmExecution(toolName, args) {
   else if (toolName === "delete_file") description = `\u5220\u9664\u6587\u4EF6: ${args.path}`;
   else if (toolName === "git_commit") description = `Git\u63D0\u4EA4: ${args.message}`;
   const answer = await askQuestion(`  \u26A1 ${description}
-  \u5141\u8BB8\u6267\u884C\uFF1F(y/n) `);
-  return answer.toLowerCase() === "y" || answer.toLowerCase() === "yes";
+  \u5141\u8BB8\u6267\u884C\uFF1F(Y/n) `);
+  const a = answer.toLowerCase().trim();
+  return a === "" || a === "y" || a === "yes";
 }
 async function executeTool(name, args, autoConfirm = false) {
   const tool = toolRegistry.get(name);
@@ -1461,7 +1540,8 @@ async function executeTool(name, args, autoConfirm = false) {
     git_log: "\u{1F4DC} \u67E5\u770BGit\u65E5\u5FD7"
   };
   printToolCall(name, descMap[name] || name);
-  if (tool.requiresConfirmation && !autoConfirm) {
+  const shouldConfirm = tool.requiresConfirmation && !autoConfirm && !_yoloMode;
+  if (shouldConfirm) {
     const confirmed = await confirmExecution(name, args);
     if (!confirmed) {
       return { success: false, output: "", error: "User declined execution." };
@@ -1550,6 +1630,20 @@ async function buildProjectContext(rootDir) {
   } catch {
   }
   ctx.structure = await buildStructureTree(rootDir);
+  try {
+    const memoryPaths = ["SXAI.md", ".sxai.md", "sxai.md"];
+    for (const mp of memoryPaths) {
+      try {
+        const content = await fs4.readFile(path4.join(rootDir, mp), "utf-8");
+        if (content.trim()) {
+          ctx.projectMemory = content.trim();
+          break;
+        }
+      } catch {
+      }
+    }
+  } catch {
+  }
   return ctx;
 }
 async function buildStructureTree(rootDir, maxDepth = 2) {
@@ -1612,6 +1706,11 @@ ${ctx.structure}`);
 [\u5DE5\u5177] \u4F60\u6709\u6587\u4EF6\u8BFB\u5199(read_file/write_file/edit_file/delete_file)\u3001\u76EE\u5F55(list_files)\u3001\u641C\u7D22(search_code)\u3001\u547D\u4EE4(run_command)\u3001Git(git_status/git_diff/git_commit/git_log)\u7B49\u5DE5\u5177\u3002\u5411\u7528\u6237\u63CF\u8FF0\u64CD\u4F5C\u65F6\u7528\u81EA\u7136\u8BED\u8A00\uFF0C\u4E0D\u66B4\u9732\u5DE5\u5177\u51FD\u6570\u540D\u3002`);
   parts.push(`
 [\u683C\u5F0F] Markdown\u56DE\u590D\uFF0C\u4EE3\u7801\u5757\u6807\u6CE8\u8BED\u8A00\uFF0C\u64CD\u4F5C\u524D\u89E3\u91CA\u539F\u56E0\uFF0C\u5B8C\u6210\u540E\u7B80\u8981\u603B\u7ED3\u3002\u9996\u6B21\u5BF9\u8BDD\u4ECB\u7ECD\u81EA\u5DF1\u5E76\u8BE2\u95EE\u9700\u6C42\u3002`);
+  if (ctx.projectMemory) {
+    parts.push(`
+[\u9879\u76EE\u6307\u4EE4 - \u6765\u81EA SXAI.md\uFF0C\u8BF7\u4E25\u683C\u9075\u5B88]
+${ctx.projectMemory}`);
+  }
   return parts.join("\n");
 }
 function getFrameworkGuide(framework) {
@@ -1775,13 +1874,16 @@ var Agent = class {
       const spinner = ora({ text: t("thinking"), spinner: "dots" }).start();
       let spinnerStopped = false;
       let hasTextOutput = false;
+      let hasThinkingOutput = 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")) {
+          if (!spinnerStopped && (chunk.type === "text" || chunk.type === "tool_call" || chunk.type === "thinking")) {
             spinner.stop();
             spinnerStopped = true;
-            if (chunk.type === "text") {
+            if (chunk.type === "thinking") {
+              process.stdout.write("\n");
+            } else if (chunk.type === "text") {
               process.stdout.write("\n");
             }
             if (chunk.type === "tool_call" && !hasTextOutput) {
@@ -1791,7 +1893,17 @@ var Agent = class {
             }
           }
           switch (chunk.type) {
+            case "thinking":
+              if (!hasThinkingOutput) {
+                hasThinkingOutput = true;
+                streamThinking("\u{1F4AD} ");
+              }
+              streamThinking(chunk.content || "");
+              break;
             case "text":
+              if (hasThinkingOutput && !hasTextOutput) {
+                process.stdout.write("\n\n");
+              }
               responseText += chunk.content || "";
               hasTextOutput = true;
               streamChunk(chunk.content || "");
@@ -1973,9 +2085,12 @@ var Agent = class {
 };
 
 // src/commands/chat.ts
-async function chatCommand() {
+async function chatCommand(options = {}) {
   const config3 = getConfig();
   setLocale(config3.locale);
+  if (options.yolo) {
+    setYoloMode(true);
+  }
   process.on("unhandledRejection", (reason) => {
     const msg = reason instanceof Error ? reason.message : String(reason);
     if (msg.includes("abort") || msg.includes("cancel") || msg.includes("ABORT")) return;
@@ -1987,6 +2102,9 @@ async function chatCommand() {
   \u26A0 \u672A\u6355\u83B7\u9519\u8BEF: ${err.message}`));
   });
   console.log(banner());
+  if (isYoloMode()) {
+    console.log(theme.warning("  \u26A1 \u81EA\u52A8\u6A21\u5F0F (YOLO) \u2014 \u6240\u6709\u64CD\u4F5C\u81EA\u52A8\u6267\u884C\uFF0C\u65E0\u9700\u786E\u8BA4"));
+  }
   console.log(theme.dim(`  ${t("exitHint")}`));
   const providerName = getProviderName(config3.model);
   const hasOwnKey = !!getProviderApiKey(providerName);
@@ -2022,6 +2140,9 @@ async function chatCommand() {
       }
       console.log(theme.dim(`  \u9879\u76EE ${theme.info(parts.join(" \xB7 "))}`));
     }
+    if (projectCtx?.projectMemory) {
+      console.log(theme.dim(`  \u{1F4CB} \u5DF2\u52A0\u8F7D SXAI.md \u9879\u76EE\u6307\u4EE4`));
+    }
     printSeparator();
     console.log();
   } catch (err) {
@@ -2142,6 +2263,7 @@ async function handleSlashCommand(input, agent) {
   \u53EF\u7528\u547D\u4EE4:
     /help      \u663E\u793A\u5E2E\u52A9
     /model     \u67E5\u770B/\u5207\u6362\u6A21\u578B
+    /auto      \u5207\u6362\u81EA\u52A8\u6A21\u5F0F\uFF08\u8DF3\u8FC7\u786E\u8BA4\uFF09
     /clear     \u6E05\u9664\u5BF9\u8BDD\u5386\u53F2
     /account   \u67E5\u770B\u8D26\u6237\u4FE1\u606F
     /status    \u67E5\u770B\u9879\u76EE\u4FE1\u606F
@@ -2150,6 +2272,17 @@ async function handleSlashCommand(input, agent) {
     /exit      \u9000\u51FA\u7A0B\u5E8F
 `);
       break;
+    case "/auto":
+    case "/yolo": {
+      const newState = !isYoloMode();
+      setYoloMode(newState);
+      if (newState) {
+        printWarning("\u26A1 \u81EA\u52A8\u6A21\u5F0F\u5DF2\u5F00\u542F \u2014 \u6240\u6709\u64CD\u4F5C\u5C06\u81EA\u52A8\u6267\u884C\uFF0C\u65E0\u9700\u786E\u8BA4");
+      } else {
+        printSuccess("\u81EA\u52A8\u6A21\u5F0F\u5DF2\u5173\u95ED \u2014 \u5371\u9669\u64CD\u4F5C\u5C06\u518D\u6B21\u8BF7\u6C42\u786E\u8BA4");
+      }
+      break;
+    }
     case "/clear":
       agent.clearHistory();
       printSuccess("\u5BF9\u8BDD\u5386\u53F2\u5DF2\u6E05\u9664\u3002");
@@ -2945,12 +3078,12 @@ function fmtTokens3(n) {
 var program = new Command();
 var config2 = getConfig();
 setLocale(config2.locale);
-program.name("sxai").description(t("description")).version("0.1.0");
-program.command("chat", { isDefault: true }).description("\u542F\u52A8\u4EA4\u4E92\u5F0FAI\u5BF9\u8BDD\uFF08\u9ED8\u8BA4\u547D\u4EE4\uFF09").option("-m, --model <model>", "\u6307\u5B9AAI\u6A21\u578B", config2.model).action(async (options) => {
+program.name("sxai").description(t("description")).version("0.4.0");
+program.command("chat", { isDefault: true }).description("\u542F\u52A8\u4EA4\u4E92\u5F0FAI\u5BF9\u8BDD\uFF08\u9ED8\u8BA4\u547D\u4EE4\uFF09").option("-m, --model <model>", "\u6307\u5B9AAI\u6A21\u578B", config2.model).option("-y, --yolo", "\u81EA\u52A8\u6A21\u5F0F\uFF1A\u8DF3\u8FC7\u6240\u6709\u786E\u8BA4\uFF0C\u8BA9AI\u81EA\u4E3B\u6267\u884C").action(async (options) => {
   if (options.model && options.model !== config2.model) {
     setConfig("model", options.model);
   }
-  await chatCommand();
+  await chatCommand({ yolo: !!options.yolo });
 });
 program.command("register").description("\u6CE8\u518C\u65B0\u8D26\u6237").action(async () => {
   await registerCommand();

package/package.json

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