@ai-zen/air 0.2.0 → 0.2.2

package/src/chat/commands/index.ts

@@ -0,0 +1,38 @@
+import type { ChatCtx } from "../shared.js";
+import { cmdExit } from "./exit.js";
+import { cmdHelp } from "./help.js";
+import { cmdSave } from "./save.js";
+import { cmdNew } from "./new.js";
+import { cmdLoad } from "./load.js";
+import { cmdEditor } from "./editor.js";
+import { cmdBack } from "./back.js";
+
+export async function dispatchCommand(ctx: ChatCtx, cmd: string): Promise<void> {
+  switch (cmd) {
+    case "/exit":
+    case "/quit":
+      cmdExit(ctx);
+      break;
+    case "/save":
+      await cmdSave(ctx);
+      break;
+    case "/new":
+      await cmdNew(ctx);
+      break;
+    case "/load":
+      await cmdLoad(ctx);
+      break;
+    case "/back":
+      await cmdBack(ctx);
+      break;
+    case "/editor":
+      await cmdEditor(ctx);
+      break;
+    case "/help":
+      cmdHelp();
+      break;
+    default:
+      console.log(`\n❌ 未知命令: ${cmd}\n`);
+      break;
+  }
+}

package/src/chat/commands/load.ts

@@ -0,0 +1,38 @@
+import { Message } from "@ai-zen/agents-core";
+import inquirer from "inquirer";
+import type { ChatCtx } from "../shared.js";
+import { SYSTEM_PROMPT } from "../shared.js";
+import { listSnapshots, loadSnapshot, saveMessages } from "../../config.js";
+import { buildAgent } from "../../agent-factory.js";
+
+export async function cmdLoad(ctx: ChatCtx): Promise<void> {
+  const snapshots = listSnapshots();
+  if (snapshots.length === 0) {
+    console.log("\n📭 没有可用的快照\n");
+    return;
+  }
+  const { selectedName } = await inquirer.prompt([
+    {
+      type: "list",
+      name: "selectedName",
+      message: "选择要加载的快照:",
+      pageSize: 15,
+      choices: [
+        { name: "\u21a9\ufe0f  取消操作", value: "" },
+        ...snapshots.map((s) => ({ name: s.date, value: s.name })),
+      ],
+    },
+  ]);
+  if (!selectedName) {
+    console.log("\n已取消\n");
+    return;
+  }
+  let loaded = loadSnapshot(selectedName);
+  if (loaded.length === 0) {
+    loaded = [Message.System(SYSTEM_PROMPT)];
+  }
+  saveMessages(loaded);
+  const snap = snapshots.find((s) => s.name === selectedName);
+  console.log("\n✅ 已加载快照: " + (snap ? snap.date : selectedName) + "\n");
+  ctx.agent = await buildAgent(loaded);
+}

package/src/chat/commands/new.ts

@@ -0,0 +1,12 @@
+import type { ChatCtx } from "../shared.js";
+import { Message } from "@ai-zen/agents-core";
+import { saveMessages } from "../../config.js";
+import { buildAgent } from "../../agent-factory.js";
+import { SYSTEM_PROMPT } from "../shared.js";
+
+export async function cmdNew(ctx: ChatCtx): Promise<void> {
+  const msgs = [Message.System(SYSTEM_PROMPT)];
+  saveMessages(msgs);
+  ctx.agent = await buildAgent(msgs);
+  console.log("\n🆕 新会话\n");
+}

package/src/chat/commands/save.ts

@@ -0,0 +1,6 @@
+import type { ChatCtx } from "../shared.js";
+import { saveSnapshot } from "../../config.js";
+
+export async function cmdSave(ctx: ChatCtx) {
+  console.log(`\n✅ 快照: ${saveSnapshot(ctx.agent.messages)}\n`);
+}

package/src/chat/message.ts

