@ai-zen/air 0.1.3

package/dist/delta-renderer.js

@@ -0,0 +1,104 @@
+import chalk from "chalk";
+import { AgentNS } from "@ai-zen/agents-core";
+export class DeltaRenderer {
+    reasoningPrinted = false;
+    contentPrinted = false;
+    toolPrints = {};
+    opts;
+    constructor(opts = {}) {
+        this.opts = {
+            reasoningHeader: opts.reasoningHeader ?? "",
+            contentHeader: opts.contentHeader ?? "",
+            reasoningStyle: opts.reasoningStyle ?? chalk.blue,
+            contentStyle: opts.contentStyle ?? chalk.white,
+            indent: opts.indent ?? "",
+        };
+    }
+    render(delta, finishReason = null) {
+        if (delta.tool_calls) {
+            this.renderToolCalls(delta, finishReason);
+        }
+        if (delta.reasoning_content) {
+            if (!this.reasoningPrinted && this.opts.reasoningHeader) {
+                process.stdout.write(this.opts.indent + this.opts.reasoningHeader);
+                this.reasoningPrinted = true;
+            }
+            process.stdout.write(this.opts.reasoningStyle(delta.reasoning_content));
+        }
+        if (delta.content) {
+            if (!this.contentPrinted && this.opts.contentHeader) {
+                process.stdout.write(this.opts.indent + this.opts.contentHeader);
+                this.contentPrinted = true;
+            }
+            if (typeof delta.content === "string") {
+                process.stdout.write(this.opts.contentStyle(delta.content));
+            }
+            else if (Array.isArray(delta.content)) {
+                for (const s of delta.content) {
+                    if (s.type === "text" && s.text) {
+                        process.stdout.write(this.opts.contentStyle(s.text));
+                    }
+                }
+            }
+        }
+    }
+    reset() {
+        this.reasoningPrinted = false;
+        this.contentPrinted = false;
+        for (const k in this.toolPrints)
+            delete this.toolPrints[k];
+    }
+    renderToolCalls(delta, finishReason) {
+        if (!delta.tool_calls || delta.tool_calls.length === 0)
+            return;
+        const isFirstToolCall = Object.keys(this.toolPrints).length === 0 &&
+            delta.tool_calls.some((tc) => tc.function?.name || tc.function?.arguments);
+        if (isFirstToolCall)
+            process.stdout.write(chalk.blue.bold("\n\n💭 工具调用中..."));
+        for (const tc of delta.tool_calls) {
+            const index = tc.index ?? 0;
+            const func = tc.function;
+            if (!this.toolPrints[index]) {
+                this.toolPrints[index] = {
+                    name: "",
+                    arguments: "",
+                    namePrinted: false,
+                    argsPrinted: false,
+                    completed: false,
+                };
+            }
+            const p = this.toolPrints[index];
+            if (func?.name)
+                p.name += func.name;
+            if (func?.arguments)
+                p.arguments += func.arguments;
+            if (p.name && !p.namePrinted) {
+                process.stdout.write(chalk.magenta.bold(`\n🔧 ${index} ${p.name}\n`));
+                p.namePrinted = true;
+            }
+            if (p.arguments && !p.argsPrinted && p.namePrinted)
+                p.argsPrinted = true;
+            if (func?.arguments && p.argsPrinted)
+                process.stdout.write(chalk.gray(func.arguments));
+            if (finishReason === AgentNS.FinishReason.ToolCalls)
+                p.completed = true;
+        }
+        if (finishReason === AgentNS.FinishReason.ToolCalls) {
+            for (const idx of Object.keys(this.toolPrints).map(Number)) {
+                const p = this.toolPrints[idx];
+                if (p.completed && p.arguments) {
+                    process.stdout.write("\n");
+                    try {
+                        const parsed = JSON.parse(p.arguments);
+                        process.stdout.write(chalk.gray(`    ${JSON.stringify(parsed, null, 4)}\n`));
+                    }
+                    catch {
+                        process.stdout.write(chalk.gray(`    ${p.arguments}\n`));
+                    }
+                }
+            }
+            process.stdout.write("\n");
+        }
+    }
+}
+//# sourceMappingURL=delta-renderer.js.map

