@ai-zen/air 0.2.0 → 0.2.2

package/README.md

@@ -85,20 +85,35 @@ $ gred "hello" file.txt
 ```
 src/
 ├── cli.ts              # CLI entry, commander
-├── config.ts           # Config read/write
-├── delta-renderer.ts   # Stream renderer (from agents project)
+├── config.ts           # Config, context, snapshot read/write
+├── delta-renderer.ts   # Stream renderer
+├── hook.ts             # Fallback terminal hook (install/uninstall)
+├── migration.ts        # Context counting & migration
 ├── tools.ts            # Tool definitions — shell
 ├── agent-factory.ts    # Agent factory — build model & agent
-├── migration.ts        # Context counting & migration
-├── agent-runtime.ts    # Runtime — conversation loop & commands
-├── hook.ts             # Fallback terminal hook (install/uninstall)
+├── chat/
+│   ├── shared.ts       # ChatCtx type & SYSTEM_PROMPT
+│   ├── runtime.ts      # runChat() & chatLoop() — core runtime
+│   ├── message.ts      # handleMessage() — send & migrate
+│   ├── print.ts        # sendAndPrint() — stream output
+│   └── commands/
+│       ├── index.ts    # dispatchCommand() — command router
+│       ├── back.ts     # /back — recall & resend
+│       ├── editor.ts   # /editor — multi-line input
+│       ├── exit.ts     # /exit — quit
+│       ├── help.ts     # /help
+│       ├── load.ts     # /load — load snapshot
+│       ├── new.ts      # /new — new session
+│       └── save.ts     # /save — save snapshot
 └── __tests__/
+    ├── chat.test.ts    # Chat session tests
     ├── config.test.ts  # Config/context/snapshot tests
     ├── main.test.ts    # contextSize/shouldMigrate tests
+    ├── e2e.test.ts     # End-to-end tests
     └── tools.test.ts   # Shell tool structure tests
 ```
 
-~46 KB, 870 lines.
+~49 KB, 881 lines.
 
 ## Tests
 

package/README.zh.md

@@ -52,6 +52,16 @@ air hook uninstall
 | `/editor` | 打开系统编辑器输入多行文本 |
 | `/help` | 帮助 |
 
+### 兜底终端钩子
+
+安装后，任何 shell 中不存在的命令都会自动转发到 `air`，AI 会解读你的意图并帮助你。
+
+```bash
+$ gred "hello" file.txt
+# → command not found → 自动转发到 air
+# → AI: "你是不是想用 grep？"
+```
+
 ## 设计
 
 ```
@@ -75,20 +85,35 @@ air hook uninstall
 ```
 src/
 ├── cli.ts              # CLI 入口，commander
-├── config.ts           # 配置与文件读写
-├── delta-renderer.ts   # 流式渲染器（复用自 agents 项目）
+├── config.ts           # 配置、上下文、快照读写
+├── delta-renderer.ts   # 流式渲染器
+├── hook.ts             # 兜底终端钩子（install/uninstall）
+├── migration.ts        # 上下文计数与迁移
 ├── tools.ts            # 工具定义——shell
 ├── agent-factory.ts    # Agent 工厂——构建模型与 Agent
-├── migration.ts        # 上下文计数与任务迁移
-├── agent-runtime.ts    # 运行时——对话循环与命令处理
-├── hook.ts             # 兜底终端钩子（install/uninstall）
+├── chat/
+│   ├── shared.ts       # ChatCtx 类型 & SYSTEM_PROMPT
+│   ├── runtime.ts      # runChat() & chatLoop() — 核心运行时
+│   ├── message.ts      # handleMessage() — 发送与迁移
+│   ├── print.ts        # sendAndPrint() — 流式输出
+│   └── commands/
+│       ├── index.ts    # dispatchCommand() — 命令分发入口
+│       ├── back.ts     # /back — 撤回消息
+│       ├── editor.ts   # /editor — 多行编辑器输入
+│       ├── exit.ts     # /exit — 退出
+│       ├── help.ts     # /help
+│       ├── load.ts     # /load — 加载快照
+│       ├── new.ts      # /new — 新会话
+│       └── save.ts     # /save — 保存快照
 └── __tests__/
+    ├── chat.test.ts    # 聊天测试
     ├── config.test.ts  # 配置/上下文/快照测试
     ├── main.test.ts    # contextSize/shouldMigrate 测试
