@ai-zen/air 0.2.0 → 0.2.2

package/dist/session/print.js

@@ -0,0 +1,24 @@
+import chalk from "chalk";
+import { DeltaRenderer } from "../delta-renderer.js";
+export async function sendAndPrint(agent, text) {
+    console.log(chalk.green.bold("\n🤖 AI:"));
+    const renderer = new DeltaRenderer({
+        reasoningHeader: "\n\n💭 思考中...\n",
+        contentHeader: "\n\n💭 回答中...\n",
+    });
+    function onChunk(chunk) {
+        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();
+}
+//# sourceMappingURL=print.js.map

package/dist/session/print.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"print.js","sourceRoot":"","sources":["../../src/session/print.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAGrD,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,KAAY,EAAE,IAAY;IAC3D,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1C,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC;QACjC,eAAe,EAAE,iBAAiB;QAClC,aAAa,EAAE,iBAAiB;KACjC,CAAC,CAAC;IAEH,SAAS,OAAO,CAAC,KAAiC;QAChD,MAAM,KAAK,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;QACzC,MAAM,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,IAAI,CAAC;QACtD,IAAI,KAAK;YAAE,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,SAAS,KAAK,KAAK,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACtC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC9B,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAClC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvB,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC/B,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACnC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC7B,OAAO,CAAC,GAAG,EAAE,CAAC;AAChB,CAAC"}

package/dist/session/runtime.d.ts

@@ -0,0 +1,2 @@
+export declare function runConversation(initialMessage?: string): Promise<void>;
+//# sourceMappingURL=runtime.d.ts.map

package/dist/session/runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../../src/session/runtime.ts"],"names":[],"mappings":"AAOA,wBAAsB,eAAe,CAAC,cAAc,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAuC5E"}

package/dist/session/runtime.js

@@ -0,0 +1,46 @@
+import { Message } from "@ai-zen/agents-core";
+import { readConfig, readMessages, saveMessages, saveSnapshot } from "../config.js";
+import { buildAgent } from "../agent-factory.js";
+import { contextSize } from "../migration.js";
+import { SYSTEM_PROMPT, saveAndNew, ask } from "./shared.js";
+import { sendAndPrint } from "./print.js";
+export async function runConversation(initialMessage) {
+    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 = await buildAgent(msgs);
+    const ctx = { 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 saveAndNew(ctx);
+        console.log(`💬 你: ${initialMessage}`);
+        try {
+            await sendAndPrint(ctx.agent, initialMessage);
+            saveMessages(ctx.agent.messages);
+        }
+        catch (err) {
+            console.error(`\n❌ ${err.message}`);
+        }
+        console.log("\n💬 继续对话 (输入 /exit 退出)\n");
+        await ask(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 ask(ctx);
+}
+//# sourceMappingURL=runtime.js.map

package/dist/session/runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../../src/session/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACpF,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAc,aAAa,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AACzE,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,cAAuB;IAC3D,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACpD,OAAO,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACnE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,IAAI,GAAG,YAAY,EAAE,CAAC;IAC1B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;QACvC,YAAY,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,KAAK,GAAU,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,GAAG,GAAe,EAAE,KAAK,EAAE,CAAC;IAElC,IAAI,cAAc,EAAE,CAAC;QACnB,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAC7F,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9C,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC;QACtC,CAAC;QACD,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,SAAS,cAAc,EAAE,CAAC,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;YAC9C,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAAC,OAAO,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAAC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;QACzC,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;QACf,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAC;IAC9E,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC;QACtB,CAAC,CAAC,gBAAgB,QAAQ,OAAO,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,qBAAqB;QACrF,CAAC,CAAC,uCAAuC,CAAC,CAAC;IAE7C,MAAM,GAAG,CAAC,GAAG,CAAC,CAAC;AACjB,CAAC"}

package/dist/session/shared.d.ts

@@ -0,0 +1,8 @@
+import { Agent } from "@ai-zen/agents-core";
+export interface SessionCtx {
+    agent: Agent;
+}
+export declare const SYSTEM_PROMPT: string;
+export declare function saveAndNew(ctx: SessionCtx): Promise<void>;
+export declare function ask(ctx: SessionCtx): Promise<void>;
+//# sourceMappingURL=shared.d.ts.map

package/dist/session/shared.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shared.d.ts","sourceRoot":"","sources":["../../src/session/shared.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAW,MAAM,qBAAqB,CAAC;AAarD,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,KAAK,CAAC;CACd;AAED,eAAO,MAAM,aAAa,QAcd,CAAC;AAEb,wBAAsB,UAAU,CAAC,GAAG,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAI/D;AAgCD,wBAAsB,GAAG,CAAC,GAAG,EAAE,UAAU,iBAaxC"}

package/dist/session/shared.js

@@ -0,0 +1,77 @@
+import { Message } from "@ai-zen/agents-core";
+import inquirer from "inquirer";
+import { saveMessages } from "../config.js";
+import { buildAgent } from "../agent-factory.js";
+import { handleMessage } from "./message.js";
+import { cmdExit } from "./commands/exit.js";
+import { cmdHelp } from "./commands/help.js";
+import { cmdSave } from "./commands/save.js";
+import { cmdNew } from "./commands/new.js";
+import { cmdLoad } from "./commands/load.js";
+import { cmdEditor } from "./commands/editor.js";
+import { cmdBack } from "./commands/back.js";
+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");
+export async function saveAndNew(ctx) {
+    const msgs = [Message.System(SYSTEM_PROMPT)];
+    saveMessages(msgs);
+    ctx.agent = await buildAgent(msgs);
+}
+async function dispatchCommand(ctx, cmd) {
+    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;
+    }
+}
+export async function ask(ctx) {
+    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);
+    }
+}
+//# sourceMappingURL=shared.js.map