package/dist/delta-renderer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"delta-renderer.js","sourceRoot":"","sources":["../src/delta-renderer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAkB9C,MAAM,OAAO,aAAa;IAChB,gBAAgB,GAAG,KAAK,CAAC;IACzB,cAAc,GAAG,KAAK,CAAC;IACvB,UAAU,GAAkC,EAAE,CAAC;IAC/C,IAAI,CAAiC;IAE7C,YAAY,OAA6B,EAAE;QACzC,IAAI,CAAC,IAAI,GAAG;YACV,eAAe,EAAE,IAAI,CAAC,eAAe,IAAI,EAAE;YAC3C,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,EAAE;YACvC,cAAc,EAAE,IAAI,CAAC,cAAc,IAAI,KAAK,CAAC,IAAI;YACjD,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,KAAK;YAC9C,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,KAAoB,EAAE,eAA4C,IAAI;QAC3E,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACrB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;gBACxD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;gBACnE,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC/B,CAAC;YACD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;gBACpD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACjE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC7B,CAAC;YACD,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACtC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YAC9D,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;gBACxC,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAC9B,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;wBAChC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;oBACvD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAC9B,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,eAAe,CAAC,KAAoB,EAAE,YAAyC;QACrF,IAAI,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE/D,MAAM,eAAe,GACnB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,KAAK,CAAC;YACzC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAE7E,IAAI,eAAe;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAE3D,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC;YAC5B,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC;YAEzB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG;oBACvB,IAAI,EAAE,EAAE;oBACR,SAAS,EAAE,EAAE;oBACb,WAAW,EAAE,KAAK;oBAClB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,KAAK;iBACjB,CAAC;YACJ,CAAC;YAED,MAAM,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEjC,IAAI,IAAI,EAAE,IAAI;gBAAE,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;YACpC,IAAI,IAAI,EAAE,SAAS;gBAAE,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;YAEnD,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;gBAC7B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;gBACtE,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC;YACvB,CAAC;YAED,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,WAAW;gBAAE,CAAC,CAAC,WAAW,GAAG,IAAI,CAAC;YACzE,IAAI,IAAI,EAAE,SAAS,IAAI,CAAC,CAAC,WAAW;gBAClC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;YAEnD,IAAI,YAAY,KAAK,OAAO,CAAC,YAAY,CAAC,SAAS;gBAAE,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;QAC1E,CAAC;QAED,IAAI,YAAY,KAAK,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YACpD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3D,MAAM,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBAC/B,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;oBAC/B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC3B,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;wBACvC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;oBAC/E,CAAC;oBAAC,MAAM,CAAC;wBACP,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;YACH,CAAC;YACG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;CACF"}

package/dist/migration.d.ts

@@ -0,0 +1,5 @@
+export declare const MAX_CONTEXT_CHARS = 500000;
+export declare function contextSize(messages: any[]): number;
+export declare function shouldMigrate(messages: any[]): boolean;
+export declare function generateMigrationDoc(messages: any[]): Promise<string>;
+//# sourceMappingURL=migration.d.ts.map

package/dist/migration.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"migration.d.ts","sourceRoot":"","sources":["../src/migration.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,iBAAiB,SAAS,CAAC;AAExC,wBAAgB,WAAW,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,MAAM,CAEnD;AAED,wBAAgB,aAAa,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,OAAO,CAEtD;AAED,wBAAsB,oBAAoB,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CA0D3E"}

package/dist/migration.js

