fix(chat): handle question and todo state

2026-06-08 18:10:28 +08:00
parent f20847399a
commit 15c3263369
10 changed files with 2173 additions and 724 deletions
@@ -0,0 +1,127 @@
+import { describe, expect, it } from "bun:test";
+
+import {
+  cancelBackendTodos,
+  upsertBackendQuestion,
+} from "../../src/routes/chatUiState.js";
+
+describe("upsertBackendQuestion", () => {
+  it("replaces a tool-call placeholder with the actionable question request", () => {
+    const questions = upsertBackendQuestion(
+      [
+        {
+          requestId: "call-1",
+          sessionId: "session-1",
+          questions: [
+            {
+              header: "测试问题",
+              question: "你觉得这个 question 工具好用吗？",
+              options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
+            },
+          ],
+          tool: { messageID: "message-1", callID: "call-1" },
+          createdAt: 123,
+          status: "pending",
+        },
+      ],
+      {
+        session_id: "session-1",
+        request_id: "question-1",
+        questions: [
+          {
+            header: "测试问题",
+            question: "你觉得这个 question 工具好用吗？",
+            options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
+          },
+        ],
+        tool: { messageID: "message-1", callID: "call-1" },
+        created_at: 456,
+      },
+    );
+
+    expect(questions).toHaveLength(1);
+    expect(questions[0]).toMatchObject({
+      requestId: "question-1",
+      tool: { callID: "call-1" },
+      status: "pending",
+    });
+  });
+
+  it("does not replace an actionable question request with a later tool-call placeholder", () => {
+    const questions = upsertBackendQuestion(
+      [
+        {
+          requestId: "question-1",
+          sessionId: "session-1",
+          questions: [
+            {
+              header: "测试问题",
+              question: "你觉得这个 question 工具好用吗？",
+              options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
+            },
+          ],
+          tool: { messageID: "message-1", callID: "call-1" },
+          createdAt: 123,
+          status: "pending",
+        },
+      ],
+      {
+        session_id: "session-1",
+        request_id: "call-1",
+        questions: [
+          {
+            header: "测试问题",
+            question: "你觉得这个 question 工具好用吗？",
+            options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
+          },
+        ],
+        tool: { messageID: "message-1", callID: "call-1" },
+        created_at: 456,
+      },
+    );
+
+    expect(questions).toHaveLength(1);
+    expect(questions[0]).toMatchObject({
+      requestId: "question-1",
+      tool: { callID: "call-1" },
+      status: "pending",
+    });
+  });
+});
+
+describe("cancelBackendTodos", () => {
+  it("marks pending and in-progress todos as cancelled", () => {
+    const cancelled = cancelBackendTodos([
+      {
+        sessionId: "session-1",
+        todos: [
+          { id: "todo-1", content: "分析水位", status: "in_progress" },
+          { id: "todo-2", content: "生成建议", status: "pending" },
+          { id: "todo-3", content: "完成报告", status: "completed" },
+        ],
+        createdAt: 123,
+      },
+    ]);
+
+    expect(cancelled).toEqual([
+      expect.objectContaining({
+        todos: [
+          expect.objectContaining({
+            id: "todo-1",
+            status: "cancelled",
+            updatedAt: expect.any(Number),
+          }),
+          expect.objectContaining({
+            id: "todo-2",
+            status: "cancelled",
+            updatedAt: expect.any(Number),
+          }),
+          expect.objectContaining({
+            id: "todo-3",
+            status: "completed",
+          }),
+        ],
+      }),
+    ]);
+  });
+});