feat(agent): add controlled task execution

2026-09-24 04:19:02 +08:00
parent d018bcae67
commit b31033e9ce
48 changed files with 14498 additions and 86 deletions
@@ -0,0 +1,50 @@
+import { describe, expect, it } from "bun:test";
+import type { UIMessage } from "ai";
+
+import {
+  buildFailedTaskExecutionPlan,
+  toCollaborationMessages,
+} from "../../src/routes/chatStreamRoute.js";
+
+describe("task collaboration history", () => {
+  it("persists only user and assistant text instead of execution trajectory", () => {
+    const messages = [
+      { id: "user-1", role: "user", parts: [{ type: "text", text: "为什么这样判断？" }] },
+      {
+        id: "assistant-1",
+        role: "assistant",
+        parts: [
+          { type: "data-progress", data: { title: "查询 SCADA" } },
+          { type: "text", text: "因为上游数据正常，而投诉集中在同一小区。" },
+          { type: "data-tool_call", data: { tool: "query_operational_task_data" } },
+        ],
+      },
+    ] as unknown as UIMessage[];
+
+    expect(toCollaborationMessages(messages)).toEqual([
+      { id: "user-1", role: "user", parts: [{ type: "text", text: "为什么这样判断？" }] },
+      { id: "assistant-1", role: "assistant", parts: [{ type: "text", text: "因为上游数据正常，而投诉集中在同一小区。" }] },
+    ]);
+  });
+});
+
+describe("failed task execution plan", () => {
+  it("turns active work into an explicit failed task artifact", () => {
+    const plan = buildFailedTaskExecutionPlan([{
+      title: "调查",
+      reason: "开始",
+      nodes: [
+        { id: "READ", title: "读取", status: "completed", owner: "agent", detail: "完成", evidence: ["evidence-1"], action_refs: [] },
+        { id: "ANALYZE", title: "分析", status: "running", owner: "agent", detail: "分析中", evidence: [], action_refs: [] },
+        { id: "SAMPLE", title: "采样", status: "waiting", owner: "human", detail: "等待现场", evidence: [], action_refs: [] },
+      ],
+      edges: [{ from: "READ", to: "ANALYZE", label: "数据已读取" }],
+    }]);
+
+    expect(plan.nodes).toEqual([
+      { id: "READ", title: "读取", status: "completed", owner: "agent", detail: "完成", evidence: ["evidence-1"], action_refs: [] },
+      expect.objectContaining({ id: "ANALYZE", status: "failed" }),
+      expect.objectContaining({ id: "SAMPLE", status: "cancelled" }),
+    ]);
+  });
+});