package/dist/session/shared.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../src/session/shared.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,QAAQ,MAAM,UAAU,CAAC;AAChC,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAM7C,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,kCAAkC;IAClC,EAAE;IACF,SAAS;IACT,kCAAkC;IAClC,6CAA6C;IAC7C,yFAAyF;IACzF,2GAA2G;IAC3G,EAAE;IACF,OAAO;IACP,8DAA8D;IAC9D,iCAAiC;IACjC,sCAAsC;IACtC,+BAA+B;CAChC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEb,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,GAAe;IAC9C,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC;IAC7C,YAAY,CAAC,IAAI,CAAC,CAAC;IACnB,GAAG,CAAC,KAAK,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;AACrC,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,GAAe,EAAE,GAAW;IACzD,QAAQ,GAAG,EAAE,CAAC;QACZ,KAAK,OAAO,CAAC;QACb,KAAK,OAAO;YACV,OAAO,CAAC,GAAG,CAAC,CAAC;YACb,MAAM;QACR,KAAK,OAAO;YACV,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;YACnB,MAAM;QACR,KAAK,MAAM;YACT,MAAM,MAAM,CAAC,GAAG,CAAC,CAAC;YAClB,MAAM;QACR,KAAK,OAAO;YACV,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;YACnB,MAAM;QACR,KAAK,OAAO;YACV,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;YACnB,MAAM;QACR,KAAK,SAAS;YACZ,MAAM,SAAS,CAAC,GAAG,CAAC,CAAC;YACrB,MAAM;QACR,KAAK,OAAO;YACV,OAAO,EAAE,CAAC;YACV,MAAM;QACR;YACE,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC;YAClC,MAAM;IACV,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,GAAG,CAAC,GAAe;IACvC,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;YACtC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE;SAChD,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,CAAC;YAAE,SAAS;QACjB,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,eAAe,CAAC,GAAG,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;YAC5C,SAAS;QACX,CAAC;QACD,MAAM,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-zen/air",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "Minimalist AI CLI assistant \u2014 one shell tool, filesystem memory, auto context migration",
   "type": "module",
   "main": "./dist/agent-runtime.js",

package/src/__tests__/chat.test.ts

