@downcity/agent 1.1.408 → 1.1.437

package/scripts/session-attachments.test.mjs

@@ -1,104 +0,0 @@
-/**
- * @file 验证 Session prompt 的 Data URL 附件会先落盘，再以路径持久化。
- */
-
-import assert from "node:assert/strict";
-import fs from "node:fs/promises";
-import os from "node:os";
-import path from "node:path";
-import test from "node:test";
-import { LocalFileSystem } from "@downcity/workspace";
-import { LocalSessionDataStore } from "../bin/workspace/store/LocalSessionDataStore.js";
-import { SessionMessages } from "../bin/session/SessionMessages.js";
-
-async function create_harness() {
-  const root_path = await fs.mkdtemp(path.join(os.tmpdir(), "downcity-session-attachment-"));
-  const files = new LocalFileSystem(root_path);
-  const store = new LocalSessionDataStore({
-    files,
-    storage_root_path: root_path,
-    agent_id: "attachment-test-agent",
-    session_id: "attachment-test-session",
-  });
-  const messages = new SessionMessages({
-    session_id: "attachment-test-session",
-    store: store.messages,
-    attachment_store: store.attachments,
-    publish: () => {},
-  });
-  await messages.initialize();
-  return { root_path, store, messages };
-}
-
-test("Data URL 落盘后保留用户文件名并保存相对路径", async () => {
-  const harness = await create_harness();
-  const result = await harness.messages.append_prompt_message({
-    turn_id: "turn-1",
-    input_type: "prompt",
-    prompt: {
-      query: [{
-        type: "file",
-        media_type: "image/png",
-        url: "data:image/png;base64,aGVsbG8=",
-        filename: "diagram.png",
-      }],
-    },
-    project_root: harness.root_path,
-  });
-
-  const part = result.parts[0];
-  assert.equal(part.type, "file");
-  assert.equal(path.isAbsolute(part.url), true);
-  assert.match(part.url, /sessions\/attachment-test-session\/attachments\/att_/);
-  assert.equal(part.url.endsWith(".png"), true);
-  assert.equal(part.filename, "diagram.png");
-  assert.equal(
-    (await fs.readFile(part.url)).toString(),
-    "hello",
-  );
-  assert.equal(part.url.includes("base64"), false);
-});
-
-test("视频 Data URL 使用 MIME 类型和扩展名落盘", async () => {
-  const harness = await create_harness();
-  const result = await harness.messages.append_prompt_message({
-    turn_id: "turn-1",
-    input_type: "prompt",
-    prompt: {
-      query: [{
-        type: "file",
-        media_type: "video/mp4",
-        url: "data:video/mp4;base64,AQID",
-        filename: "demo.mp4",
-      }],
-    },
-    project_root: harness.root_path,
-  });
-
-  const part = result.parts[0];
-  assert.equal(part.url.endsWith(".mp4"), true);
-  assert.deepEqual(
-    [...await fs.readFile(part.url)],
-    [1, 2, 3],
-  );
-});
-
-test("无效 Data URL 不会写入 Session Message", async () => {
-  const harness = await create_harness();
-  await assert.rejects(
-    harness.messages.append_prompt_message({
-      turn_id: "turn-1",
-      input_type: "prompt",
-      prompt: {
-        query: [{
-          type: "file",
-          media_type: "image/png",
-          url: "data:image/png;base64",
-        }],
-      },
-      project_root: harness.root_path,
-    }),
-    /Invalid data URL/,
-  );
-  assert.deepEqual(await harness.store.messages.list_messages(), []);
-});

package/scripts/session-compact-handle.test.mjs

