vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-06-03 21:44:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: share block reply coalescer enqueue

Browse Source
This commit is contained in:
Vincent Koc
2026-05-30 13:38:50 +02:00
parent 3e8d06a6be
commit cd37dbd4e5
1 changed files with 28 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/auto-reply/reply/block-reply-pipeline.ts
View File

@@ -235,6 +235,32 @@ export function createBlockReplyPipeline(params: {
bufferedPayloadKeys.clear();
};
const enqueueCoalescedPayload = (payload: ReplyPayload) => {
if (!coalescer) {
return;
}
const assistantMessageIndex = getReplyPayloadMetadata(payload)?.assistantMessageIndex;
if (
assistantMessageIndex !== undefined &&
bufferedAssistantMessageIndex !== undefined &&
assistantMessageIndex !== bufferedAssistantMessageIndex &&
coalescer.hasBuffered()
) {
// Logical assistant blocks must not be merged together by the generic
// coalescer. Force-flush the previous buffered block before starting a
// new assistant-message block.
flushBufferedAssistantBlock();
}
const payloadKey = createBlockReplyPayloadKey(payload);
if (hasSeenOrQueuedPayloadKey(payloadKey) || bufferedKeys.has(payloadKey)) {
return;
}
seenKeys.add(payloadKey);
bufferedKeys.add(payloadKey);
bufferedAssistantMessageIndex = assistantMessageIndex;
coalescer.enqueue(payload);
};
const enqueue = (payload: ReplyPayload) => {
if (aborted) {
return;
@@ -248,23 +274,7 @@ export function createBlockReplyPipeline(params: {
{ trimText: true },
);
if (reply.hasMedia && coalescer && !hasNonTextContent) {
const assistantMessageIndex = getReplyPayloadMetadata(payload)?.assistantMessageIndex;
if (
assistantMessageIndex !== undefined &&
bufferedAssistantMessageIndex !== undefined &&
assistantMessageIndex !== bufferedAssistantMessageIndex &&
coalescer.hasBuffered()
) {
flushBufferedAssistantBlock();
}
const payloadKey = createBlockReplyPayloadKey(payload);
if (hasSeenOrQueuedPayloadKey(payloadKey) || bufferedKeys.has(payloadKey)) {
return;
}
seenKeys.add(payloadKey);
bufferedKeys.add(payloadKey);
bufferedAssistantMessageIndex = assistantMessageIndex;
coalescer.enqueue(payload);
enqueueCoalescedPayload(payload);
return;
}
if (reply.hasMedia || hasNonTextContent) {
@@ -273,26 +283,7 @@ export function createBlockReplyPipeline(params: {
return;
}
if (coalescer) {
const assistantMessageIndex = getReplyPayloadMetadata(payload)?.assistantMessageIndex;
if (
assistantMessageIndex !== undefined &&
bufferedAssistantMessageIndex !== undefined &&
assistantMessageIndex !== bufferedAssistantMessageIndex &&
coalescer.hasBuffered()
) {
// Logical assistant blocks must not be merged together by the generic
// coalescer. Force-flush the previous buffered block before starting a
// new assistant-message block.
flushBufferedAssistantBlock();
}
const payloadKey = createBlockReplyPayloadKey(payload);
if (hasSeenOrQueuedPayloadKey(payloadKey) || bufferedKeys.has(payloadKey)) {
return;
}
seenKeys.add(payloadKey);
bufferedKeys.add(payloadKey);
bufferedAssistantMessageIndex = assistantMessageIndex;
coalescer.enqueue(payload);
enqueueCoalescedPayload(payload);
return;
}
sendPayload(payload, /* bypassSeenCheck */ false);