@@ -0,0 +1,147 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import type { ChatCtx } from "../chat/shared.js";
+
+// Mock all external dependencies
+vi.mock("inquirer", () => ({ default: { prompt: vi.fn() } }));
+vi.mock("../config.js", () => ({
+  saveMessages: vi.fn(),
+  saveSnapshot: vi.fn(() => "snapshot-123"),
+  listSnapshots: vi.fn(() => []),
+  loadSnapshot: vi.fn(() => []),
+}));
+vi.mock("../agent-factory.js", () => ({
+  buildAgent: vi.fn(async () => ({
+    messages: [],
+    events: { on: vi.fn(), off: vi.fn() },
+    send: vi.fn(),
+  })),
+}));
+vi.mock("../chat/print.js", () => ({
+  sendAndPrint: vi.fn(),
+}));
+
+const mockPrompt = vi.mocked((await import("inquirer")).default.prompt);
+const { saveMessages, saveSnapshot, listSnapshots, loadSnapshot } = await import("../config.js");
+const { buildAgent } = await import("../agent-factory.js");
+
+function makeCtx(msgs: any[] = []): ChatCtx {
+  return { agent: { messages: msgs, events: { on: vi.fn(), off: vi.fn() }, send: vi.fn() } as any };
+}
+
+// ==================== cmdNew ====================
+
+describe("cmdNew", () => {
+  it("写入 system prompt 并构建 agent", async () => {
+    const { cmdNew } = await import("../chat/commands/new.js");
+    const ctx = makeCtx();
+
+    await cmdNew(ctx);
+
+    expect(saveMessages).toHaveBeenCalledWith(
+      expect.arrayContaining([expect.objectContaining({ role: "system" })])
+    );
+    expect(buildAgent).toHaveBeenCalled();
+    expect(ctx.agent).toBeTruthy();
+  });
+});
+
+// ==================== dispatchCommand 路由 ====================
+
+describe("dispatchCommand", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('"/exit" 调用 cmdExit（process.exit）', async () => {
+    // Can't easily test process.exit, just verify it doesn't throw
+    const { runChat } = await import("../chat/runtime.js");
+    // dispatchCommand is internal, tested via routing only
+  });
+
+  it('"/save" 调用 cmdSave', async () => {
+    const { cmdSave } = await import("../chat/commands/save.js");
+    const ctx = makeCtx([{ role: "system", content: "sys" }]);
+    await cmdSave(ctx);
+    expect(saveSnapshot).toHaveBeenCalledWith(ctx.agent.messages);
+  });
+
+  it('"/new" 调用 cmdNew 并打印提示', async () => {
+    const { cmdNew } = await import("../chat/commands/new.js");
+    const ctx = makeCtx();
+    const spy = vi.spyOn(console, "log").mockImplementation(() => {});
+    await cmdNew(ctx);
+    expect(spy).toHaveBeenCalledWith("\n🆕 新会话\n");
+    spy.mockRestore();
+  });
+
+  it('"/help" 打印帮助信息', async () => {
+    const { cmdHelp } = await import("../chat/commands/help.js");
+    const spy = vi.spyOn(console, "log").mockImplementation(() => {});
+    cmdHelp();
+    expect(spy).toHaveBeenCalled();
+    spy.mockRestore();
+  });
+});
+
+// ==================== cmdBack ====================
+
+describe("cmdBack", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("没有消息时提示无可撤回", async () => {
+    const { cmdBack } = await import("../chat/commands/back.js");
+    const ctx = makeCtx([{ role: "system", content: "sys" }]);
+    const spy = vi.spyOn(console, "log").mockImplementation(() => {});
+    await cmdBack(ctx);
+    expect(spy).toHaveBeenCalledWith("\n❌ 还没有消息可以撤回\n");
+    spy.mockRestore();
+  });
+
+  it("取消操作时不修改消息", async () => {
+    const { cmdBack } = await import("../chat/commands/back.js");
+    const msgs = [
+      { role: "system", content: "sys" },
+      { role: "user", content: "你好" },
+    ];
+    const ctx = makeCtx(msgs);
+    mockPrompt.mockResolvedValueOnce({ selectedIndex: -1 });
+    await cmdBack(ctx);
+    expect(ctx.agent.messages.length).toBe(2);
+  });
+});
+
+// ==================== cmdLoad ====================
+
+describe("cmdLoad", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("没有快照时提示", async () => {
+    const { cmdLoad } = await import("../chat/commands/load.js");
+    const ctx = makeCtx();
+    vi.mocked(listSnapshots).mockReturnValueOnce([]);
+    const spy = vi.spyOn(console, "log").mockImplementation(() => {});
+    await cmdLoad(ctx);
+    expect(spy).toHaveBeenCalledWith("\n📭 没有可用的快照\n");
+    spy.mockRestore();
+  });
+});
+
+// ==================== cmdEditor ====================
+
+describe("cmdEditor", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("取消编辑不报错", async () => {
+    const { cmdEditor } = await import("../chat/commands/editor.js");
+    const ctx = makeCtx();
+    mockPrompt.mockResolvedValueOnce({ content: "" });
+    await cmdEditor(ctx);
+    expect(saveMessages).not.toHaveBeenCalled();
+  });
+});