@@ -0,0 +1,70 @@
+import { Agent, Message, OpenAI, ChatGPT } from "@ai-zen/agents-core";
+import { readConfig } from "./config.js";
+export const MAX_CONTEXT_CHARS = 500000;
+export function contextSize(messages) {
+    return JSON.stringify(messages).length;
+}
+export function shouldMigrate(messages) {
+    return contextSize(messages) >= MAX_CONTEXT_CHARS;
+}
+export async function generateMigrationDoc(messages) {
+    const config = readConfig();
+    const endpoint = new OpenAI({ openai_endpoint: "https://api.deepseek.com/v1", api_key: config.apiKey });
+    const model = new ChatGPT({
+        model_config: {},
+        request_config: await endpoint.chatCompletion("deepseek-v4-flash"),
+    });
+    const agent = new Agent({
+        model,
+        messages: [Message.System(`你是一个专业的任务交接分析师。你的任务是阅读一段完整的AI助手与用户的对话历史，筛选出对后续工作有用的关键信息，生成一份结构清晰的交接文档。
+
+这份交接文档将作为新会话的第一条用户消息，让新的AI助手了解任务背景并继续工作。
+
+请按以下模板生成 Markdown 格式的文档：
+
+## ✅ 已完成的任务
+列出已经完成的任务及其产出物（文件路径、代码片段位置等）。
+如果没有已完成的任务，保留此标题并注明"无"。
+注意：只记录任务标题和产出路径，不需要描述完成过程的细节。
+
+## 📋 未完成的任务
+列出所有待继续完成的任务，包含：
+- 任务描述
+- 当前进度
+- 下一步需要做什么
+- 相关的文件路径
+- 优先级（如果对话中提到过）
+
+## 🧠 重要记忆
+记录所有对后续工作有影响的信息，包括但不限于：
+- 用户的技术偏好和约定
+- 踩过的坑和教训
+- 项目特定的架构决策
+- 任何对后续工作有指导意义的信息
+
+## 📁 文件索引
+按用途分类列出对话中涉及的重要文件路径，方便新 Agent 按需阅读。
+**请对每个文件注明其用途或作用**，让接手者能快速判断哪些文件需要优先阅读。
+
+## 🔔 接手指令
+接手后请按以下步骤操作：
+1. **读文件** — 使用 shell 工具（cat、grep、find 等）读取「文件索引」中列出的关键文件，以实际代码为准，不要依赖训练数据或过往经验
+2. **对状态** — 确认代码中的关键常量、函数签名、配置等是否与文档描述一致
+3. **再行动** — 确认无误后再继续未完成任务，做任何改动前先跟用户商量
+
+---
+
+注意事项：
+1. 只包含确定的信息，不要猜测或补充对话中没有的内容
+2. 已完成的任务只需列出任务标题和产出路径，不要罗列完成过程的每一步
+3. 如果某个部分没有需要记录的内容，标注"无"即可
+4. 语言风格与原始对话一致（中文）`)],
+        tools: [],
+    });
+    const result = await agent.send(`请阅读以下对话历史，生成交接文档：\n\n${messages.map((m) => `[${m.role}] ${typeof m.content === "string" ? m.content : ""}`).join("\n\n")}`);
+    const last = result.at(-1);
+    if (!last || last.status === "error")
+        throw new Error("生成交接文档失败");
+    return typeof last.content === "string" ? last.content : "";
+}
+//# sourceMappingURL=migration.js.map

package/dist/migration.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"migration.js","sourceRoot":"","sources":["../src/migration.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,CAAC,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAExC,MAAM,UAAU,WAAW,CAAC,QAAe;IACzC,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,QAAe;IAC3C,OAAO,WAAW,CAAC,QAAQ,CAAC,IAAI,iBAAiB,CAAC;AACpD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,QAAe;IACxD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,EAAE,eAAe,EAAE,6BAA6B,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACxG,MAAM,KAAK,GAAG,IAAI,OAAO,CAAC;QACxB,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,MAAM,QAAQ,CAAC,cAAc,CAAC,mBAAmB,CAAC;KACnE,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;QACtB,KAAK;QACL,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mBA0CX,CAAC,CAAC;QACjB,KAAK,EAAE,EAAE;KACV,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,wBAAwB,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7J,MAAM,IAAI,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;IAClE,OAAO,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC9D,CAAC"}

package/dist/tools.d.ts

@@ -0,0 +1,3 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+export declare const shellTool: CallbackTool;
+//# sourceMappingURL=tools.d.ts.map

package/dist/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGnD,eAAO,MAAM,SAAS,cAmBpB,CAAC"}

package/dist/tools.js

@@ -0,0 +1,24 @@
+import { CallbackTool } from "@ai-zen/agents-core";
+import { execSync } from "node:child_process";
+export const shellTool = new CallbackTool({
+    function: {
+        name: "shell",
+        description: "执行 shell 命令并返回输出",
+        parameters: {
+            type: "object",
+            properties: {
+                command: { type: "string", description: "要执行的命令" },
+            },
+            required: ["command"],
+        },
+    },
+    callback(args) {
+        try {
+            return execSync(args.command, { encoding: "utf-8", timeout: 30000 });
+        }
+        catch (e) {
+            return `错误: ${e.message}`;
+        }
+    },
+});
+//# sourceMappingURL=tools.js.map

