fix(chat): wire question and todo cards

2026-06-08 18:10:28 +08:00
parent 2691f42581
commit b23cb6acdd
9 changed files with 1713 additions and 10 deletions
@@ -1,7 +1,9 @@
 import {
  abortAgentChat,
  forkAgentChat,
+  rejectAgentQuestion,
  replyAgentPermission,
+  replyAgentQuestion,
  type StreamEvent,
  resumeAgentChatStream,
  streamAgentChat,
@@ -218,6 +220,69 @@ describe("streamAgentChat", () => {
    ]);
  });

+  it("parses question request, response, and todo update events", async () => {
+    apiFetch.mockResolvedValue({
+      ok: true,
+      body: makeStream([
+        'event: question_request\ndata: {"session_id":"s1","request_id":"q-1","questions":[{"header":"范围","question":"选择范围","options":[{"label":"城区","description":"中心城区"}],"multiple":false,"custom":true}],"tool":{"message_id":"m1","call_id":"c1"},"created_at":123}\n\n',
+        'event: question_response\ndata: {"session_id":"s1","request_id":"q-1","answers":[["城区","补充说明"]]}\n\n',
+        'event: todo_update\ndata: {"session_id":"s1","todos":[{"id":"t1","content":"分析水位","status":"in_progress","priority":"high","updated_at":456}],"created_at":456}\n\n',
+      ]),
+    });
+
+    const events: StreamEvent[] = [];
+
+    await streamAgentChat({
+      message: "hi",
+      onEvent: (event) => events.push(event),
+    });
+
+    expect(events).toEqual([
+      {
+        type: "question_request",
+        sessionId: "s1",
+        requestId: "q-1",
+        questions: [
+          {
+            header: "范围",
+            question: "选择范围",
+            options: [{ label: "城区", description: "中心城区" }],
+            multiple: false,
+            custom: true,
+          },
+        ],
+        tool: {
+          messageID: "m1",
+          callID: "c1",
+        },
+        createdAt: 123,
+      },
+      {
+        type: "question_response",
+        sessionId: "s1",
+        requestId: "q-1",
+        answers: [["城区", "补充说明"]],
+        rejected: false,
+      },
+      {
+        type: "todo_update",
+        sessionId: "s1",
+        messageId: undefined,
+        todos: [
+          {
+            id: "t1",
+            content: "分析水位",
+            status: "in_progress",
+            priority: "high",
+            createdAt: undefined,
+            updatedAt: 456,
+          },
+        ],
+        createdAt: 456,
+      },
+    ]);
+  });
+
  it("emits error when response is not ok", async () => {
    apiFetch.mockResolvedValue({
      ok: false,
@@ -314,6 +379,41 @@ describe("streamAgentChat", () => {
    );
  });

+  it("calls question reply and reject endpoints", async () => {
+    apiFetch.mockResolvedValue({
+      ok: true,
+      status: 202,
+      text: async () => "",
+    });
+
+    await replyAgentQuestion("s1", "q-1", [["城区"]]);
+    await rejectAgentQuestion("s1", "q-2");
+
+    expect(apiFetch).toHaveBeenCalledWith(
+      expect.stringContaining("/api/v1/agent/chat/question/q-1/reply"),
+      expect.objectContaining({
+        method: "POST",
+        projectHeaderMode: "include",
+        skipAuthRedirect: true,
+        body: JSON.stringify({
+          session_id: "s1",
+          answers: [["城区"]],
+        }),
+      }),
+    );
+    expect(apiFetch).toHaveBeenCalledWith(
+      expect.stringContaining("/api/v1/agent/chat/question/q-2/reject"),
+      expect.objectContaining({
+        method: "POST",
+        projectHeaderMode: "include",
+        skipAuthRedirect: true,
+        body: JSON.stringify({
+          session_id: "s1",
+        }),
+      }),
+    );
+  });
+
  it("calls fork endpoint and returns new session id", async () => {
    apiFetch.mockResolvedValue({
      ok: true,