package/src/__tests__/e2e.test.ts

@@ -0,0 +1,187 @@
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
+import { spawn } from "node:child_process";
+import { existsSync, readFileSync, rmSync, mkdirSync, readdirSync } from "node:fs";
+import { join } from "node:path";
+import { tmpdir } from "node:os";
+
+// ==================== Helpers ====================
+
+const CLI = join(process.cwd(), "dist", "cli.js");
+
+interface CliResult {
+  stdout: string;
+  stderr: string;
+  exitCode: number;
+}
+
+function runCli(args: string[], stdinInput?: string, extraEnv?: Record<string, string>): Promise<CliResult> {
+  return new Promise((resolve) => {
+    const env = {
+      ...process.env,
+      ...extraEnv,
+    };
+
+    const proc = spawn("node", [CLI, ...args], { env, stdio: ["pipe", "pipe", "pipe"] });
+
+    let stdout = "";
+    let stderr = "";
+
+    proc.stdout.on("data", (data: Buffer) => { stdout += data.toString(); });
+    proc.stderr.on("data", (data: Buffer) => { stderr += data.toString(); });
+
+    if (stdinInput !== undefined) {
+      proc.stdin.write(stdinInput);
+      proc.stdin.end();
+    }
+
+    proc.on("close", (code) => {
+      resolve({ stdout, stderr, exitCode: code ?? -1 });
+    });
+    proc.on("error", () => {
+      resolve({ stdout, stderr, exitCode: -1 });
+    });
+  });
+}
+
+let tmpDirs: string[] = [];
+
+afterAll(() => {
+  for (const d of tmpDirs) {
+    try { rmSync(d, { recursive: true, force: true }); } catch {}
+  }
+});
+
+function makeTestDir(): string {
+  const dir = join(tmpdir(), `air-e2e-${Date.now()}-${Math.random().toString(36).slice(2)}`);
+  mkdirSync(dir, { recursive: true });
+  tmpDirs.push(dir);
+  return dir;
+}
+
+// Read API key from .env.local
+function getApiKey(): string {
+  try {
+    const envContent = readFileSync(join(process.cwd(), ".env.local"), "utf-8");
+    const match = envContent.match(/DEEPSEEK_API_KEY=(.+)/);
+    if (match) return match[1].trim();
+  } catch {}
+  return process.env.DEEPSEEK_API_KEY || "";
+}
+
+const API_KEY = getApiKey();
+
+function skipIfNoKey(ctx: any) {
+  if (!API_KEY) ctx.skip();
+}
+
+// ==================== Tests ====================
+
+describe("E2E: CLI 基本功能", () => {
+  it("air --help 显示帮助信息", async () => {
+    const result = await runCli(["--help"]);
+    expect(result.stdout).toContain("Usage:");
+    expect(result.stdout).toContain("air");
+    expect(result.exitCode).toBe(0);
+  });
+
+  it("air key <key> 设置 API Key", async () => {
+    const airDir = makeTestDir();
+    const result = await runCli(["key", "sk-test-12345"], undefined, { AIR_DIR: airDir });
+    expect(result.stdout).toContain("✅ API Key 已设置");
+
+    const configFile = join(airDir, "config.json");
+    expect(existsSync(configFile)).toBe(true);
+    const config = JSON.parse(readFileSync(configFile, "utf-8"));
+    expect(config.apiKey).toBe("sk-test-12345");
+  });
+
+  it("air config 显示配置信息", async () => {
+    const airDir = makeTestDir();
+    await runCli(["key", "sk-test-abcde"], undefined, { AIR_DIR: airDir });
+    const result = await runCli(["config"], undefined, { AIR_DIR: airDir });
+    expect(result.stdout).toContain("API Key:");
+    expect(result.stdout).toContain("bcde");
+  });
+});
+
+
+describe("E2E: 对话命令", () => {
+  it("/save 保存快照", async (ctx) => {
+    skipIfNoKey(ctx);
+
+    const airDir = makeTestDir();
+    const env = { AIR_DIR: airDir };
+    await runCli(["key", API_KEY], undefined, env);
+
+    // 先发一条消息建立历史
+    await runCli(["测试消息"], "/exit\n", env);
+
+    // 交互模式下 /save 后 /exit
+    const result = await runCli([], "/save\n/exit\n", env);
+    expect(result.stdout).toContain("✅ 快照:");
+
+    // 验证快照文件被创建
+    const snapDir = join(airDir, "snapshots");
+    const files = readdirSync(snapDir).filter((f) => f.endsWith(".json"));
+    expect(files.length).toBeGreaterThanOrEqual(1);
+  }, 30000);
+});
+
+describe("E2E: 真实 API 对话", () => {
+  it("air <message> 发送消息并接收真实回复", async (ctx) => {
+    skipIfNoKey(ctx);
+
+    const airDir = makeTestDir();
+    const env = { AIR_DIR: airDir };
+
+    // Set up real key
+    await runCli(["key", API_KEY], undefined, env);
+
+    // Send a simple message, pipe /exit to leave interactive mode
+    const result = await runCli(["用一句话介绍你自己"], "/exit\n", env);
+
+    // Should have AI response
+    expect(result.stdout).toContain("🤖 AI:");
+    // Should have some actual content (not empty)
+    expect(result.stdout.length).toBeGreaterThan(100);
+
+    // Verify context was saved with user message + AI response
+    const contextFile = join(airDir, "context.json");
+    expect(existsSync(contextFile)).toBe(true);
+    const msgs = JSON.parse(readFileSync(contextFile, "utf-8"));
+    expect(msgs.length).toBeGreaterThanOrEqual(2);
+    expect(msgs[1].role).toBe("user");
+    expect(msgs[1].content).toBe("用一句话介绍你自己");
+
+    // Verify there's an assistant response
+    const assistantMsg = msgs.find((m: any) => m.role === "assistant");
+    expect(assistantMsg).toBeTruthy();
+    expect(assistantMsg.content.length).toBeGreaterThan(10);
+  }, 30000);  // 30s timeout for real API
+
+  it("/new 命令重置对话", async (ctx) => {
+    skipIfNoKey(ctx);
+
+    const airDir = makeTestDir();
+    const env = { AIR_DIR: airDir };
+
+    await runCli(["key", API_KEY], undefined, env);
+
+    // Send a message to create history
+    await runCli(["第一轮消息"], "/exit\n", env);
+
+    // Verify there's history
+    let contextFile = join(airDir, "context.json");
+    let msgs = JSON.parse(readFileSync(contextFile, "utf-8"));
+    expect(msgs.length).toBeGreaterThanOrEqual(2);
+
+    // Interactive mode: /new then exit
+    const result = await runCli([], "/new\n/exit\n", env);
+    expect(result.stdout).toContain("🆕 重新开始");
+
+    // Verify context was reset
+    msgs = JSON.parse(readFileSync(contextFile, "utf-8"));
+    expect(msgs.length).toBe(1);
+    expect(msgs[0].role).toBe("system");
+  }, 30000);
+});

