@ai-zen/air 0.2.0 → 0.2.1

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/__tests__/session.test.ts

@@ -0,0 +1,147 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import type { SessionCtx } from "../session/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("../session/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[] = []): SessionCtx {
+  return { agent: { messages: msgs, events: { on: vi.fn(), off: vi.fn() }, send: vi.fn() } as any };
+}
+
+// ==================== shared.ts ====================
+
+describe("saveAndNew", () => {
+  it("写入 system prompt 并构建 agent", async () => {
+    const { saveAndNew } = await import("../session/shared.js");
+    const ctx = makeCtx();
+
+    await saveAndNew(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 { runConversation } = await import("../session/runtime.js");
+    // dispatchCommand is internal, tested via routing only
+  });
+
+  it('"/save" 调用 cmdSave', async () => {
+    const { cmdSave } = await import("../session/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("../session/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("../session/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("../session/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("../session/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("../session/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("../session/commands/editor.js");
+    const ctx = makeCtx();
+    mockPrompt.mockResolvedValueOnce({ content: "" });
+    await cmdEditor(ctx);
+    expect(saveMessages).not.toHaveBeenCalled();
+  });
+});

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/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 { runConversation } from "./session/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 runConversation(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/session/commands/back.ts

@@ -0,0 +1,90 @@
+import inquirer from "inquirer";
+import type { SessionCtx } from "../shared.js";
+import { sendAndPrint } from "../print.js";
+import { saveMessages } from "../../config.js";
+import { buildAgent } from "../../agent-factory.js";
+
+export async function cmdBack(ctx: SessionCtx): 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/session/commands/editor.ts

@@ -0,0 +1,18 @@
+import inquirer from "inquirer";
+import type { SessionCtx } from "../shared.js";
+import { sendAndPrint } from "../print.js";
+import { saveMessages } from "../../config.js";
+
+export async function cmdEditor(ctx: SessionCtx) {
+  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/session/commands/exit.ts

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

package/src/session/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");
+}

package/src/session/commands/load.ts

@@ -0,0 +1,38 @@
+import { Message } from "@ai-zen/agents-core";
+import inquirer from "inquirer";
+import type { SessionCtx } 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: SessionCtx): 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/session/commands/new.ts

@@ -0,0 +1,7 @@
+import type { SessionCtx } from "../shared.js";
+import { saveAndNew } from "../shared.js";
+
+export async function cmdNew(ctx: SessionCtx): Promise<void> {
+  await saveAndNew(ctx);
+  console.log("\n🆕 重新开始\n");
+}

package/src/session/commands/save.ts

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

package/src/session/message.ts

@@ -0,0 +1,27 @@
+import { Message } from "@ai-zen/agents-core";
+import type { SessionCtx } 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: SessionCtx, 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/session/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/session/runtime.ts

@@ -0,0 +1,47 @@
+import { Agent, 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 { SessionCtx, SYSTEM_PROMPT, saveAndNew, ask } from "./shared.js";
+import { sendAndPrint } from "./print.js";
+
+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);
+  }
+  const agent: Agent = await buildAgent(msgs);
+  const ctx: SessionCtx = { 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: any) { 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);
+}