@@ -0,0 +1,27 @@
+import { Message } from "@ai-zen/agents-core";
+import type { ChatCtx } from "./shared.js";
+import { SYSTEM_PROMPT } from "./shared.js";
+import { sendAndPrint } from "./print.js";
+import { saveMessages, saveSnapshot } from "../config.js";
+import { buildAgent } from "../agent-factory.js";
+import { contextSize, shouldMigrate, generateMigrationDoc, MAX_CONTEXT_CHARS } from "../migration.js";
+
+export async function handleMessage(ctx: ChatCtx, text: string): Promise<void> {
+  try {
+    await sendAndPrint(ctx.agent, text);
+    saveMessages(ctx.agent.messages);
+
+    if (shouldMigrate(ctx.agent.messages)) {
+      console.log(`🔄 上下文 ${contextSize(ctx.agent.messages)}/${MAX_CONTEXT_CHARS}，准备迁移...`);
+      try {
+        const snap = saveSnapshot(ctx.agent.messages);
+        console.log(`  💾 快照: ${snap}`);
+        const summary = await generateMigrationDoc(ctx.agent.messages);
+        const msgs = [Message.System(SYSTEM_PROMPT), Message.User(summary)];
+        saveMessages(msgs);
+        console.log("✅ 迁移完成\n");
+        ctx.agent = await buildAgent(msgs);
+      } catch (err: any) { console.error(`❌ 迁移失败: ${err.message}\n`); }
+    }
+  } catch (err: any) { console.error(`\n❌ ${err.message}`); }
+}

package/src/chat/print.ts

@@ -0,0 +1,27 @@
+import chalk from "chalk";
+import { Agent } from "@ai-zen/agents-core";
+import { DeltaRenderer } from "../delta-renderer.js";
+import type { AgentNS } from "@ai-zen/agents-core";
+
+export async function sendAndPrint(agent: Agent, text: string): Promise<void> {
+  console.log(chalk.green.bold("\n🤖 AI:"));
+  const renderer = new DeltaRenderer({
+    reasoningHeader: "\n\n💭 思考中...\n",
+    contentHeader: "\n\n💭 回答中...\n",
+  });
+
+  function onChunk(chunk: AgentNS.StreamResponseData) {
+    const delta = chunk?.choices?.[0]?.delta;
+    const fr = chunk?.choices?.[0]?.finish_reason ?? null;
+    if (delta) renderer.render(delta, fr);
+  }
+
+  function onRun() { renderer.reset(); }
+  agent.events.on("run", onRun);
+  agent.events.on("chunk", onChunk);
+  await agent.send(text);
+  agent.events.off("run", onRun);
+  agent.events.off("chunk", onChunk);
+  process.stdout.write("\n\n");
+  console.log();
+}

package/src/chat/runtime.ts

@@ -0,0 +1,65 @@
+import { Agent, Message } from "@ai-zen/agents-core";
+import inquirer from "inquirer";
+import { readConfig, readMessages, saveMessages, saveSnapshot } from "../config.js";
+import { buildAgent } from "../agent-factory.js";
+import { contextSize } from "../migration.js";
+import { ChatCtx, SYSTEM_PROMPT } from "./shared.js";
+import { dispatchCommand } from "./commands/index.js";
+import { handleMessage } from "./message.js";
+import { sendAndPrint } from "./print.js";
+
+async function chatLoop(ctx: ChatCtx) {
+  while (true) {
+    const { input } = await inquirer.prompt([
+      { type: "input", name: "input", message: "你:" },
+    ]);
+    const t = input.trim();
+    if (!t) continue;
+    if (t.startsWith("/")) {
+      await dispatchCommand(ctx, t.toLowerCase());
+      continue;
+    }
+    await handleMessage(ctx, t);
+  }
+}
+
+export async function runChat(initialMessage?: string): Promise<void> {
+  const config = readConfig();
+  if (!config.apiKey) {
+    console.error("❌ 请先设置 API Key: air key <your-key>");
+    console.error("   获取 Key: https://platform.deepseek.com/api_keys");
+    process.exit(1);
+  }
+
+  let msgs = readMessages();
+  if (msgs.length === 0) {
+    msgs = [Message.System(SYSTEM_PROMPT)];
+    saveMessages(msgs);
+  }
+  const agent: Agent = await buildAgent(msgs);
+  const ctx: ChatCtx = { agent };
+
+  if (initialMessage) {
+    const hasHistory = ctx.agent.messages.some(m => m.role === "user" || m.role === "assistant");
+    if (hasHistory) {
+      const snap = saveSnapshot(ctx.agent.messages);
+      console.log(`💾 已存档旧对话: ${snap}\n`);
+    }
+    await dispatchCommand(ctx, "/new");
+    console.log(`💬 你: ${initialMessage}`);
+    try {
+      await sendAndPrint(ctx.agent, initialMessage);
+      saveMessages(ctx.agent.messages);
+    } catch (err: any) { console.error(`\n❌ ${err.message}`); }
+    console.log("\n💬 继续对话 (输入 /exit 退出)\n");
+    await chatLoop(ctx);
+    return;
+  }
+
+  const msgCount = ctx.agent.messages.filter((m) => m.role !== "system").length;
+  console.log(msgCount > 0
+    ? `\n💬 继续上次对话 (${msgCount} 条，/${contextSize(ctx.agent.messages)} 字符，输入 /new 重新开始)\n`
+    : "\n💬 air — 极简 AI 助手 (输入 /help 查看命令)\n");
+
+  await chatLoop(ctx);
+}