package/src/agent-factory.ts

@@ -4,7 +4,7 @@ import { readConfig } from "./config.js";
 import { shellTool } from "./tools.js";
 
 const MODEL_NAME = "deepseek-v4-flash";
-const API_ENDPOINT = "https://api.deepseek.com/v1";
+const API_ENDPOINT = process.env.AIR_API_ENDPOINT || "https://api.deepseek.com/v1";
 
 async function buildModel(apiKey: string) {
   const endpoint = new OpenAI({ openai_endpoint: API_ENDPOINT, api_key: apiKey });

package/src/chat/commands/back.ts

@@ -0,0 +1,90 @@
+import inquirer from "inquirer";
+import type { ChatCtx } from "../shared.js";
+import { sendAndPrint } from "../print.js";
+import { saveMessages } from "../../config.js";
+import { buildAgent } from "../../agent-factory.js";
+
+export async function cmdBack(ctx: ChatCtx): Promise<void> {
+  const targets: { index: number; role: string; label: string; preview: string }[] = [];
+  for (let i = 0; i < ctx.agent.messages.length; i++) {
+    const msg = ctx.agent.messages[i];
+    if (msg.role === "user") {
+      const text = typeof msg.content === "string" ? msg.content : "";
+      if (text) {
+        targets.push({ index: i, role: "user", label: "\u{1F464} 用户", 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: "\u{1F527} 工具", 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: "\u21a9\ufe0f  取消操作", value: -1 },
+        ...targets.map((t) => ({ name: t.label + " " + t.preview, value: t.index })),
+      ],
+    },
+  ]);
+
+  if (selectedIndex === -1) { console.log("\n已取消\n"); return; }
+
+  const selectedMsg = ctx.agent.messages[selectedIndex];
+  const isUserMsg = selectedMsg.role === "user";
+  const originalText = typeof selectedMsg.content === "string" ? selectedMsg.content : "";
+  const sliceEnd = isUserMsg ? selectedIndex : selectedIndex + 1;
+  ctx.agent.messages = ctx.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: "\u270f\ufe0f  修改后重新发送", value: "edit" },
+          { name: "\u{1F504} 直接重新发送", value: "resend" },
+          { name: "\u21a9\ufe0f  取消操作", 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();
+    }
+
+    ctx.agent = await buildAgent(ctx.agent.messages);
+    await sendAndPrint(ctx.agent, textToSend);
+    saveMessages(ctx.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; }
+    ctx.agent = await buildAgent(ctx.agent.messages);
+    await sendAndPrint(ctx.agent, newMessage.trim());
+    saveMessages(ctx.agent.messages);
+  }
+}

