@downcity/agent 1.1.338 → 1.1.400

package/scripts/session-list-title.test.mjs

@@ -2,8 +2,8 @@
  * @file 验证 session 列表从正确目录读取持久化 title。
  *
  * 关键点（中文）
- * - 普通列表只读取 `.downcity/agents/<agent_id>/sessions` 下的 meta。
- * - 归档列表只读取 `.downcity/agents/<agent_id>/archived-sessions` 下的 meta。
+ * - 普通列表只读取 Workspace `sessions` 下当前 Agent 的 meta。
+ * - 归档列表只读取 Workspace `archived-sessions` 下当前 Agent 的 meta。
  * - title 允许为空；这里仅验证已由模型生成并落盘的 title 能被列表返回。
  */
 
@@ -13,11 +13,13 @@ import os from "node:os";
 import path from "node:path";
 import fs from "node:fs/promises";
 
-import { MockLanguageModelV3 } from "ai/test";
-import { Agent, Workspace } from "../bin/index.js";
+import { MockModelClient } from "./ModelClientMock.mjs";
+import { Agent } from "../bin/index.js";
+import { create_workspace_entry } from "../bin/internal/index.js";
+import { Workspace } from "@downcity/workspace";
 
 function create_mock_title_model(title_text) {
-  return new MockLanguageModelV3({
+  return new MockModelClient({
     modelId: "mock-session-list-title-model",
     doStream: async () => ({
       stream: new ReadableStream({
@@ -60,10 +62,10 @@ async function create_agent_with_titled_session(input) {
   );
   const agent = new Agent({
     id: input.agent_id,
-    workspace: new Workspace({ path: agent_path }),
     model: create_mock_title_model(input.title),
   });
-  const collection = agent.sessions;
+  const entry = create_workspace_entry(agent, new Workspace({ id: "test_workspace", path: agent_path, data_root_path: path.join(agent_path, "data") }));
+  const collection = entry.sessions;
   const session = await collection.create({
     session_id: input.session_id,
   });
@@ -81,13 +83,14 @@ async function create_agent_with_titled_session(input) {
 
   return {
     agent,
+    entry,
     collection,
     session,
   };
 }
 
 test("list_sessions returns persisted title from active session metadata", async () => {
-  const { agent, collection, session } = await create_agent_with_titled_session({
+  const { agent, entry, collection, session } = await create_agent_with_titled_session({
     tmp_prefix: "downcity-agent-session-list-title-",
     agent_id: "list_title_agent",
     session_id: "active_session",
@@ -97,38 +100,7 @@ test("list_sessions returns persisted title from active session metadata", async
 
   try {
     const page = await collection.list();
-    const meta = JSON.parse(
-      await fs.readFile(
-        path.join(
-          agent.workspace.path,
-          ".downcity",
-          "agents",
-          encodeURIComponent(agent.id),
-          "sessions",
-          encodeURIComponent(session.id),
-          "messages",
-          "meta.json",
-        ),
-        "utf8",
-      ),
-    );
-
     assert.equal(page.total, 1);
-    assert.equal(meta.message_count, 1);
-    assert.equal(meta.preview_text, "Need the session list to show the generated title");
-    const history_stat = await fs.stat(
-      path.join(
-        agent.workspace.path,
-        ".downcity",
-        "agents",
-        encodeURIComponent(agent.id),
-        "sessions",
-        encodeURIComponent(session.id),
-        "messages",
-        "active.jsonl",
-      ),
-    );
-    assert.equal(meta.historyBytes, history_stat.size);
     assert.deepEqual(
       page.items.map((item) => ({
         session_id: item.session_id,
@@ -149,7 +121,7 @@ test("list_sessions returns persisted title from active session metadata", async
 });
 
 test("list_sessions reflects canonical SessionMessages changes", async () => {
-  const { agent, collection, session } = await create_agent_with_titled_session({
+  const { agent, entry, collection, session } = await create_agent_with_titled_session({
     tmp_prefix: "downcity-agent-session-summary-repair-",
     agent_id: "summary_repair_agent",
     session_id: "summary_repair_session",
@@ -157,16 +129,13 @@ test("list_sessions reflects canonical SessionMessages changes", async () => {
     first_user_text: "Initial history",
   });
   const messages_dir = path.join(
-    agent.workspace.path,
-    ".downcity",
-    "agents",
-    encodeURIComponent(agent.id),
+    entry.data_path,
     "sessions",
     encodeURIComponent(session.id),
     "messages",
   );
   const messages_path = path.join(messages_dir, "active.jsonl");
-  const meta_path = path.join(messages_dir, "meta.json");
+  const meta_path = path.join(path.dirname(messages_dir), "meta.json");
 
   try {
     await session.append_assistant_message({
@@ -186,20 +155,9 @@ test("list_sessions reflects canonical SessionMessages changes", async () => {
 
     const page = await collection.list();
     const info = await session.get_info();
-    const repaired_meta = JSON.parse(await fs.readFile(meta_path, "utf8"));
-    const history_stat = await fs.stat(messages_path);
-    const segment_stat = await fs.stat(path.join(
-      messages_dir,
-      "segments",
-      "000000000001-000000000001.jsonl",
-    ));
-
     assert.equal(page.items[0].message_count, 2);
     assert.equal(info.message_count, 2);
     assert.equal(page.items[0].preview_text, "Recorder appended history");
-    assert.equal(repaired_meta.message_count, 2);
-    assert.equal(repaired_meta.preview_text, "Recorder appended history");
-    assert.equal(repaired_meta.historyBytes, history_stat.size + segment_stat.size);
   } finally {
     await agent.dispose();
   }