@@ -1,118 +0,0 @@
-/**
- * @file 验证 session.compact() 的队列顺序与 Handle 完成语义。
- *
- * 关键点（中文）
- * - `compact()` 只返回已入队 Handle，不在空闲期主动创建 Turn。
- * - 后续 Prompt 必须等待排在它前面的 Compact Command 真正完成后才启动。
- * - `result` 与 `finished` 使用同一个稳定领域结果。
- */
-
-import assert from "node:assert/strict";
-import fs from "node:fs/promises";
-import os from "node:os";
-import path from "node:path";
-import test from "node:test";
-import { MockModelClient } from "./ModelClientMock.mjs";
-
-import {
-  Agent,
-  DefaultSessionComposer,
-  Session,
-} from "../bin/index.js";
-import { create_workspace_entry } from "../bin/internal/index.js";
-import { Workspace } from "@downcity/workspace";
-
-function create_deferred() {
-  let resolve;
-  const promise = new Promise((inner_resolve) => {
-    resolve = inner_resolve;
-  });
-  return { promise, resolve };
-}
-
-function create_stream_text_result(text) {
-  return {
-    stream: new ReadableStream({
-      start(controller) {
-        controller.enqueue({ type: "stream-start", warnings: [] });
-        controller.enqueue({ type: "text-start", id: "text_1" });
-        controller.enqueue({ type: "text-delta", id: "text_1", delta: text });
-        controller.enqueue({ type: "text-end", id: "text_1" });
-        controller.enqueue({
-          type: "finish",
-          finishReason: { unified: "stop", raw: "stop" },
-          usage: {
-            inputTokens: { total: 0, noCache: 0, cacheRead: 0, cacheWrite: 0 },
-            outputTokens: { total: 0, text: 0, reasoning: 0 },
-          },
-        });
-        controller.close();
-      },
-    }),
-  };
-}
-
-test("compact Handle 在队列命令完成后兑现并阻塞后续 Prompt", async () => {
-  const compact_started = create_deferred();
-  const release_compact = create_deferred();
-
-  class CompactComposer extends DefaultSessionComposer {
-    async compact() {
-      compact_started.resolve();
-      await release_compact.promise;
-      return null;
-    }
-  }
-
-  class CompactSession extends Session {
-    constructor(options) {
-      super({ ...options, composer: new CompactComposer() });
-    }
-  }
-
-  const project_root = await fs.mkdtemp(
-    path.join(os.tmpdir(), "downcity-session-compact-handle-"),
-  );
-  const agent = new Agent({
-    id: "compact_handle_agent",
-    model: new MockModelClient({
-      modelId: "compact-handle-model",
-      doStream: async () => create_stream_text_result("done"),
-    }),
-    session_class: CompactSession,
-  });
-  const entry = create_workspace_entry(agent, new Workspace({ id: "test_workspace", path: project_root, data_root_path: path.join(project_root, "data") }));
-
-  try {
-    const session = await entry.sessions.create({
-      session_id: "compact_handle_session",
-    });
-    const compact_handle = await session.compact();
-    assert.equal(compact_handle.result, null);
-
-    let prompt_returned = false;
-    const prompt_promise = session.prompt({ query: "continue" }).then((handle) => {
-      prompt_returned = true;
-      return handle;
-    });
-    await compact_started.promise;
-    assert.equal(prompt_returned, false);
-    assert.equal(compact_handle.result, null);
-
-    release_compact.resolve();
-    const turn_handle = await prompt_promise;
-    const compact_result = await compact_handle.finished;
-    assert.deepEqual(compact_result, {
-      compact_id: compact_handle.id,
-      success: true,
-      compacted: false,
-      reason: "nothing_to_compact",
-    });
-    assert.deepEqual(compact_handle.result, compact_result);
-    assert.equal((await turn_handle.finished).success, true);
-  } finally {
-    release_compact.resolve();
-    await agent.dispose();
-    await fs.rm(project_root, { recursive: true, force: true });
-  }
-});

package/scripts/session-compaction-prompts.test.mjs