package/src/chat/commands/editor.ts

@@ -0,0 +1,18 @@
+import inquirer from "inquirer";
+import type { ChatCtx } from "../shared.js";
+import { sendAndPrint } from "../print.js";
+import { saveMessages } from "../../config.js";
+
+export async function cmdEditor(ctx: ChatCtx) {
+  const { content } = await inquirer.prompt([
+    { type: "editor", name: "content", message: "编辑消息:" },
+  ]);
+  if (!content.trim()) {
+    console.log("\n已取消\n");
+    return;
+  }
+  try {
+    await sendAndPrint(ctx.agent, content.trim());
+    saveMessages(ctx.agent.messages);
+  } catch (err: any) { console.error(`\n❌ ${err.message}`); }
+}

package/src/chat/commands/exit.ts

@@ -0,0 +1,6 @@
+import type { ChatCtx } from "../shared.js";
+
+export async function cmdExit(ctx: ChatCtx): Promise<void> {
+  console.log("\n👋 再见！");
+  process.exit(0);
+}

package/src/chat/commands/help.ts

@@ -0,0 +1,3 @@
+export function cmdHelp() {
+  console.log("\n/exit /quit  退出\n/save        保存快照\n/load        加载快照\n/new         重新开始\n/back        撤回消息\n/help        帮助\n");
+}