package/dist/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../src/tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAE9C,MAAM,CAAC,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC;IACxC,QAAQ,EAAE;QACR,IAAI,EAAE,OAAO;QACb,WAAW,EAAE,kBAAkB;QAC/B,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE;aACnD;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF;IACD,QAAQ,CAAO,IAAyB;QACtC,IAAI,CAAC;YACH,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,OAAO,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;CACF,CAAC,CAAC"}

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "@ai-zen/air",
+  "version": "0.1.3",
+  "description": "极简 AI 命令行助手",
+  "type": "module",
+  "main": "./dist/agent-runtime.js",
+  "bin": {
+    "air": "./dist/cli.js"
+  },
+  "dependencies": {
+    "@ai-zen/agents-core": "file:../agents/packages/core",
+    "chalk": "^5.6.2",
+    "commander": "^15.0.0",
+    "inquirer": "^9.3.8"
+  },
+  "devDependencies": {
+    "@types/inquirer": "^9.0.10",
+    "@types/node": "^24.12.2",
+    "typescript": "^5.3.3",
+    "vitest": "^4.1.9"
+  },
+  "scripts": {
+    "build": "tsc",
+    "start": "npm run build && node ./dist/cli.js",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  }
+}

package/src/__tests__/config.test.ts

@@ -0,0 +1,111 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import { existsSync, rmSync, mkdirSync, writeFileSync, readFileSync, readdirSync } from "node:fs";
+import { join } from "node:path";
+import { tmpdir } from "node:os";
+import { readConfig, saveConfig, readMessages, saveMessages, clearMessages, saveSnapshot, listSnapshots, loadSnapshot } from "../config.js";
+
+describe("config", () => {
+  let tempDir: string;
+  let origHome: string | undefined;
+
+  beforeEach(() => {
+    tempDir = join(tmpdir(), `air-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);
+    mkdirSync(tempDir, { recursive: true });
+    origHome = process.env.HOME;
+    process.env.HOME = tempDir;
+  });
+
+  afterEach(() => {
+    if (origHome) process.env.HOME = origHome;
+    else delete process.env.HOME;
+    try { rmSync(tempDir, { recursive: true, force: true }); } catch {}
+  });
+
+  describe("config (apiKey)", () => {
+    it("默认 apiKey 为空", () => {
+      expect(readConfig().apiKey).toBe("");
+    });
+
+    it("保存和读取 apiKey", () => {
+      saveConfig("sk-test-key-12345");
+      expect(readConfig().apiKey).toBe("sk-test-key-12345");
+    });
+
+    it("配置文件损坏时返回空 key", () => {
+      mkdirSync(join(tempDir, ".ai-zen", "air"), { recursive: true });
+      writeFileSync(join(tempDir, ".ai-zen", "air", "config.json"), "不是JSON", "utf-8");
+      expect(readConfig().apiKey).toBe("");
+    });
+  });
+
+  describe("context (messages 数组)", () => {
+    it("不存在时返回空数组", () => {
+      expect(readMessages()).toEqual([]);
+    });
+
+    it("保存和读取 messages", () => {
+      const msgs = [
+        { role: "user", content: "你好" },
+        { role: "assistant", content: "你好！" },
+      ];
+      saveMessages(msgs);
+      expect(readMessages()).toEqual(msgs);
+    });
+
+    it("clearMessages 清空为 []", () => {
+      saveMessages([{ role: "user", content: "hi" }]);
+      clearMessages();
+      expect(readMessages()).toEqual([]);
+    });
+
+    it("文件内容不是数组时返回空数组", () => {
+      mkdirSync(join(tempDir, ".ai-zen", "air"), { recursive: true });
+      writeFileSync(join(tempDir, ".ai-zen", "air", "context.json"), '"不是数组"', "utf-8");
+      expect(readMessages()).toEqual([]);
+    });
+
+    it("文件损坏时返回空数组", () => {
+      mkdirSync(join(tempDir, ".ai-zen", "air"), { recursive: true });
+      writeFileSync(join(tempDir, ".ai-zen", "air", "context.json"), "不是JSON", "utf-8");
+      expect(readMessages()).toEqual([]);
+    });
+  });
+
+  describe("snapshots", () => {
+    it("保存快照，文件名是 ISO 时间", () => {
+      const msgs = [{ role: "user", content: "测试快照" }];
+      const name = saveSnapshot(msgs);
+      expect(name).toBeTruthy();
+
+      const snapDir = join(tempDir, ".ai-zen", "air", "snapshots");
+      expect(existsSync(snapDir)).toBe(true);
+      const files = readdirSync(snapDir).filter((f) => f.endsWith(".json"));
+      expect(files.length).toBe(1);
+      expect(files[0]).toBe(name + ".json");
+
+      const loaded = JSON.parse(readFileSync(join(snapDir, files[0]), "utf-8"));
+      expect(loaded).toEqual(msgs);
+    });
+  });
+
+
+  describe("listSnapshots / loadSnapshot", () => {
+    it("加载不存在的快照返回空数组", () => {
+      expect(loadSnapshot("nonexistent")).toEqual([]);
+    });
+
+    it("保存后能列出并加载快照", () => {
+      const msgs = [{ role: "user", content: "测试" }];
+      const name = saveSnapshot(msgs);
+
+      const list = listSnapshots();
+      expect(list.length).toBe(1);
+      expect(list[0].name).toBe(name);
+      expect(list[0].date).toBeTruthy();
+
+      const loaded = loadSnapshot(name);
+      expect(loaded).toEqual(msgs);
+    });
+  });
+
+});

package/src/__tests__/main.test.ts

@@ -0,0 +1,43 @@
+import { describe, it, expect } from "vitest";
+import { contextSize, shouldMigrate } from "../migration.js";
+
+describe("contextSize", () => {
+  it("空数组", () => {
+    expect(contextSize([])).toBe(2);
+  });
+
+  it("一条消息", () => {
+    const msgs = [{ role: "user", content: "你好" }];
+    expect(contextSize(msgs)).toBe(JSON.stringify(msgs).length);
+  });
+
+  it("多条消息", () => {
+    const msgs = [
+      { role: "user", content: "你好" },
+      { role: "assistant", content: "你好！有什么可以帮你的吗？" },
+    ];
+    expect(contextSize(msgs)).toBe(JSON.stringify(msgs).length);
+  });
+});
+
+describe("shouldMigrate", () => {
+  it("空消息不迁移", () => {
+    expect(shouldMigrate([])).toBe(false);
+  });
+
+  it("小消息不迁移", () => {
+    expect(shouldMigrate([{ role: "user", content: "hi" }])).toBe(false);
+  });
+
+  it("达到 66 万时迁移", () => {
+    const n = 500000 - JSON.stringify([{ role: "user", content: "" }]).length;
+    const bigMsg = { role: "user", content: "x".repeat(n) };
+    expect(shouldMigrate([bigMsg])).toBe(true);
+  });
+
+  it("差一点不到时不迁移", () => {
+    const n = 500000 - JSON.stringify([{ role: "user", content: "" }]).length - 1;
+    const bigMsg = { role: "user", content: "x".repeat(n) };
+    expect(shouldMigrate([bigMsg])).toBe(false);
+  });
+});

package/src/__tests__/tools.test.ts

@@ -0,0 +1,15 @@
+import { describe, it, expect } from "vitest";
+import { shellTool } from "../tools.js";
+
+describe("shellTool", () => {
+  it("具有正确的名称和描述", () => {
+    expect(shellTool.function.name).toBe("shell");
+    expect(shellTool.function.description).toBe("执行 shell 命令并返回输出");
+  });
+
+  it("参数定义正确", () => {
+    const params = shellTool.function.parameters;
+    expect(params.properties.command.type).toBe("string");
+    expect(params.required).toEqual(["command"]);
+  });
+});

package/src/agent-factory.ts

@@ -0,0 +1,23 @@
+import { Agent, Message, OpenAI, ChatGPT } from "@ai-zen/agents-core";
+import type { AgentNS } from "@ai-zen/agents-core";
+import { readConfig } from "./config.js";
+import { shellTool } from "./tools.js";
+
+const MODEL_NAME = "deepseek-v4-flash";
+const API_ENDPOINT = "https://api.deepseek.com/v1";
+
+async function buildModel(apiKey: string) {
+  const endpoint = new OpenAI({ openai_endpoint: API_ENDPOINT, api_key: apiKey });
+  return new ChatGPT({
+    model_config: {},
+    request_config: await endpoint.chatCompletion(MODEL_NAME),
+  });
+}
+
+export async function buildAgent(savedMessages: any[]): Promise<Agent> {
+  const config = readConfig();
+  const model = await buildModel(config.apiKey);
+  const messages: AgentNS.Message[] = [];
+  for (const m of savedMessages) messages.push(new Message(m));
+  return new Agent({ model, messages, tools: [shellTool] });
+}