mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-05-22 01:22:48 +00:00
82ba63e148
Thinking and Used tools are both auxiliary trace rows, but the thread list was applying the same large gap used between full chat turns. That made alternating Thinking / Used tools sequences look uneven and too airy. Move row spacing from a fixed flex gap to per-row margins: full chat turns keep mt-5, while consecutive auxiliary rows use mt-2. Add coverage for Thinking -> Used tools -> Thinking spacing. Co-authored-by: Cursor <cursoragent@cursor.com>
53 lines
1.4 KiB
TypeScript
53 lines
1.4 KiB
TypeScript
import { render } from "@testing-library/react";
|
|
import { describe, expect, it } from "vitest";
|
|
|
|
import { ThreadMessages } from "@/components/thread/ThreadMessages";
|
|
import type { UIMessage } from "@/lib/types";
|
|
|
|
describe("ThreadMessages", () => {
|
|
it("uses compact spacing between consecutive auxiliary rows", () => {
|
|
const messages: UIMessage[] = [
|
|
{
|
|
id: "r1",
|
|
role: "assistant",
|
|
content: "",
|
|
reasoning: "thinking",
|
|
reasoningStreaming: false,
|
|
isStreaming: true,
|
|
createdAt: Date.now(),
|
|
},
|
|
{
|
|
id: "t1",
|
|
role: "tool",
|
|
kind: "trace",
|
|
content: "search()",
|
|
traces: ["search()"],
|
|
createdAt: Date.now(),
|
|
},
|
|
{
|
|
id: "r2",
|
|
role: "assistant",
|
|
content: "",
|
|
reasoning: "more thinking",
|
|
reasoningStreaming: false,
|
|
isStreaming: true,
|
|
createdAt: Date.now(),
|
|
},
|
|
{
|
|
id: "a1",
|
|
role: "assistant",
|
|
content: "final answer",
|
|
createdAt: Date.now(),
|
|
},
|
|
];
|
|
|
|
const { container } = render(<ThreadMessages messages={messages} />);
|
|
const rows = Array.from(container.firstElementChild?.children ?? []);
|
|
|
|
expect(rows[0]).not.toHaveClass("mt-2", "mt-5");
|
|
expect(rows[1]).toHaveClass("mt-2");
|
|
expect(rows[2]).toHaveClass("mt-2");
|
|
expect(rows[3]).toHaveClass("mt-5");
|
|
});
|
|
});
|