+    ├── e2e.test.ts     # 端到端测试
     └── tools.test.ts   # shell 工具结构测试
 ```
 
-共 46 KB，870 行。
+共 49 KB，881 行。
 
 ## 测试
 

package/dist/agent-factory.js

@@ -2,7 +2,7 @@ import { Agent, Message, OpenAI, ChatGPT } 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";
+const API_ENDPOINT = process.env.AIR_API_ENDPOINT || "https://api.deepseek.com/v1";
 async function buildModel(apiKey) {
     const endpoint = new OpenAI({ openai_endpoint: API_ENDPOINT, api_key: apiKey });
     return new ChatGPT({

package/dist/agent-factory.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-factory.js","sourceRoot":"","sources":["../src/agent-factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEtE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,UAAU,GAAG,mBAAmB,CAAC;AACvC,MAAM,YAAY,GAAG,6BAA6B,CAAC;AAEnD,KAAK,UAAU,UAAU,CAAC,MAAc;IACtC,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,EAAE,eAAe,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;IAChF,OAAO,IAAI,OAAO,CAAC;QACjB,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,MAAM,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC;KAC1D,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,aAAoB;IACnD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAsB,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,IAAI,aAAa;QAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,OAAO,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;AAC5D,CAAC"}
+{"version":3,"file":"agent-factory.js","sourceRoot":"","sources":["../src/agent-factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEtE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,UAAU,GAAG,mBAAmB,CAAC;AACvC,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,6BAA6B,CAAC;AAEnF,KAAK,UAAU,UAAU,CAAC,MAAc;IACtC,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,EAAE,eAAe,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;IAChF,OAAO,IAAI,OAAO,CAAC;QACjB,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,MAAM,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC;KAC1D,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,aAAoB;IACnD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAsB,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,IAAI,aAAa;QAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,OAAO,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;AAC5D,CAAC"}

package/dist/agent-runtime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-runtime.d.ts","sourceRoot":"","sources":["../src/agent-runtime.ts"],"names":[],"mappings":"AAwDA,wBAAsB,eAAe,CAAC,cAAc,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAgN5E"}
+{"version":3,"file":"agent-runtime.d.ts","sourceRoot":"","sources":["../src/agent-runtime.ts"],"names":[],"mappings":"AA0RA,wBAAsB,eAAe,CAAC,cAAc,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAyC5E"}

package/dist/agent-runtime.js

@@ -2,7 +2,7 @@ import chalk from "chalk";
 import { Message } from "@ai-zen/agents-core";
 import { DeltaRenderer } from "./delta-renderer.js";
 import inquirer from "inquirer";
-import { readConfig, readMessages, saveMessages, clearMessages, saveSnapshot, listSnapshots, loadSnapshot } from "./config.js";
+import { readConfig, readMessages, saveMessages, 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 = [
@@ -20,8 +20,9 @@ const SYSTEM_PROMPT = [
     "- 项目记忆: $(cwd)/.ai-zen/air/temp/*.md",
     "下次启动时用 shell 读取即可。这是你唯一的记忆方式。",
 ].join("\n");
-// 流式发送
+// ==================== 流式发送 ====================
 async function sendAndPrint(agent, text) {
+    console.log(chalk.green.bold("\n🤖 AI:"));
     const renderer = new DeltaRenderer({
         reasoningHeader: "\n\n💭 思考中...\n",
         contentHeader: "\n\n💭 回答中...\n",
@@ -41,261 +42,273 @@ async function sendAndPrint(agent, text) {
     process.stdout.write("\n\n");
     console.log();
 }
-async function sendAndSave(agent, text) {
-    console.log(chalk.green.bold("\n🤖 AI:"));
-    await sendAndPrint(agent, text);
+// ==================== 公共操作 ====================
+async function saveAndNew(ctx) {
+    const msgs = [Message.System(SYSTEM_PROMPT)];
+    saveMessages(msgs);
+    ctx.agent = await buildAgent(msgs);
 }
-// 对话循环
-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);
-    }
-    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) {
-            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() {
+// ==================== 对话循环 ====================
+async function ask(ctx) {
+    while (true) {
         const { input } = await inquirer.prompt([
             { type: "input", name: "input", message: "你:" },
         ]);
         const t = input.trim();
-        if (!t) {
-            await ask();
-            return;
-        }
+        if (!t)
+            continue;
         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)];
-                }
+            await dispatchCommand(ctx, t.toLowerCase());
+            continue;
+        }
+        await handleMessage(ctx, t);
+    }
+}
+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 handleBack(ctx);
+            break;
+        case "/editor":
+            await cmdEditor(ctx);
+            break;
+        case "/help":
+            cmdHelp();
+            break;
+        default:
+            console.log(`\n❌ 未知命令: ${cmd}\n`);
+            break;
+    }
+}
+async function cmdSave(ctx) {
+    console.log(`\n✅ 快照: ${saveSnapshot(ctx.agent.messages)}\n`);
+}
+async function cmdNew(ctx) {
+    await saveAndNew(ctx);
+    console.log("\n🆕 重新开始\n");
+}
+function cmdHelp() {
+    console.log("\n/exit /quit  退出\n/save        保存快照\n/load        加载快照\n/new         重新开始\n/back        撤回消息\n/help        帮助\n");
+}
+async function cmdExit(ctx) {
+    console.log("\n👋 再见！");
+    process.exit(0);
+}
+async function cmdLoad(ctx) {
+    const snapshots = listSnapshots();
+    if (snapshots.length === 0) {
+        console.log("\n📭 没有可用的快照\n");
+        ctx.agent = await buildAgent([]);
+        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");
+        ctx.agent = await buildAgent([]);
+        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);
+}
+async function cmdEditor(ctx) {
+    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) {
+        console.error(`\n❌ ${err.message}`);
+    }
+}
+async function handleMessage(ctx, text) {
+    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);
-                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) {
-                    console.error(`\n❌ ${err.message}`);
-                }
-                await ask();
-                return;
+                console.log("✅ 迁移完成\n");
+                ctx.agent = await buildAgent(msgs);
             }
-            if (c === "/help") {
-                console.log("\n/exit /quit  退出\n/save        保存快照\n/load        加载快照\n/new         重新开始\n/back        撤回消息\n/help        帮助\n");
-                await ask();
-                return;
+            catch (err) {
+                console.error(`❌ 迁移失败: ${err.message}\n`);
             }
-            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) {
-                    console.error(`❌ 迁移失败: ${err.message}\n`);
-                }
-            }
-        }
-        catch (err) {
-            console.error(`\n❌ ${err.message}`);
-        }
-        await ask();
     }
-    async function handleBack() {
-        const targets = [];
-        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 ? "..." : "") });
-                }
+    catch (err) {
+        console.error(`\n❌ ${err.message}`);
+    }
+}
+async function handleBack(ctx) {
+    const targets = [];
+    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 ? "..." : "") });
             }
         }
-        if (targets.length === 0) {
-            console.log("\n❌ 还没有消息可以撤回\n");
-            return;
+        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 ? "..." : "") });
+            }
         }
-        console.log("\n📋 选择要撤回到哪条消息（将删除所选及其之后的所有内容）：\n");
-        const { selectedIndex } = await inquirer.prompt([
+    }
+    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: "selectedIndex",
-                message: "撤回到:",
-                pageSize: 15,
+                name: "editChoice",
+                message: "请选择:",
                 choices: [
-                    { name: "↩️  取消操作", value: -1 },
-                    ...targets.map((t) => ({ name: t.label + " " + t.preview, value: t.index })),
+                    { name: "\u270f\ufe0f  修改后重新发送", value: "edit" },
+                    { name: "\u{1F504} 直接重新发送", value: "resend" },
+                    { name: "\u21a9\ufe0f  取消操作", value: "cancel" },
                 ],
             },
         ]);
-        if (selectedIndex === -1) {
+        if (editChoice === "cancel") {
             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" },
-                    ],
-                },
+        let textToSend = originalText;
+        if (editChoice === "edit") {
+            const { editedContent } = await inquirer.prompt([
+                { type: "input", name: "editedContent", message: "修改消息:", default: originalText },
             ]);
-            if (editChoice === "cancel") {
-                console.log("\n已取消\n");
+            if (!editedContent.trim()) {
+                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);
+            textToSend = editedContent.trim();
         }
-        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);
+        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);
     }
-    await ask();
+}
+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);
     process.on("SIGINT", () => { console.log("\n\n👋 再见！"); process.exit(0); });
 }
 //# sourceMappingURL=agent-runtime.js.map