refactor: extract execution prompt assembly

2026-04-02 22:32:19 +08:00
parent e348d6cc5d
commit 384dd570de
7 changed files with 415 additions and 157 deletions
--- a/tests/execution-memory-resolver.test.ts
+++ b/tests/execution-memory-resolver.test.ts
@@ -0,0 +1,138 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+import {
+  resolveRelevantMemoriesForTesting,
+  resolveRuntimeRelevantMemoriesForTesting,
+} from "@/lib/execution/memory-resolver";
+
+test("MemoryResolver 在 master-agent 会话下优先挑当前请求命中的项目记忆", () => {
+  const resolved = resolveRelevantMemoriesForTesting({
+    projectId: "master-agent",
+    requestText: "boss-console 的审批流",
+    memories: [
+      {
+        memoryId: "m1",
+        scope: "project",
+        projectId: "boss-console",
+        title: "审批流",
+        content: "boss-console approval",
+        tags: ["approval"],
+        memoryType: "project_progress",
+        createdAt: "2026-01-01T00:00:00.000Z",
+        updatedAt: "2026-01-01T00:00:00.000Z",
+      },
+      {
+        memoryId: "m2",
+        scope: "project",
+        projectId: "wenshenapp",
+        title: "UI",
+        content: "wechat ui",
+        tags: ["ui"],
+        memoryType: "project_progress",
+        createdAt: "2026-01-02T00:00:00.000Z",
+        updatedAt: "2026-01-02T00:00:00.000Z",
+      },
+    ],
+  });
+
+  assert.equal(resolved.projectMemories.length, 1);
+  assert.equal(resolved.projectMemories[0]?.projectId, "boss-console");
+});
+
+test("MemoryResolver 会保留全局记忆的输入顺序并只截断到 8 条", () => {
+  const resolved = resolveRelevantMemoriesForTesting({
+    projectId: "master-agent",
+    memories: [
+      {
+        memoryId: "g1",
+        scope: "global",
+        title: "一",
+        content: "one",
+        tags: [],
+        memoryType: "decision",
+        createdAt: "2026-01-01T00:00:00.000Z",
+        updatedAt: "2026-01-01T00:00:00.000Z",
+      },
+      {
+        memoryId: "g2",
+        scope: "global",
+        title: "二",
+        content: "two",
+        tags: [],
+        memoryType: "decision",
+        createdAt: "2026-01-02T00:00:00.000Z",
+        updatedAt: "2026-01-02T00:00:00.000Z",
+      },
+    ],
+  });
+
+  assert.deepEqual(
+    resolved.userMemories.map((memory) => memory.memoryId),
+    ["g1", "g2"],
+  );
+});
+
+test("Runtime MemoryResolver 会优先排布 workflow_rule 和 user_preference 全局记忆", () => {
+  const resolved = resolveRuntimeRelevantMemoriesForTesting({
+    projectId: "master-agent",
+    memories: [
+      {
+        memoryId: "g1",
+        scope: "global",
+        title: "普通记忆",
+        content: "normal",
+        tags: [],
+        memoryType: "decision",
+        createdAt: "2026-01-01T00:00:00.000Z",
+        updatedAt: "2026-01-01T00:00:00.000Z",
+      },
+      {
+        memoryId: "g2",
+        scope: "global",
+        title: "规则记忆",
+        content: "workflow",
+        tags: [],
+        memoryType: "workflow_rule",
+        createdAt: "2026-01-03T00:00:00.000Z",
+        updatedAt: "2026-01-03T00:00:00.000Z",
+      },
+      {
+        memoryId: "g3",
+        scope: "global",
+        title: "偏好记忆",
+        content: "preference",
+        tags: [],
+        memoryType: "user_preference",
+        createdAt: "2026-01-02T00:00:00.000Z",
+        updatedAt: "2026-01-02T00:00:00.000Z",
+      },
+    ],
+  });
+
+  assert.deepEqual(
+    resolved.userMemories.map((memory) => memory.memoryId),
+    ["g2", "g3", "g1"],
+  );
+});
+
+test("Runtime MemoryResolver 在 master-agent 非空请求但无 lexical 命中时回退到前 6 个项目记忆", () => {
+  const resolved = resolveRuntimeRelevantMemoriesForTesting({
+    projectId: "master-agent",
+    requestText: "xyzxyz no overlap",
+    memories: [
+      { memoryId: "p1", scope: "project", projectId: "boss-console", title: "alpha", content: "alpha-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-01T00:00:00.000Z", updatedAt: "2026-01-01T00:00:00.000Z" },
+      { memoryId: "p2", scope: "project", projectId: "boss-console", title: "bravo", content: "bravo-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-02T00:00:00.000Z", updatedAt: "2026-01-02T00:00:00.000Z" },
+      { memoryId: "p3", scope: "project", projectId: "boss-console", title: "charlie", content: "charlie-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-03T00:00:00.000Z", updatedAt: "2026-01-03T00:00:00.000Z" },
+      { memoryId: "p4", scope: "project", projectId: "boss-console", title: "delta", content: "delta-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-04T00:00:00.000Z", updatedAt: "2026-01-04T00:00:00.000Z" },
+      { memoryId: "p5", scope: "project", projectId: "boss-console", title: "echo", content: "echo-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-05T00:00:00.000Z", updatedAt: "2026-01-05T00:00:00.000Z" },
+      { memoryId: "p6", scope: "project", projectId: "boss-console", title: "foxtrot", content: "foxtrot-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-06T00:00:00.000Z", updatedAt: "2026-01-06T00:00:00.000Z" },
+      { memoryId: "p7", scope: "project", projectId: "boss-console", title: "golf", content: "golf-content", tags: [], memoryType: "project_progress", createdAt: "2026-01-07T00:00:00.000Z", updatedAt: "2026-01-07T00:00:00.000Z" },
+    ],
+  });
+
+  assert.equal(resolved.projectMemories.length, 6);
+  assert.deepEqual(
+    resolved.projectMemories.map((memory) => memory.memoryId),
+    ["p1", "p2", "p3", "p4", "p5", "p6"],
+  );
+});