github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-08 03:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(tui): dedupe handlers and formatter test setup

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-22 14:05:51 +00:00
parent 66f814a0af
commit 38752338dc
20 changed files with 430 additions and 477 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
src/tui/tui-stream-assembler.test.ts
View File

@@ -1,6 +1,23 @@
import { describe, expect, it } from "vitest";
import { TuiStreamAssembler } from "./tui-stream-assembler.js";
const STREAM_WITH_TOOL_BLOCKS = {
role: "assistant",
content: [
{ type: "text", text: "Before tool call" },
{ type: "tool_use", name: "search" },
{ type: "text", text: "After tool call" },
],
} as const;
const STREAM_AFTER_TOOL_BLOCKS = {
role: "assistant",
content: [
{ type: "tool_use", name: "search" },
{ type: "text", text: "After tool call" },
],
} as const;
describe("TuiStreamAssembler", () => {
it("keeps thinking before content even when thinking arrives later", () => {
const assembler = new TuiStreamAssembler();
@@ -92,61 +109,19 @@ describe("TuiStreamAssembler", () => {
it("keeps richer streamed text when final payload drops earlier blocks", () => {
const assembler = new TuiStreamAssembler();
assembler.ingestDelta(
"run-5",
{
role: "assistant",
content: [
{ type: "text", text: "Before tool call" },
{ type: "tool_use", name: "search" },
{ type: "text", text: "After tool call" },
],
},
false,
);
assembler.ingestDelta("run-5", STREAM_WITH_TOOL_BLOCKS, false);
const finalText = assembler.finalize(
"run-5",
{
role: "assistant",
content: [
{ type: "tool_use", name: "search" },
{ type: "text", text: "After tool call" },
],
},
false,
);
const finalText = assembler.finalize("run-5", STREAM_AFTER_TOOL_BLOCKS, false);
expect(finalText).toBe("Before tool call\nAfter tool call");
});
it("does not regress streamed text when a delta drops boundary blocks after tool calls", () => {
const assembler = new TuiStreamAssembler();
const first = assembler.ingestDelta(
"run-5-stream",
{
role: "assistant",
content: [
{ type: "text", text: "Before tool call" },
{ type: "tool_use", name: "search" },
{ type: "text", text: "After tool call" },
],
},
false,
);
const first = assembler.ingestDelta("run-5-stream", STREAM_WITH_TOOL_BLOCKS, false);
expect(first).toBe("Before tool call\nAfter tool call");
const second = assembler.ingestDelta(
"run-5-stream",
{
role: "assistant",
content: [
{ type: "tool_use", name: "search" },
{ type: "text", text: "After tool call" },
],
},
false,
);
const second = assembler.ingestDelta("run-5-stream", STREAM_AFTER_TOOL_BLOCKS, false);
expect(second).toBeNull();
});