重构 Agent 聊天，支持分支管理与消息克隆

src/lib/chatStream.test.ts

@@ -1,4 +1,4 @@
-import { streamAgentChat } from "./chatStream";
+import { abortAgentChat, forkAgentChat, streamAgentChat } from "./chatStream";
 import { ReadableStream } from "stream/web";
 import { TextEncoder, TextDecoder } from "util";
 
@@ -147,4 +147,49 @@ describe("streamAgentChat", () => {
       { type: "error", message: "network request failed", detail: "Failed to fetch" },
     ]);
   });
+
+  it("calls abort endpoint for an active session", async () => {
+    apiFetch.mockResolvedValue({
+      ok: true,
+      status: 202,
+      text: async () => "",
+    });
+
+    await abortAgentChat("s1");
+
+    expect(apiFetch).toHaveBeenCalledWith(
+      expect.stringContaining("/api/v1/agent/chat/abort"),
+      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,
+      status: 200,
+      json: async () => ({ session_id: "forked-s1" }),
+      text: async () => "",
+    });
+
+    const sessionId = await forkAgentChat("s1", 3);
+
+    expect(sessionId).toBe("forked-s1");
+    expect(apiFetch).toHaveBeenCalledWith(
+      expect.stringContaining("/api/v1/agent/chat/fork"),
+      expect.objectContaining({
+        method: "POST",
+        body: JSON.stringify({
+          session_id: "s1",
+          keep_message_count: 3,
+        }),
+      }),
+    );
+  });
 });

src/lib/chatStream.ts

@@ -181,3 +181,52 @@ export const streamAgentChat = async ({
     }
   }
 };
+
+export const abortAgentChat = async (sessionId?: string) => {
+  if (!sessionId) {
+    return;
+  }
+
+  const response = await apiFetch(`${config.AGENT_URL}/api/v1/agent/chat/abort`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      session_id: sessionId,
+    }),
+    projectHeaderMode: "include",
+    skipAuthRedirect: true,
+  });
+
+  if (!response.ok) {
+    const detail = await response.text();
+    throw new Error(detail || `abort request failed: ${response.status}`);
+  }
+};
+
+export const forkAgentChat = async (sessionId: string | undefined, keepMessageCount: number) => {
+  const response = await apiFetch(`${config.AGENT_URL}/api/v1/agent/chat/fork`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      session_id: sessionId,
+      keep_message_count: keepMessageCount,
+    }),
+    projectHeaderMode: "include",
+    skipAuthRedirect: true,
+  });
+
+  if (!response.ok) {
+    const detail = await response.text();
+    throw new Error(detail || `fork request failed: ${response.status}`);
+  }
+
+  const payload = (await response.json()) as { session_id?: string };
+  if (!payload.session_id) {
+    throw new Error("fork request returned no session_id");
+  }
+  return payload.session_id;
+};