package/src/chat/shared.ts

@@ -0,0 +1,21 @@
+import { Agent } from "@ai-zen/agents-core";
+
+export interface ChatCtx {
+  agent: Agent;
+}
+
+export const SYSTEM_PROMPT = [
+  "你是一个AI助手，专门帮助用户回答问题和执行任务。请用中文回复。",
+  "",
+  "## 行为准则",
+  "1. 做任何改动之前，必须先跟用户商量，获得明确指示之后再行动。",
+  "2. 用户没有明确要求产出文件时，不得自行创建任何文件到项目中。讨论就停留在讨论层面。",
+  "3. 区分危险操作：删除文件、覆盖文件、安装卸载软件、修改系统配置、执行耗时任务等属于危险操作。执行前必须评估影响范围，并向用户说明风险，获得用户明确的书面的确认之后再执行。",
+  "4. 追责原则：你的每一步操作都应当基于用户的明确指令。如果出了问题，可以从用户说过的话追溯责任——是用户让你做的，用户承担责任。所以你不需要畏手畏脚，只要用户明确说了\u201c做\u201d，你就放心去做。",
+  "",
+  "## 记忆",
+  "你可以使用 shell 工具执行任何命令。如果你有需要长期记住的信息（用户偏好、项目约定、任务进度等），请写入以下位置：",
+  "- 全局记忆: ~/.ai-zen/air/temp/*.md",
+  "- 项目记忆: $(cwd)/.ai-zen/air/temp/*.md",
+  "下次启动时用 shell 读取即可。这是你唯一的记忆方式。",
+].join("\n");

package/src/cli.ts

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import { readFileSync } from "node:fs";
 import { Command } from "commander";
-import { runConversation } from "./agent-runtime.js";
+import { runChat } from "./chat/runtime.js";
 import { installHook, uninstallHook } from "./hook.js";
 import { readConfig, saveConfig } from "./config.js";
 
@@ -49,15 +49,15 @@ program
   });
 
 program
