github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-04-03 03:03:24 -04:00
Code Issues Packages Projects Releases Wiki Activity

test(agents): dedupe subscribe reasoning tag fixtures

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-19 09:11:07 +00:00
parent 749edf25ca
commit 317b7d363d
3 changed files with 20 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/agents/pi-embedded-subscribe.e2e-harness.ts
View File

@@ -7,6 +7,13 @@ type SubscribeEmbeddedPiSessionParams = Parameters<SubscribeEmbeddedPiSession>[0
type PiSession = Parameters<SubscribeEmbeddedPiSession>[0]["session"];
type OnBlockReply = NonNullable<SubscribeEmbeddedPiSessionParams["onBlockReply"]>;
export const THINKING_TAG_CASES = [
{ tag: "think", open: "<think>", close: "</think>" },
{ tag: "thinking", open: "<thinking>", close: "</thinking>" },
{ tag: "thought", open: "<thought>", close: "</thought>" },
{ tag: "antthinking", open: "<antthinking>", close: "</antthinking>" },
] as const;
export function createStubSessionHarness(): {
session: PiSession;
emit: (evt: unknown) => void;

26
src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.emits-reasoning-as-separate-message-enabled.e2e.test.ts
View File

@@ -1,16 +1,12 @@
import type { AssistantMessage } from "@mariozechner/pi-ai";
import { describe, expect, it, vi } from "vitest";
import { createStubSessionHarness } from "./pi-embedded-subscribe.e2e-harness.js";
import {
THINKING_TAG_CASES,
createStubSessionHarness,
} from "./pi-embedded-subscribe.e2e-harness.js";
import { subscribeEmbeddedPiSession } from "./pi-embedded-subscribe.js";
describe("subscribeEmbeddedPiSession", () => {
const THINKING_TAG_CASES = [
{ tag: "think", open: "<think>", close: "</think>" },
{ tag: "thinking", open: "<thinking>", close: "</thinking>" },
{ tag: "thought", open: "<thought>", close: "</thought>" },
{ tag: "antthinking", open: "<antthinking>", close: "</antthinking>" },
] as const;
function createReasoningBlockReplyHarness() {
const { session, emit } = createStubSessionHarness();
const onBlockReply = vi.fn();
@@ -26,6 +22,12 @@ describe("subscribeEmbeddedPiSession", () => {
return { emit, onBlockReply };
}
function expectReasoningAndAnswerCalls(onBlockReply: ReturnType<typeof vi.fn>) {
expect(onBlockReply).toHaveBeenCalledTimes(2);
expect(onBlockReply.mock.calls[0][0].text).toBe("Reasoning:\n_Because it helps_");
expect(onBlockReply.mock.calls[1][0].text).toBe("Final answer");
}
it("emits reasoning as a separate message when enabled", () => {
const { emit, onBlockReply } = createReasoningBlockReplyHarness();
@@ -39,9 +41,7 @@ describe("subscribeEmbeddedPiSession", () => {
emit({ type: "message_end", message: assistantMessage });
expect(onBlockReply).toHaveBeenCalledTimes(2);
expect(onBlockReply.mock.calls[0][0].text).toBe("Reasoning:\n_Because it helps_");
expect(onBlockReply.mock.calls[1][0].text).toBe("Final answer");
expectReasoningAndAnswerCalls(onBlockReply);
});
it.each(THINKING_TAG_CASES)(
"promotes <%s> tags to thinking blocks at write-time",
@@ -60,9 +60,7 @@ describe("subscribeEmbeddedPiSession", () => {
emit({ type: "message_end", message: assistantMessage });
expect(onBlockReply).toHaveBeenCalledTimes(2);
expect(onBlockReply.mock.calls[0][0].text).toBe("Reasoning:\n_Because it helps_");
expect(onBlockReply.mock.calls[1][0].text).toBe("Final answer");
expectReasoningAndAnswerCalls(onBlockReply);
expect(assistantMessage.content).toEqual([
{ type: "thinking", thinking: "Because it helps" },

8
src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.subscribeembeddedpisession.e2e.test.ts
View File

@@ -1,6 +1,7 @@
import type { AssistantMessage } from "@mariozechner/pi-ai";
import { describe, expect, it, vi } from "vitest";
import {
THINKING_TAG_CASES,
createStubSessionHarness,
emitMessageStartAndEndForAssistantText,
expectSingleAgentEventText,
@@ -13,13 +14,6 @@ type StubSession = {
};
describe("subscribeEmbeddedPiSession", () => {
const THINKING_TAG_CASES = [
{ tag: "think", open: "<think>", close: "</think>" },
{ tag: "thinking", open: "<thinking>", close: "</thinking>" },
{ tag: "thought", open: "<thought>", close: "</thought>" },
{ tag: "antthinking", open: "<antthinking>", close: "</antthinking>" },
] as const;
function createAgentEventHarness(options?: { runId?: string; sessionKey?: string }) {
const { session, emit } = createStubSessionHarness();
const onAgentEvent = vi.fn();