feat(workbench): redesign investigation reports and diagrams

src/features/workbench/hooks/agent-session-message-state.test.ts

@@ -0,0 +1,46 @@
+import { describe, expect, it } from "vitest";
+import { toAgentChatMessages, toAgentUiMessages } from "./agent-session-message-state";
+
+describe("task context in persisted answers", () => {
+  it("keeps the report reference of each answer when history is restored", () => {
+    const reference = {
+      task_id: "22222222-2222-4222-8222-222222222222",
+      execution_generation: 1,
+      report_revision: 3
+    };
+    const stored = [
+      {
+        id: "answer-1",
+        role: "assistant",
+        parts: [
+          { type: "data-task_context", data: reference },
+          { type: "text", text: "该判断仅覆盖投诉时段的上游记录。" }
+        ]
+      }
+    ];
+    const messages = toAgentChatMessages(toAgentUiMessages(stored), {}, {});
+    expect(messages[0]?.taskContext).toEqual(reference);
+    expect(messages[0]?.content).toBe("该判断仅覆盖投诉时段的上游记录。");
+  });
+  it("keeps the answer readable when a report reference is invalid", () => {
+    const messages = toAgentChatMessages(
+      toAgentUiMessages([
+        {
+          id: "answer-1",
+          role: "assistant",
+          parts: [
+            {
+              type: "data-task_context",
+              data: { task_id: "unknown", execution_generation: -1, report_revision: 0 }
+            },
+            { type: "text", text: "报告引用未记录。" }
+          ]
+        }
+      ]),
+      {},
+      {}
+    );
+    expect(messages[0]?.taskContext).toBeUndefined();
+    expect(messages[0]?.content).toBe("报告引用未记录。");
+  });
+});