fix: unify chat process and reply presentation

src/features/agent/presentation/message-process-groups.test.ts

@@ -0,0 +1,83 @@
+import { describe, expect, it } from "vitest";
+import type { InteractionRequest } from "../api/interaction-requests";
+import type { AgentChatMessage } from "../types";
+import { groupMessageProcesses } from "./message-process-groups";
+
+const assistant = (id: string, content = ""): AgentChatMessage => ({
+  id, role: "assistant", content,
+  progress: [{ id: "query", phase: "tool", status: "completed", title: id }]
+});
+const user = (id: string): AgentChatMessage => ({ id, role: "user", content: id });
+const request = (source: string, reply: string, extra: Partial<InteractionRequest> = {}): InteractionRequest => ({
+  version: "agent-interaction-request@1", interaction_id: `question-${source}`, message_id: source,
+  turn_id: `turn-${source}`, call_id: `call-${source}`, status: "answered", revision: 1, draft_revision: 0,
+  questions: [{ question_id: "scope", title: "范围", body: "核查范围？", mode: "single", required: true, allow_custom: true, options: [] }],
+  draft: [], created_at: "2026-10-28T10:00:00Z", delivery: { native_message_id: reply, revision: 1, status: "delivered" },
+  ...extra
+});
+
+describe("process groups across clarification replies", () => {
+  const source = assistant("question");
+  const reply = user("reply");
+  const result = assistant("answer_reply", "结果正文");
+  const messages = [user("original"), source, reply, result];
+
+  it("moves the source process to the related continuation without editing stored messages or progress ids", () => {
+    const original = structuredClone(messages);
+    const groups = groupMessageProcesses(messages, [request(source.id, reply.id)]);
+    expect(groups.get(source.id)).toEqual([]);
+    expect(groups.get(result.id)).toEqual([source, result]);
+    expect(groups.get(result.id)?.map((message) => message.progress?.[0].id)).toEqual(["query", "query"]);
+    expect(messages).toEqual(original);
+  });
+
+  it("groups an existing result even when delivery is blocked and there is no final_answer", () => {
+    const record = request(source.id, reply.id, { delivery: { native_message_id: reply.id, revision: 2, status: "blocked", error_code: "CONTINUATION_UNCONFIRMED" } });
+    expect(groupMessageProcesses(messages, [record]).get(result.id)).toEqual([source, result]);
+    expect(record.delivery?.status).toBe("blocked");
+  });
+
+  it("keeps processes in place while waiting, cancelled, missing a source page or missing a continuation", () => {
+    for (const extra of [{ status: "pending" }, { status: "cancelled" }, { delivery: undefined },
+      { delivery: { native_message_id: reply.id, revision: 2, status: "cancelled" } }] satisfies Partial<InteractionRequest>[]) {
+      expect(groupMessageProcesses(messages, [request(source.id, reply.id, extra)]).get(source.id)).toEqual([source]);
+    }
+    expect(groupMessageProcesses(messages.slice(0, -1), [request(source.id, reply.id)]).get(source.id)).toEqual([source]);
+    expect(groupMessageProcesses(messages.slice(2), [request(source.id, reply.id)]).get(result.id)).toEqual([result]);
+  });
+
+  it("does not merge across an unrelated user request on either side of the saved reply", () => {
+    for (const list of [
+      [source, user("other-request"), reply, result],
+      [source, reply, user("other-request"), result],
+      [reply, source, result]
+    ]) {
+      const groups = groupMessageProcesses(list, [request(source.id, reply.id)]);
+      expect(groups.get(source.id)).toEqual([source]);
+      expect(groups.get(result.id)).toEqual([result]);
+    }
+  });
+
+  it("combines successive clarifications in order and ignores duplicate records", () => {
+    const reply2 = user("reply2");
+    const final = assistant("answer_reply2", "最终结果");
+    const first = request(source.id, reply.id);
+    const second = request(result.id, reply2.id);
+    const groups = groupMessageProcesses([...messages, reply2, final], [second, first, first]);
+    expect(groups.get(source.id)).toEqual([]);
+    expect(groups.get(result.id)).toEqual([]);
+    expect(groups.get(final.id)).toEqual([source, result, final]);
+  });
+
+  it("retains actionable questions and permissions at their original message", () => {
+    const pendingQuestion: AgentChatMessage = { ...source, questions: [{
+      requestId: "pending", sessionId: "s", questions: [], createdAt: 0, status: "checking"
+    }] };
+    const pendingPermission: AgentChatMessage = { ...source, permissions: [{
+      requestId: "pending", sessionId: "s", permission: "simulate", patterns: [], always: [], createdAt: 0, status: "pending"
+    }] };
+    for (const pending of [pendingQuestion, pendingPermission]) {
+      expect(groupMessageProcesses([pending, reply, result], [request(source.id, reply.id)]).get(source.id)).toEqual([pending]);
+    }
+  });
+});