@@ -1,49 +0,0 @@
-/**
- * @file 验证 Session compact prompt 的结构化摘要格式。
- *
- * 关键点（中文）
- * - compact prompt 必须保持结构化摘要格式稳定。
- * - 迭代更新 prompt 需要携带 previous-summary 标签。
- */
-
-import test from "node:test";
-import assert from "node:assert/strict";
-
-import {
-  SESSION_COMPACTION_SYSTEM_PROMPT,
-  build_initial_session_compaction_prompt,
-  build_turn_prefix_session_compaction_prompt,
-  build_update_session_compaction_prompt,
-} from "../bin/executor/composer/compaction/jsonl/JsonlSessionCompactionPrompts.js";
-
-test("session compaction prompts preserve structured formats", () => {
-  assert.match(
-    SESSION_COMPACTION_SYSTEM_PROMPT,
-    /Do NOT continue the conversation/,
-  );
-
-  const initial_prompt = build_initial_session_compaction_prompt({
-    conversation_text: "user: 请修改 packages/agent/src/foo.ts",
-  });
-  assert.match(initial_prompt, /## Goal/);
-  assert.match(initial_prompt, /## Constraints & Preferences/);
-  assert.match(initial_prompt, /## Critical Context/);
-  assert.match(
-    initial_prompt,
-    /Keep each section concise\. Preserve exact file paths/,
-  );
-
-  const update_prompt = build_update_session_compaction_prompt({
-    previous_summary: "## Goal\n旧目标",
-    new_conversation_text: "assistant: 已完成一部分",
-  });
-  assert.match(update_prompt, /<previous-summary>\n## Goal\n旧目标\n<\/previous-summary>/);
-  assert.match(update_prompt, /PRESERVE all existing information/);
-  assert.match(update_prompt, /## Next Steps/);
-
-  const prefix_prompt = build_turn_prefix_session_compaction_prompt({
-    prefix_text: "user: 这个 turn 太长",
-  });
-  assert.match(prefix_prompt, /## Original Request/);
-  assert.match(prefix_prompt, /## Context for Suffix/);
-});

package/scripts/session-composer.test.mjs

@@ -1,208 +0,0 @@
-/**
- * @file 验证统一 SessionComposer 只读取快照并返回模型输入或压缩计划。
- */
-
-import assert from "node:assert/strict";
-import test from "node:test";
-import fs from "node:fs/promises";
-import os from "node:os";
-import path from "node:path";
-import { MockModelClient } from "./ModelClientMock.mjs";
-import {
-  Agent,
-  DefaultSessionComposer,
-  Session,
-} from "../bin/index.js";
-import { create_workspace_entry } from "../bin/internal/index.js";
-import { Workspace } from "@downcity/workspace";
-
-function create_input(model) {
-  return {
-    session: {
-      agent_id: "composer-agent",
-      session_id: "composer-session",
-      project_root: "/tmp/composer-project",
-      created_at: 1,
-      timezone: "UTC",
-    },
-    state: {
-      model,
-      env: {},
-      systems: ["Base instruction"],
-      tools: {},
-      instruction_system_blocks: [{
-        source: "instruction",
-        name: "agent",
-        content: "Base instruction",
-      }],
-      managed_plugin_system_blocks: [],
-      plugin_system_blocks: [],
-    },
-    history: {
-      summary: null,
-      messages: [{
-        message_id: "user-1",
-        session_id: "composer-session",
-        turn_id: "turn-1",
-        sequence: 1,
-        revision: 1,
-        visibility: "visible",
-        created_at: 1,
-        updated_at: 1,
-        type: "user",
-        input_type: "prompt",
-        parts: [{
-          part_id: "text-1",
-          type: "text",
-          text: "hello",
-          state: "done",
-        }],
-      }],
-    },
-    turn: { turn_id: "turn-1", retry_count: 0 },
-  };
-}
-
-test("DefaultSessionComposer 从 canonical 快照组装 Step 输入", async () => {
-  const model = new MockModelClient({ modelId: "composer-model" });
-  const input = create_input(model);
-  const step = await new DefaultSessionComposer().compose(input);
-
-  assert.equal(step.messages.length, 1);
-  assert.equal(step.messages[0].content[0].text, "hello");
-  assert.equal(step.system_blocks[0].content, "Base instruction");
-  assert.match(step.system.at(-1).content, /composer-session/);
-});
-
-test("Custom Composer 可以覆盖组装结果而不接触持久化", async () => {
-  class CustomComposer extends DefaultSessionComposer {
-    async compose(input) {
-      const step = await super.compose(input);
-      return {
-        ...step,
-        system: [
-          ...step.system,
-          { role: "system", content: "Custom behavior" },
-        ],
-      };
-    }
-  }
-
-  const model = new MockModelClient({ modelId: "custom-composer-model" });
-  const input = create_input(model);
-  const before = structuredClone(input.history);
-  const step = await new CustomComposer().compose(input);
-
-  assert.equal(step.system.at(-1).content, "Custom behavior");
-  assert.deepEqual(input.history, before);
-});
-
-test("Session system 快照与 Custom Composer 的实际模型输入一致", async () => {
-  class CustomComposer extends DefaultSessionComposer {
-    async compose(input) {
-      const step = await super.compose(input);
-      return {
-        ...step,
-        system: [
-          ...step.system,
-          { role: "system", content: "Custom snapshot behavior" },
-        ],
-      };
-    }
-  }
-
-  class CustomSession extends Session {
-    constructor(options) {
-      super({ ...options, composer: new CustomComposer() });
-    }
-  }
-
-  const project_root = await fs.mkdtemp(
-    path.join(os.tmpdir(), "downcity-session-composer-system-"),
-  );
-  const agent = new Agent({
-    id: "custom_composer_agent",
-    model: new MockModelClient({ modelId: "custom-composer-model" }),
-    session_class: CustomSession,
-  });
-  const entry = create_workspace_entry(agent, new Workspace({ id: "test_workspace", path: project_root, data_root_path: path.join(project_root, "data") }));
-  try {
-    const session = await entry.sessions.create({
-      session_id: "custom_composer_session",
-    });
-    const snapshot = await session.system();
-    assert.equal(snapshot.blocks.at(-1).content, "Custom snapshot behavior");
-    assert.equal(snapshot.blocks.at(-1).source, "session");
-  } finally {
-    await agent.dispose();
-    await fs.rm(project_root, { recursive: true, force: true });
-  }
-});
-
-test("Composer compact 只返回计划，不修改 Message 快照", async () => {
-  const prompts = [];
-  const model = new MockModelClient({
-    modelId: "composer-compact-model",
-    doGenerate: async (options) => {
-      prompts.push(JSON.stringify(options.prompt));
-      return {
-        content: [{ type: "text", text: "Composer summary" }],
-        finishReason: { unified: "stop", raw: "stop" },
-        usage: {
-          inputTokens: { total: 0, noCache: 0, cacheRead: 0, cacheWrite: 0 },
-          outputTokens: { total: 0, text: 0, reasoning: 0 },
-        },
-        warnings: [],
-      };
-    },
-  });
-  const input = create_input(model);
-  for (let sequence = 2; sequence <= 3; sequence += 1) {
-    input.history.messages.push({
-      ...structuredClone(input.history.messages[0]),
-      message_id: `user-${String(sequence)}`,
-      sequence,
-      parts: [{
-        part_id: `text-${String(sequence)}`,
-        type: "text",
-        text: `message ${String(sequence)}`,
-        state: "done",
-      }],
-    });
-  }
-  const before = structuredClone(input.history);
-  const plan = await new DefaultSessionComposer().compact({
-    session: input.session,
-    model: input.state.model,
-    history: input.history,
-  });
-
-  assert.equal(plan.through_sequence, 1);
-  assert.equal(plan.summary.text, "Composer summary");
-  assert.equal(prompts.length, 1);
-  assert.match(prompts[0], /hello/);
-  assert.doesNotMatch(prompts[0], /message 2/);
-  assert.doesNotMatch(prompts[0], /message 3/);
-  assert.deepEqual(input.history, before);
-});
-
-test("Composer compact 在 Active 少于两条上下文消息时不调用模型", async () => {
-  let generation_count = 0;
-  const model = new MockModelClient({
-    modelId: "composer-no-compact-model",
-    doGenerate: async () => {
-      generation_count += 1;
-      throw new Error("summary should not run");
-    },
-  });
-  const input = create_input(model);
-
-  const plan = await new DefaultSessionComposer().compact({
-    session: input.session,
-    model: input.state.model,
-    history: input.history,
-  });
-
-  assert.equal(plan, null);
-  assert.equal(generation_count, 0);
-});