feat: add image generation tool and WebUI mode

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-05-08 20:06:23 +08:00
parent 3a2f47d720
commit e936ed48bd
45 changed files with 2979 additions and 89 deletions
@@ -134,6 +134,89 @@ describe("useNanobotStream", () => {
    ]);
  });

+  it("suppresses redundant stream confirmation after assistant media", () => {
+    const fake = fakeClient();
+    const { result } = renderHook(() => useNanobotStream("chat-img-result", EMPTY_MESSAGES), {
+      wrapper: wrap(fake.client),
+    });
+
+    act(() => {
+      fake.emit("chat-img-result", {
+        event: "message",
+        chat_id: "chat-img-result",
+        text: "image ready",
+        media_urls: [{ url: "/api/media/sig/image", name: "generated.png" }],
+      });
+      fake.emit("chat-img-result", {
+        event: "message",
+        chat_id: "chat-img-result",
+        text: "message()",
+        kind: "tool_hint",
+      });
+      fake.emit("chat-img-result", {
+        event: "delta",
+        chat_id: "chat-img-result",
+        text: "发送成功",
+      });
+      fake.emit("chat-img-result", {
+        event: "stream_end",
+        chat_id: "chat-img-result",
+      });
+      fake.emit("chat-img-result", {
+        event: "turn_end",
+        chat_id: "chat-img-result",
+      });
+    });
+
+    expect(result.current.messages).toHaveLength(1);
+    expect(result.current.messages[0].content).toBe("image ready");
+    expect(result.current.messages[0].media).toHaveLength(1);
+  });
+
+  it("passes image generation options to the websocket client", () => {
+    const fake = fakeClient();
+    const { result } = renderHook(() => useNanobotStream("chat-img", EMPTY_MESSAGES), {
+      wrapper: wrap(fake.client),
+    });
+
+    act(() => {
+      result.current.send(
+        "draw a square icon",
+        undefined,
+        { imageGeneration: { enabled: true, aspect_ratio: "1:1" } },
+      );
+    });
+
+    expect(fake.client.sendMessage).toHaveBeenCalledWith(
+      "chat-img",
+      "draw a square icon",
+      undefined,
+      { imageGeneration: { enabled: true, aspect_ratio: "1:1" } },
+    );
+  });
+
+  it("stops the active turn without adding a user slash command bubble", () => {
+    const fake = fakeClient();
+    const { result } = renderHook(() => useNanobotStream("chat-stop", EMPTY_MESSAGES), {
+      wrapper: wrap(fake.client),
+    });
+
+    act(() => {
+      result.current.send("long task");
+    });
+    expect(result.current.messages).toHaveLength(1);
+    expect(result.current.isStreaming).toBe(true);
+
+    act(() => {
+      result.current.stop();
+    });
+
+    expect(fake.client.sendMessage).toHaveBeenLastCalledWith("chat-stop", "/stop");
+    expect(result.current.isStreaming).toBe(false);
+    expect(result.current.messages).toHaveLength(1);
+    expect(result.current.messages[0].content).toBe("long task");
+  });
+
  it("keeps assistant buttons on complete messages", () => {
    const fake = fakeClient();
    const { result } = renderHook(() => useNanobotStream("chat-q", EMPTY_MESSAGES), {