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

refactor(agents): drop pi subscribe fallback changes from this pr

Browse Source
This commit is contained in:
Ayaan Zaidi
2026-02-19 20:29:29 +05:30
parent d29776c0d6
commit 735644e002
2 changed files with 2 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/agents/pi-embedded-subscribe.handlers.messages.ts
View File

@@ -165,10 +165,9 @@ export function handleMessageUpdate(
}
}
const streamedReasoning = extractThinkingFromTaggedStream(ctx.state.deltaBuffer);
if (ctx.state.streamReasoning) {
// Handle partial <think> tags: stream whatever reasoning is visible so far.
ctx.emitReasoningStream(streamedReasoning);
ctx.emitReasoningStream(extractThinkingFromTaggedStream(ctx.state.deltaBuffer));
}
const next = ctx
@@ -277,9 +276,7 @@ export function handleMessageEnd(
});
const rawThinking =
ctx.state.includeReasoning || ctx.state.streamReasoning
? extractAssistantThinking(assistantMessage) ||
extractThinkingFromTaggedStream(ctx.state.deltaBuffer) ||
extractThinkingFromTaggedText(rawText)
? extractAssistantThinking(assistantMessage) || extractThinkingFromTaggedText(rawText)
: "";
const formattedReasoning = rawThinking ? formatReasoningMessage(rawThinking) : "";
const trimmedText = text.trim();

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

@@ -68,36 +68,4 @@ describe("subscribeEmbeddedPiSession", () => {
]);
},
);
it.each(THINKING_TAG_CASES)(
"falls back to streamed <%s> reasoning when message_end has no thinking block",
({ open, close }) => {
const { emit, onBlockReply } = createReasoningBlockReplyHarness();
emit({ type: "message_start", message: { role: "assistant" } });
emit({
type: "message_update",
message: { role: "assistant" },
assistantMessageEvent: { type: "text_delta", delta: `${open}Because` },
});
emit({
type: "message_update",
message: { role: "assistant" },
assistantMessageEvent: {
type: "text_delta",
delta: ` it helps${close}\n\nFinal answer`,
},
});
emit({
type: "message_end",
message: {
role: "assistant",
content: [{ type: "text", text: "Final answer" }],
} as AssistantMessage,
});
expectReasoningAndAnswerCalls(onBlockReply);
},
);
});