-  .argument("[message]", "要发送的消息（不传则进入交互模式）")
-  .action(async (message?: string) => {
+  .argument("[message...]", "要发送的消息（不传则进入交互模式）")
+  .action(async (message?: string[]) => {
     const config = readConfig();
     if (!config.apiKey) {
       console.error("❌ 请先设置 API Key: air key <your-key>");
       console.error("   获取 Key: https://platform.deepseek.com/api_keys");
       process.exit(1);
     }
-    await runConversation(message);
+    await runChat(message?.join(" "));
   });
 
 program.parse(process.argv);

package/src/config.ts

@@ -10,7 +10,7 @@ export interface Config {
 
 // ==================== 路径 ====================
 
-const AIR_DIR = () => join(homedir(), ".ai-zen", "air");
+const AIR_DIR = () => process.env.AIR_DIR || join(homedir(), ".ai-zen", "air");
 const CONFIG_FILE = () => join(AIR_DIR(), "config.json");
 const CONTEXT_FILE = () => join(AIR_DIR(), "context.json");
 const SNAPSHOTS_DIR = () => join(AIR_DIR(), "snapshots");

package/src/agent-runtime.ts

@@ -1,265 +0,0 @@
-import chalk from "chalk";
-import { Agent, Message } from "@ai-zen/agents-core";
-import { DeltaRenderer } from "./delta-renderer.js";
-import type { AgentNS } from "@ai-zen/agents-core";
-import inquirer from "inquirer";
-import { readConfig, readMessages, saveMessages, clearMessages, saveSnapshot, listSnapshots, loadSnapshot } from "./config.js";
-import { buildAgent } from "./agent-factory.js";
-import { contextSize, shouldMigrate, generateMigrationDoc, MAX_CONTEXT_CHARS } from "./migration.js";
-
-const SYSTEM_PROMPT = [
-  "你是一个AI助手，专门帮助用户回答问题和执行任务。请用中文回复。",
-  "",
-  "## 行为准则",
-  "1. 做任何改动之前，必须先跟用户商量，获得明确指示之后再行动。",
-  "2. 用户没有明确要求产出文件时，不得自行创建任何文件到项目中。讨论就停留在讨论层面。",
-  "3. 区分危险操作：删除文件、覆盖文件、安装卸载软件、修改系统配置、执行耗时任务等属于危险操作。执行前必须评估影响范围，并向用户说明风险，获得用户明确的书面的确认之后再执行。",
-  "4. 追责原则：你的每一步操作都应当基于用户的明确指令。如果出了问题，可以从用户说过的话追溯责任——是用户让你做的，用户承担责任。所以你不需要畏手畏脚，只要用户明确说了\u201c做\u201d，你就放心去做。",
-  "",
-  "## 记忆",
-  "你可以使用 shell 工具执行任何命令。如果你有需要长期记住的信息（用户偏好、项目约定、任务进度等），请写入以下位置：",
-  "- 全局记忆: ~/.ai-zen/air/temp/*.md",
-  "- 项目记忆: $(cwd)/.ai-zen/air/temp/*.md",
-  "下次启动时用 shell 读取即可。这是你唯一的记忆方式。",
-].join("\n");
-
-// 流式发送
-
-async function sendAndPrint(agent: Agent, text: string): Promise<void> {
-  const renderer = new DeltaRenderer({
-    reasoningHeader: "\n\n💭 思考中...\n",
-    contentHeader: "\n\n💭 回答中...\n",
-  });
-
-  function onChunk(chunk: AgentNS.StreamResponseData) {
-    const delta = chunk?.choices?.[0]?.delta;
-    const fr = chunk?.choices?.[0]?.finish_reason ?? null;
-    if (delta) renderer.render(delta, fr);
-  }
-
-  function onRun() { renderer.reset(); }
-  agent.events.on("run", onRun);
-  agent.events.on("chunk", onChunk);
-  await agent.send(text);
-  agent.events.off("run", onRun);
-  agent.events.off("chunk", onChunk);
-  process.stdout.write("\n\n");
-  console.log();
-}
-
-async function sendAndSave(agent: Agent, text: string): Promise<void> {
-  console.log(chalk.green.bold("\n🤖 AI:"));
-  await sendAndPrint(agent, text);
-}
-
-// 对话循环
-
-export async function runConversation(initialMessage?: string): Promise<void> {
-  const config = readConfig();
-  if (!config.apiKey) {
-    console.error("❌ 请先设置 API Key: air key <your-key>");
-    console.error("   获取 Key: https://platform.deepseek.com/api_keys");
-    process.exit(1);
-  }
-
-  let msgs = readMessages();
-  if (msgs.length === 0) {
-    msgs = [Message.System(SYSTEM_PROMPT)];
-    saveMessages(msgs);
-  }
-  let agent = await buildAgent(msgs);
-
-  if (initialMessage) {
-    // 有旧对话则先存档，再开新对话
-    const hasHistory = agent.messages.some(m => m.role === "user" || m.role === "assistant");
-    if (hasHistory) {
-      const snap = saveSnapshot(agent.messages);
-      console.log(`💾 已存档旧对话: ${snap}\n`);
-    }
-    msgs = [Message.System(SYSTEM_PROMPT)];
-    saveMessages(msgs);
-    agent = await buildAgent(msgs);
-    console.log(`💬 你: ${initialMessage}`);
-    try {
-      await sendAndSave(agent, initialMessage);
-      saveMessages(agent.messages);
-      console.log("💾 已保存\n");
-    } catch (err: any) { console.error(`\n❌ ${err.message}`); }
-    return;
-  }
-
-  const msgCount = agent.messages.filter((m) => m.role !== "system").length;
-  console.log(msgCount > 0
-    ? `\n💬 继续上次对话 (${msgCount} 条，/${contextSize(agent.messages)} 字符，输入 /new 重新开始)\n`
-    : "\n💬 air — 极简 AI 助手 (输入 /help 查看命令)\n");
-
-  async function ask() {
-    const { input } = await inquirer.prompt([
-      { type: "input", name: "input", message: "你:" },
-    ]);
-    const t = input.trim();
-    if (!t) { await ask(); return; }
-
-    if (t.startsWith("/")) {
-      const c = t.toLowerCase();
-      if (c === "/exit" || c === "/quit") { console.log("\n👋 再见！"); process.exit(0); }
-      if (c === "/save") { console.log(`\n✅ 快照: ${saveSnapshot(agent.messages)}\n`); await ask(); return; }
-      if (c === "/new") { clearMessages(); const msgs = [Message.System(SYSTEM_PROMPT)]; saveMessages(msgs); agent = await buildAgent(msgs); console.log("\n🆕 重新开始\n"); await ask(); return; }
-      if (c === "/load") {
-        const snapshots = listSnapshots();
-        if (snapshots.length === 0) {
-          console.log("\n📭 没有可用的快照\n");
-          await ask(); return;
-        }
-        const { selectedName } = await inquirer.prompt([
-          {
-            type: "list",
-            name: "selectedName",
-            message: "选择要加载的快照:",
-            pageSize: 15,
-            choices: [
-              { name: "↩️  取消操作", value: "" },
-              ...snapshots.map((s) => ({ name: s.date, value: s.name })),
-            ],
-          },
-        ]);
-        if (!selectedName) {
-          console.log("\n已取消\n");
-          await ask(); return;
-        }
-        let msgs = loadSnapshot(selectedName);
-        if (msgs.length === 0) {
-          msgs = [Message.System(SYSTEM_PROMPT)];
-        }
-        saveMessages(msgs);
-        agent = await buildAgent(msgs);
-        const snap = snapshots.find((s) => s.name === selectedName);
-        console.log("\n✅ 已加载快照: " + (snap ? snap.date : selectedName) + "\n");
-        await ask(); return;
-      }
-      if (c === "/back") {
-        await handleBack();
-        await ask(); return;
-      }
-      if (c === "/editor") {
-        const { content } = await inquirer.prompt([
-          { type: "editor", name: "content", message: "编辑消息:" },
-        ]);
-        if (!content.trim()) { console.log("\n已取消\n"); await ask(); return; }
-        try {
-          await sendAndSave(agent, content.trim());
-          saveMessages(agent.messages);
-        } catch (err: any) { console.error(`\n❌ ${err.message}`); }
-        await ask(); return;
-      }
-      if (c === "/help") { console.log("\n/exit /quit  退出\n/save        保存快照\n/load        加载快照\n/new         重新开始\n/back        撤回消息\n/help        帮助\n"); await ask(); return; }
-      console.log(`\n❌ 未知命令: ${t}\n`); await ask(); return; }
-
-    try {
-      await sendAndSave(agent, t);
-      saveMessages(agent.messages);
-
-      if (shouldMigrate(agent.messages)) {
-        console.log(`🔄 上下文 ${contextSize(agent.messages)}/${MAX_CONTEXT_CHARS}，准备迁移...`);
-        try {
-          const snap = saveSnapshot(agent.messages);
-          console.log(`  💾 快照: ${snap}`);
-          const summary = await generateMigrationDoc(agent.messages);
-          const msgs = [Message.System(SYSTEM_PROMPT), Message.User(summary)];
-          saveMessages(msgs);
-          agent = await buildAgent(msgs);
-          console.log("✅ 迁移完成\n");
-        } catch (err: any) { console.error(`❌ 迁移失败: ${err.message}\n`); }
-      }
-    } catch (err: any) { console.error(`\n❌ ${err.message}`); }
-    await ask();
-  }
-
-  async function handleBack(): Promise<void> {
-    const targets: { index: number; role: string; label: string; preview: string }[] = [];
-    for (let i = 0; i < agent.messages.length; i++) {
-      const msg = agent.messages[i];
-      if (msg.role === "user") {
-        const text = typeof msg.content === "string" ? msg.content : "";
-        if (text) {
-          targets.push({ index: i, role: "user", label: "👤 用户", preview: text.substring(0, 60) + (text.length > 60 ? "..." : "") });
-        }
-      } else if (msg.role === "tool" || msg.role === "function") {
-        const text = typeof msg.content === "string" ? msg.content : "";
-        if (text) {
-          targets.push({ index: i, role: msg.role, label: "🔧 工具", preview: text.substring(0, 60) + (text.length > 60 ? "..." : "") });
-        }
-      }
-    }
-
-    if (targets.length === 0) {
-      console.log("\n❌ 还没有消息可以撤回\n");
-      return;
-    }
-
-    console.log("\n📋 选择要撤回到哪条消息（将删除所选及其之后的所有内容）：\n");
-
-    const { selectedIndex } = await inquirer.prompt([
-      {
-        type: "list",
-        name: "selectedIndex",
-        message: "撤回到:",
-        pageSize: 15,
-        choices: [
-          { name: "↩️  取消操作", value: -1 },
-          ...targets.map((t) => ({ name: t.label + " " + t.preview, value: t.index })),
-        ],
-      },
-    ]);
-
-    if (selectedIndex === -1) { console.log("\n已取消\n"); return; }
-
-    const selectedMsg = agent.messages[selectedIndex];
-    const isUserMsg = selectedMsg.role === "user";
-    const originalText = typeof selectedMsg.content === "string" ? selectedMsg.content : "";
-    const sliceEnd = isUserMsg ? selectedIndex : selectedIndex + 1;
-    agent.messages = agent.messages.slice(0, sliceEnd);
-
-    if (isUserMsg) {
-      console.log("\n原内容: " + originalText.substring(0, 200) + (originalText.length > 200 ? "..." : "") + "\n");
-      const { editChoice } = await inquirer.prompt([
-        {
-          type: "list",
-          name: "editChoice",
-          message: "请选择:",
-          choices: [
-            { name: "✏️  修改后重新发送", value: "edit" },
-            { name: "🔄 直接重新发送", value: "resend" },
-            { name: "↩️  取消操作", value: "cancel" },
-          ],
-        },
-      ]);
-      if (editChoice === "cancel") { console.log("\n已取消\n"); return; }
-
-      let textToSend = originalText;
-      if (editChoice === "edit") {
-        const { editedContent } = await inquirer.prompt([
-          { type: "input", name: "editedContent", message: "修改消息:", default: originalText },
-        ]);
-        if (!editedContent.trim()) { console.log("\n❌ 消息不能为空\n"); return; }
-        textToSend = editedContent.trim();
-      }
-
-      agent = await buildAgent(agent.messages);
-      await sendAndSave(agent, textToSend);
-      saveMessages(agent.messages);
-    } else {
-      console.log("\n💡 请输入一条新消息继续对话\n");
-      const { newMessage } = await inquirer.prompt([
-        { type: "input", name: "newMessage", message: "新消息:" },
-      ]);
-      if (!newMessage.trim()) { console.log("\n已取消\n"); return; }
-      agent = await buildAgent(agent.messages);
-      await sendAndSave(agent, newMessage.trim());
-      saveMessages(agent.messages);
-    }
-  }
-
-  await ask();
-  process.on("SIGINT", () => { console.log("\n\n👋 再见！"); process.exit(0); });
-}