@poolzin/pool-bot 2026.2.24 → 2026.2.26

package/dist/telegram/bot-message-dispatch.js

@@ -1,7 +1,6 @@
 import { resolveAgentDir } from "../agents/agent-scope.js";
 import { findModelInCatalog, loadModelCatalog, modelSupportsVision, } from "../agents/model-catalog.js";
 import { resolveDefaultModelForAgent } from "../agents/model-selection.js";
-import { EmbeddedBlockChunker } from "../agents/pi-embedded-block-chunker.js";
 import { resolveChunkMode } from "../auto-reply/chunk.js";
 import { clearHistoryEntriesIfEnabled } from "../auto-reply/reply/history.js";
 import { dispatchReplyWithBufferedBlockDispatcher } from "../auto-reply/reply/provider-dispatcher.js";
@@ -10,11 +9,13 @@ import { logAckFailure, logTypingFailure } from "../channels/logging.js";
 import { createReplyPrefixOptions } from "../channels/reply-prefix.js";
 import { createTypingCallbacks } from "../channels/typing.js";
 import { resolveMarkdownTableMode } from "../config/markdown-tables.js";
+import { loadSessionStore, resolveStorePath } from "../config/sessions.js";
 import { danger, logVerbose } from "../globals.js";
 import { getAgentScopedMediaLocalRoots } from "../media/local-roots.js";
 import { deliverReplies } from "./bot/delivery.js";
-import { resolveTelegramDraftStreamingChunking } from "./draft-chunking.js";
 import { createTelegramDraftStream } from "./draft-stream.js";
+import { renderTelegramHtmlText } from "./format.js";
+import { createTelegramReasoningStepState, splitTelegramReasoningText, } from "./reasoning-lane-coordinator.js";
 import { editMessageTelegram } from "./send.js";
 import { cacheSticker, describeStickerImage } from "./sticker-cache.js";
 const EMPTY_RESPONSE_FALLBACK = "No response generated. Please try again.";
@@ -34,118 +35,163 @@ async function resolveStickerVisionSupport(cfg, agentId) {
         return false;
     }
 }
+function resolveTelegramReasoningLevel(params) {
+    const { cfg, sessionKey, agentId } = params;
+    if (!sessionKey) {
+        return "off";
+    }
+    try {
+        const storePath = resolveStorePath(cfg.session?.store, { agentId });
+        const store = loadSessionStore(storePath, { skipCache: true });
+        const entry = store[sessionKey.toLowerCase()] ?? store[sessionKey];
+        const level = entry?.reasoningLevel;
+        if (level === "on" || level === "stream") {
+            return level;
+        }
+    }
+    catch {
+        // Fall through to default.
+    }
+    return "off";
+}
 export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, replyToMode, streamMode, textLimit, telegramCfg, opts, }) => {
-    const { ctxPayload, msg, chatId, isGroup, threadSpec, historyKey, historyLimit, groupHistories, route, skillFilter, sendTyping, sendRecordVoice, ackReactionPromise, reactionApi, removeAckAfterReply, } = context;
+    const { ctxPayload, msg, chatId, isGroup, threadSpec, historyKey, historyLimit, groupHistories, route, skillFilter, sendTyping, sendRecordVoice, ackReactionPromise, reactionApi, removeAckAfterReply, statusReactionController, } = context;
     const draftMaxChars = Math.min(textLimit, 4096);
+    const tableMode = resolveMarkdownTableMode({
+        cfg,
+        channel: "telegram",
+        accountId: route.accountId,
+    });
+    const renderDraftPreview = (text) => ({
+        text: renderTelegramHtmlText(text, { tableMode }),
+        parseMode: "HTML",
+    });
     const accountBlockStreamingEnabled = typeof telegramCfg.blockStreaming === "boolean"
         ? telegramCfg.blockStreaming
         : cfg.agents?.defaults?.blockStreamingDefault === "on";
-    const canStreamDraft = streamMode !== "off" && !accountBlockStreamingEnabled;
+    const resolvedReasoningLevel = resolveTelegramReasoningLevel({
+        cfg,
+        sessionKey: ctxPayload.SessionKey,
+        agentId: route.agentId,
+    });
+    const forceBlockStreamingForReasoning = resolvedReasoningLevel === "on";
+    const streamReasoningDraft = resolvedReasoningLevel === "stream";
+    const previewStreamingEnabled = streamMode !== "off";
+    const canStreamAnswerDraft = previewStreamingEnabled && !accountBlockStreamingEnabled && !forceBlockStreamingForReasoning;
+    const canStreamReasoningDraft = canStreamAnswerDraft || streamReasoningDraft;
     const draftReplyToMessageId = replyToMode !== "off" && typeof msg.message_id === "number" ? msg.message_id : undefined;
-    const draftStream = canStreamDraft
-        ? createTelegramDraftStream({
-            api: bot.api,
-            chatId,
-            maxChars: draftMaxChars,
-            thread: threadSpec,
-            replyToMessageId: draftReplyToMessageId,
-            minInitialChars: DRAFT_MIN_INITIAL_CHARS,
-            log: logVerbose,
-            warn: logVerbose,
-        })
-        : undefined;
-    const draftChunking = draftStream && streamMode === "block"
-        ? resolveTelegramDraftStreamingChunking(cfg, route.accountId)
-        : undefined;
-    const shouldSplitPreviewMessages = streamMode === "block";
-    const draftChunker = draftChunking ? new EmbeddedBlockChunker(draftChunking) : undefined;
+    const draftMinInitialChars = DRAFT_MIN_INITIAL_CHARS;
     const mediaLocalRoots = getAgentScopedMediaLocalRoots(cfg, route.agentId);
-    let lastPartialText = "";
-    let draftText = "";
-    let hasStreamedMessage = false;
-    const updateDraftFromPartial = (text) => {
-        if (!draftStream || !text) {
-            return;
+    const archivedAnswerPreviews = [];
+    const archivedReasoningPreviewIds = [];
+    const createDraftLane = (laneName, enabled) => {
+        const stream = enabled
+            ? createTelegramDraftStream({
+                api: bot.api,
+                chatId,
+                maxChars: draftMaxChars,
+                thread: threadSpec,
+                replyToMessageId: draftReplyToMessageId,
+                minInitialChars: draftMinInitialChars,
+                renderText: renderDraftPreview,
+                onSupersededPreview: laneName === "answer" || laneName === "reasoning"
+                    ? (preview) => {
+                        if (laneName === "reasoning") {
+                            if (!archivedReasoningPreviewIds.includes(preview.messageId)) {
+                                archivedReasoningPreviewIds.push(preview.messageId);
+                            }
+                            return;
+                        }
+                        archivedAnswerPreviews.push({
+                            messageId: preview.messageId,
+                            textSnapshot: preview.textSnapshot,
+                        });
+                    }
+                    : undefined,
+                log: logVerbose,
+                warn: logVerbose,
+            })
+            : undefined;
+        return {
+            stream,
+            lastPartialText: "",
+            hasStreamedMessage: false,
+        };
+    };
+    const lanes = {
+        answer: createDraftLane("answer", canStreamAnswerDraft),
+        reasoning: createDraftLane("reasoning", canStreamReasoningDraft),
+    };
+    const answerLane = lanes.answer;
+    const reasoningLane = lanes.reasoning;
+    let splitReasoningOnNextStream = false;
+    const reasoningStepState = createTelegramReasoningStepState();
+    const splitTextIntoLaneSegments = (text) => {
+        const split = splitTelegramReasoningText(text);
+        const segments = [];
+        if (split.reasoningText) {
+            segments.push({ lane: "reasoning", text: split.reasoningText });
         }
-        if (text === lastPartialText) {
-            return;
+        if (split.answerText) {
+            segments.push({ lane: "answer", text: split.answerText });
         }
-        // Mark that we've received streaming content (for forceNewMessage decision).
-        hasStreamedMessage = true;
-        if (streamMode === "partial") {
-            // Some providers briefly emit a shorter prefix snapshot (for example
-            // "Sure." -> "Sure" -> "Sure."). Keep the longer preview to avoid
-            // visible punctuation flicker.
-            if (lastPartialText &&
-                lastPartialText.startsWith(text) &&
-                text.length < lastPartialText.length) {
-                return;
-            }
-            lastPartialText = text;
-            draftStream.update(text);
+        return segments;
+    };
+    const resetDraftLaneState = (lane) => {
+        lane.lastPartialText = "";
+        lane.hasStreamedMessage = false;
+    };
+    const updateDraftFromPartial = (lane, text) => {
+        const laneStream = lane.stream;
+        if (!laneStream || !text) {
             return;
         }
-        let delta = text;
-        if (text.startsWith(lastPartialText)) {
-            delta = text.slice(lastPartialText.length);
-        }
-        else {
-            // Streaming buffer reset (or non-monotonic stream). Start fresh.
-            draftChunker?.reset();
-            draftText = "";
-        }
-        lastPartialText = text;
-        if (!delta) {
+        if (text === lane.lastPartialText) {
             return;
         }
-        if (!draftChunker) {
-            draftText = text;
-            draftStream.update(draftText);
+        // Mark that we've received streaming content (for forceNewMessage decision).
+        lane.hasStreamedMessage = true;
+        // Some providers briefly emit a shorter prefix snapshot (for example
+        // "Sure." -> "Sure" -> "Sure."). Keep the longer preview to avoid
+        // visible punctuation flicker.
+        if (lane.lastPartialText &&
+            lane.lastPartialText.startsWith(text) &&
+            text.length < lane.lastPartialText.length) {
             return;
         }
-        draftChunker.append(delta);
-        draftChunker.drain({
-            force: false,
-            emit: (chunk) => {
-                draftText += chunk;
-                draftStream.update(draftText);
-            },
-        });
+        lane.lastPartialText = text;
+        laneStream.update(text);
     };
-    const flushDraft = async () => {
-        if (!draftStream) {
-            return;
-        }
-        if (draftChunker?.hasBuffered()) {
-            draftChunker.drain({
-                force: true,
-                emit: (chunk) => {
-                    draftText += chunk;
-                },
-            });
-            draftChunker.reset();
-            if (draftText) {
-                draftStream.update(draftText);
+    const ingestDraftLaneSegments = (text) => {
+        for (const segment of splitTextIntoLaneSegments(text)) {
+            if (segment.lane === "reasoning") {
+                reasoningStepState.noteReasoningHint();
+                reasoningStepState.noteReasoningDelivered();
             }
+            updateDraftFromPartial(lanes[segment.lane], segment.text);
         }
-        await draftStream.flush();
     };
-    const disableBlockStreaming = typeof telegramCfg.blockStreaming === "boolean"
-        ? !telegramCfg.blockStreaming
-        : draftStream || streamMode === "off"
-            ? true
-            : undefined;
+    const flushDraftLane = async (lane) => {
+        if (!lane.stream) {
+            return;
+        }
+        await lane.stream.flush();
+    };
+    const disableBlockStreaming = !previewStreamingEnabled
+        ? true
+        : forceBlockStreamingForReasoning
+            ? false
+            : typeof telegramCfg.blockStreaming === "boolean"
+                ? !telegramCfg.blockStreaming
+                : canStreamAnswerDraft
+                    ? true
+                    : undefined;
     const { onModelSelected, ...prefixOptions } = createReplyPrefixOptions({
         cfg,
         agentId: route.agentId,
         channel: "telegram",
         accountId: route.accountId,
     });
-    const tableMode = resolveMarkdownTableMode({
-        cfg,
-        channel: "telegram",
-        accountId: route.accountId,
-    });
     const chunkMode = resolveChunkMode(cfg, "telegram", route.accountId);
     // Handle uncached stickers: get a dedicated vision description before dispatch
     // This ensures we cache a raw description rather than a conversational response
@@ -205,8 +251,12 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
     const deliveryState = {
         delivered: false,
         skippedNonSilent: 0,
+        failedNonSilent: 0,
+    };
+    const finalizedPreviewByLane = {
+        answer: false,
+        reasoning: false,
     };
-    let finalizedViaPreviewMessage = false;
     const clearGroupHistory = () => {
         if (isGroup && historyKey) {
             clearHistoryEntriesIfEnabled({ historyMap: groupHistories, historyKey, limit: historyLimit });
@@ -226,7 +276,174 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
         linkPreview: telegramCfg.linkPreview,
         replyQuoteText,
     };
+    const getLanePreviewText = (lane) => lane.lastPartialText;
+    const tryUpdatePreviewForLane = async (params) => {
+        const { lane, laneName, text, previewButtons, stopBeforeEdit = false, updateLaneSnapshot = false, skipRegressive, context, previewMessageId: previewMessageIdOverride, previewTextSnapshot, } = params;
+        if (!lane.stream) {
+            return false;
+        }
+        const lanePreviewMessageId = lane.stream.messageId();
+        const hadPreviewMessage = typeof previewMessageIdOverride === "number" || typeof lanePreviewMessageId === "number";
+        if (stopBeforeEdit) {
+            await lane.stream.stop();
+        }
+        const previewMessageId = typeof previewMessageIdOverride === "number"
+            ? previewMessageIdOverride
+            : lane.stream.messageId();
+        if (typeof previewMessageId !== "number") {
+            return false;
+        }
+        const currentPreviewText = previewTextSnapshot ?? getLanePreviewText(lane);
+        const shouldSkipRegressive = Boolean(currentPreviewText) &&
+            currentPreviewText.startsWith(text) &&
+            text.length < currentPreviewText.length &&
+            (skipRegressive === "always" || hadPreviewMessage);
+        if (shouldSkipRegressive) {
+            // Avoid regressive punctuation/wording flicker from occasional shorter finals.
+            deliveryState.delivered = true;
+            return true;
+        }
+        try {
+            await editMessageTelegram(chatId, previewMessageId, text, {
+                api: bot.api,
+                cfg,
+                accountId: route.accountId,
+                linkPreview: telegramCfg.linkPreview,
+                buttons: previewButtons,
+            });
+            if (updateLaneSnapshot) {
+                lane.lastPartialText = text;
+            }
+            deliveryState.delivered = true;
+            return true;
+        }
+        catch (err) {
+            logVerbose(`telegram: ${laneName} preview ${context} edit failed; falling back to standard send (${String(err)})`);
+            return false;
+        }
+    };
+    const applyTextToPayload = (payload, text) => {
+        if (payload.text === text) {
+            return payload;
+        }
+        return { ...payload, text };
+    };
+    const sendPayload = async (payload) => {
+        const result = await deliverReplies({
+            ...deliveryBaseOptions,
+            replies: [payload],
+            onVoiceRecording: sendRecordVoice,
+        });
+        if (result.delivered) {
+            deliveryState.delivered = true;
+        }
+        return result.delivered;
+    };
+    const consumeArchivedAnswerPreviewForFinal = async (params) => {
+        const archivedPreview = archivedAnswerPreviews.shift();
+        if (!archivedPreview) {
+            return undefined;
+        }
+        if (params.canEditViaPreview) {
+            const finalized = await tryUpdatePreviewForLane({
+                lane: params.lane,
+                laneName: "answer",
+                text: params.text,
+                previewButtons: params.previewButtons,
+                stopBeforeEdit: false,
+                skipRegressive: "existingOnly",
+                context: "final",
+                previewMessageId: archivedPreview.messageId,
+                previewTextSnapshot: archivedPreview.textSnapshot,
+            });
+            if (finalized) {
+                return "preview-finalized";
+            }
+        }
+        try {
+            await bot.api.deleteMessage(chatId, archivedPreview.messageId);
+        }
+        catch (err) {
+            logVerbose(`telegram: archived answer preview cleanup failed (${archivedPreview.messageId}): ${String(err)}`);
+        }
+        const delivered = await sendPayload(applyTextToPayload(params.payload, params.text));
+        return delivered ? "sent" : "skipped";
+    };
+    const deliverLaneText = async (params) => {
+        const { laneName, text, payload, infoKind, previewButtons, allowPreviewUpdateForNonFinal = false, } = params;
+        const lane = lanes[laneName];
+        const hasMedia = Boolean(payload.mediaUrl) || (payload.mediaUrls?.length ?? 0) > 0;
+        const canEditViaPreview = !hasMedia && text.length > 0 && text.length <= draftMaxChars && !payload.isError;
+        if (infoKind === "final") {
+            if (laneName === "answer") {
+                const archivedResult = await consumeArchivedAnswerPreviewForFinal({
+                    lane,
+                    text,
+                    payload,
+                    previewButtons,
+                    canEditViaPreview,
+                });
+                if (archivedResult) {
+                    return archivedResult;
+                }
+            }
+            if (canEditViaPreview && !finalizedPreviewByLane[laneName]) {
+                await flushDraftLane(lane);
+                if (laneName === "answer") {
+                    const archivedResultAfterFlush = await consumeArchivedAnswerPreviewForFinal({
+                        lane,
+                        text,
+                        payload,
+                        previewButtons,
+                        canEditViaPreview,
+                    });
+                    if (archivedResultAfterFlush) {
+                        return archivedResultAfterFlush;
+                    }
+                }
+                const finalized = await tryUpdatePreviewForLane({
+                    lane,
+                    laneName,
+                    text,
+                    previewButtons,
+                    stopBeforeEdit: true,
+                    skipRegressive: "existingOnly",
+                    context: "final",
+                });
+                if (finalized) {
+                    finalizedPreviewByLane[laneName] = true;
+                    return "preview-finalized";
+                }
+            }
+            else if (!hasMedia && !payload.isError && text.length > draftMaxChars) {
+                logVerbose(`telegram: preview final too long for edit (${text.length} > ${draftMaxChars}); falling back to standard send`);
+            }
+            await lane.stream?.stop();
+            const delivered = await sendPayload(applyTextToPayload(payload, text));
+            return delivered ? "sent" : "skipped";
+        }
+        if (allowPreviewUpdateForNonFinal && canEditViaPreview) {
+            const updated = await tryUpdatePreviewForLane({
+                lane,
+                laneName,
+                text,
+                previewButtons,
+                stopBeforeEdit: false,
+                updateLaneSnapshot: true,
+                skipRegressive: "always",
+                context: "update",
+            });
+            if (updated) {
+                return "preview-updated";
+            }
+        }
+        const delivered = await sendPayload(applyTextToPayload(payload, text));
+        return delivered ? "sent" : "skipped";
+    };
     let queuedFinal = false;
+    if (statusReactionController) {
+        void statusReactionController.setThinking();
+    }
     try {
         ({ queuedFinal } = await dispatchReplyWithBufferedBlockDispatcher({
             ctx: ctxPayload,
@@ -234,91 +451,74 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
             dispatcherOptions: {
                 ...prefixOptions,
                 deliver: async (payload, info) => {
-                    if (info.kind === "final") {
-                        await flushDraft();
-                        const hasMedia = Boolean(payload.mediaUrl) || (payload.mediaUrls?.length ?? 0) > 0;
-                        const previewMessageId = draftStream?.messageId();
-                        const finalText = payload.text;
-                        const currentPreviewText = streamMode === "block" ? draftText : lastPartialText;
-                        const previewButtons = payload.channelData?.telegram?.buttons;
-                        let draftStoppedForPreviewEdit = false;
-                        // Skip preview edit for error payloads to avoid overwriting previous content
-                        const canFinalizeViaPreviewEdit = !finalizedViaPreviewMessage &&
-                            !hasMedia &&
-                            typeof finalText === "string" &&
-                            finalText.length > 0 &&
-                            typeof previewMessageId === "number" &&
-                            finalText.length <= draftMaxChars &&
-                            !payload.isError;
-                        if (canFinalizeViaPreviewEdit) {
-                            await draftStream?.stop();
-                            draftStoppedForPreviewEdit = true;
-                            if (currentPreviewText &&
-                                currentPreviewText.startsWith(finalText) &&
-                                finalText.length < currentPreviewText.length) {
-                                // Ignore regressive final edits (e.g., "Okay." -> "Ok"), which
-                                // can appear transiently in some provider streams.
-                                return;
-                            }
-                            try {
-                                await editMessageTelegram(chatId, previewMessageId, finalText, {
-                                    api: bot.api,
-                                    cfg,
-                                    accountId: route.accountId,
-                                    linkPreview: telegramCfg.linkPreview,
-                                    buttons: previewButtons,
-                                });
-                                finalizedViaPreviewMessage = true;
-                                deliveryState.delivered = true;
-                                return;
-                            }
-                            catch (err) {
-                                logVerbose(`telegram: preview final edit failed; falling back to standard send (${String(err)})`);
-                            }
+                    const previewButtons = payload.channelData?.telegram?.buttons;
+                    const segments = splitTextIntoLaneSegments(payload.text);
+                    const hasMedia = Boolean(payload.mediaUrl) || (payload.mediaUrls?.length ?? 0) > 0;
+                    const flushBufferedFinalAnswer = async () => {
+                        const buffered = reasoningStepState.takeBufferedFinalAnswer();
+                        if (!buffered) {
+                            return;
                         }
-                        if (!hasMedia &&
-                            !payload.isError &&
-                            typeof finalText === "string" &&
-                            finalText.length > draftMaxChars) {
-                            logVerbose(`telegram: preview final too long for edit (${finalText.length} > ${draftMaxChars}); falling back to standard send`);
+                        const bufferedButtons = buffered.payload.channelData?.telegram?.buttons;
+                        await deliverLaneText({
+                            laneName: "answer",
+                            text: buffered.text,
+                            payload: buffered.payload,
+                            infoKind: "final",
+                            previewButtons: bufferedButtons,
+                        });
+                        reasoningStepState.resetForNextStep();
+                    };
+                    for (const segment of segments) {
+                        if (segment.lane === "answer" &&
+                            info.kind === "final" &&
+                            reasoningStepState.shouldBufferFinalAnswer()) {
+                            reasoningStepState.bufferFinalAnswer({ payload, text: segment.text });
+                            continue;
                         }
-                        if (!draftStoppedForPreviewEdit) {
-                            await draftStream?.stop();
+                        if (segment.lane === "reasoning") {
+                            reasoningStepState.noteReasoningHint();
                         }
-                        // Check if stop() sent a message (debounce released on isFinal)
-                        // If so, edit that message instead of sending a new one
-                        const messageIdAfterStop = draftStream?.messageId();
-                        if (!finalizedViaPreviewMessage &&
-                            typeof messageIdAfterStop === "number" &&
-                            typeof finalText === "string" &&
-                            finalText.length > 0 &&
-                            finalText.length <= draftMaxChars &&
-                            !hasMedia &&
-                            !payload.isError) {
-                            try {
-                                await editMessageTelegram(chatId, messageIdAfterStop, finalText, {
-                                    api: bot.api,
-                                    cfg,
-                                    accountId: route.accountId,
-                                    linkPreview: telegramCfg.linkPreview,
-                                    buttons: previewButtons,
-                                });
-                                finalizedViaPreviewMessage = true;
-                                deliveryState.delivered = true;
-                                return;
+                        const result = await deliverLaneText({
+                            laneName: segment.lane,
+                            text: segment.text,
+                            payload,
+                            infoKind: info.kind,
+                            previewButtons,
+                            allowPreviewUpdateForNonFinal: segment.lane === "reasoning",
+                        });
+                        if (segment.lane === "reasoning") {
+                            if (result !== "skipped") {
+                                reasoningStepState.noteReasoningDelivered();
+                                await flushBufferedFinalAnswer();
                             }
-                            catch (err) {
-                                logVerbose(`telegram: post-stop preview edit failed; falling back to standard send (${String(err)})`);
+                            continue;
+                        }
+                        if (info.kind === "final") {
+                            if (reasoningLane.hasStreamedMessage) {
+                                finalizedPreviewByLane.reasoning = true;
                             }
+                            reasoningStepState.resetForNextStep();
                         }
                     }
-                    const result = await deliverReplies({
-                        ...deliveryBaseOptions,
-                        replies: [payload],
-                        onVoiceRecording: sendRecordVoice,
-                    });
-                    if (result.delivered) {
-                        deliveryState.delivered = true;
+                    if (segments.length > 0) {
+                        return;
+                    }
+                    if (info.kind === "final") {
+                        await answerLane.stream?.stop();
+                        await reasoningLane.stream?.stop();
+                        reasoningStepState.resetForNextStep();
+                    }
+                    const canSendAsIs = hasMedia || typeof payload.text !== "string" || payload.text.length > 0;
+                    if (!canSendAsIs) {
+                        if (info.kind === "final") {
+                            await flushBufferedFinalAnswer();
+                        }
+                        return;
+                    }
+                    await sendPayload(payload);
+                    if (info.kind === "final") {
+                        await flushBufferedFinalAnswer();
                     }
                 },
                 onSkip: (_payload, info) => {
@@ -327,6 +527,7 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
                     }
                 },
                 onError: (err, info) => {
+                    deliveryState.failedNonSilent += 1;
                     runtime.error?.(danger(`telegram ${info.kind} reply failed: ${String(err)}`));
                 },
                 onReplyStart: createTypingCallbacks({
@@ -344,30 +545,47 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
             replyOptions: {
                 skillFilter,
                 disableBlockStreaming,
-                onPartialReply: draftStream ? (payload) => updateDraftFromPartial(payload.text) : undefined,
-                onAssistantMessageStart: draftStream
-                    ? () => {
-                        // Only split preview bubbles in block mode. In partial mode, keep
-                        // editing one preview message to avoid flooding the chat.
-                        logVerbose(`telegram: onAssistantMessageStart called, hasStreamedMessage=${hasStreamedMessage}`);
-                        if (shouldSplitPreviewMessages && hasStreamedMessage) {
-                            logVerbose(`telegram: calling forceNewMessage()`);
-                            draftStream.forceNewMessage();
+                onPartialReply: answerLane.stream || reasoningLane.stream
+                    ? (payload) => ingestDraftLaneSegments(payload.text)
+                    : undefined,
+                onReasoningStream: reasoningLane.stream
+                    ? (payload) => {
+                        // Split between reasoning blocks only when the next reasoning
+                        // stream starts. Splitting at reasoning-end can orphan the active
+                        // preview and cause duplicate reasoning sends on reasoning final.
+                        if (splitReasoningOnNextStream) {
+                            reasoningLane.stream?.forceNewMessage();
+                            resetDraftLaneState(reasoningLane);
+                            splitReasoningOnNextStream = false;
                         }
-                        lastPartialText = "";
-                        draftText = "";
-                        draftChunker?.reset();
+                        ingestDraftLaneSegments(payload.text);
                     }
                     : undefined,
-                onReasoningEnd: draftStream
-                    ? () => {
-                        // Same policy as assistant-message boundaries: split only in block mode.
-                        if (shouldSplitPreviewMessages && hasStreamedMessage) {
-                            draftStream.forceNewMessage();
+                onAssistantMessageStart: answerLane.stream
+                    ? async () => {
+                        reasoningStepState.resetForNextStep();
+                        if (answerLane.hasStreamedMessage) {
+                            const previewMessageId = answerLane.stream?.messageId();
+                            if (typeof previewMessageId === "number") {
+                                archivedAnswerPreviews.push({
+                                    messageId: previewMessageId,
+                                    textSnapshot: answerLane.lastPartialText,
+                                });
+                            }
+                            answerLane.stream?.forceNewMessage();
                         }
-                        lastPartialText = "";
-                        draftText = "";
-                        draftChunker?.reset();
+                        resetDraftLaneState(answerLane);
+                    }
+                    : undefined,
+                onReasoningEnd: reasoningLane.stream
+                    ? () => {
+                        // Split when/if a later reasoning block begins.
+                        splitReasoningOnNextStream = reasoningLane.hasStreamedMessage;
+                    }
+                    : undefined,
+                onToolStart: statusReactionController
+                    ? async (payload) => {
+                        await statusReactionController.setTool(payload.name);
                     }
                     : undefined,
                 onModelSelected,
@@ -375,14 +593,51 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
         }));
     }
     finally {
-        // Must stop() first to flush debounced content before clear() wipes state
-        await draftStream?.stop();
-        if (!finalizedViaPreviewMessage) {
-            await draftStream?.clear();
+        // Must stop() first to flush debounced content before clear() wipes state.
+        const streamCleanupStates = new Map();
+        const lanesToCleanup = [
+            { laneName: "answer", lane: answerLane },
+            { laneName: "reasoning", lane: reasoningLane },
+        ];
+        for (const laneState of lanesToCleanup) {
+            const stream = laneState.lane.stream;
+            if (!stream) {
+                continue;
+            }
+            const shouldClear = !finalizedPreviewByLane[laneState.laneName];
+            const existing = streamCleanupStates.get(stream);
+            if (!existing) {
+                streamCleanupStates.set(stream, { shouldClear });
+                continue;
+            }
+            existing.shouldClear = existing.shouldClear && shouldClear;
+        }
+        for (const [stream, cleanupState] of streamCleanupStates) {
+            await stream.stop();
+            if (cleanupState.shouldClear) {
+                await stream.clear();
+            }
+        }
+        for (const archivedPreview of archivedAnswerPreviews) {
+            try {
+                await bot.api.deleteMessage(chatId, archivedPreview.messageId);
+            }
+            catch (err) {
+                logVerbose(`telegram: archived answer preview cleanup failed (${archivedPreview.messageId}): ${String(err)}`);
+            }
+        }
+        for (const messageId of archivedReasoningPreviewIds) {
+            try {
+                await bot.api.deleteMessage(chatId, messageId);
+            }
+            catch (err) {
+                logVerbose(`telegram: archived reasoning preview cleanup failed (${messageId}): ${String(err)}`);
+            }
         }
     }
     let sentFallback = false;
-    if (!deliveryState.delivered && deliveryState.skippedNonSilent > 0) {
+    if (!deliveryState.delivered &&
+        (deliveryState.skippedNonSilent > 0 || deliveryState.failedNonSilent > 0)) {
         const result = await deliverReplies({
             replies: [{ text: EMPTY_RESPONSE_FALLBACK }],
             ...deliveryBaseOptions,
@@ -390,26 +645,38 @@ export const dispatchTelegramMessage = async ({ context, bot, cfg, runtime, repl
         sentFallback = result.delivered;
     }
     const hasFinalResponse = queuedFinal || sentFallback;
+    if (statusReactionController && !hasFinalResponse) {
+        void statusReactionController.setError().catch((err) => {
+            logVerbose(`telegram: status reaction error finalize failed: ${String(err)}`);
+        });
+    }
     if (!hasFinalResponse) {
         clearGroupHistory();
         return;
     }
-    removeAckReactionAfterReply({
-        removeAfterReply: removeAckAfterReply,
-        ackReactionPromise,
-        ackReactionValue: ackReactionPromise ? "ack" : null,
-        remove: () => reactionApi?.(chatId, msg.message_id ?? 0, []) ?? Promise.resolve(),
-        onError: (err) => {
-            if (!msg.message_id) {
-                return;
-            }
-            logAckFailure({
-                log: logVerbose,
-                channel: "telegram",
-                target: `${chatId}/${msg.message_id}`,
-                error: err,
-            });
-        },
-    });
+    if (statusReactionController) {
+        void statusReactionController.setDone().catch((err) => {
+            logVerbose(`telegram: status reaction finalize failed: ${String(err)}`);
+        });
+    }
+    else {
+        removeAckReactionAfterReply({
+            removeAfterReply: removeAckAfterReply,
+            ackReactionPromise,
+            ackReactionValue: ackReactionPromise ? "ack" : null,
+            remove: () => reactionApi?.(chatId, msg.message_id ?? 0, []) ?? Promise.resolve(),
+            onError: (err) => {
+                if (!msg.message_id) {
+                    return;
+                }
+                logAckFailure({
+                    log: logVerbose,
+                    channel: "telegram",
+                    target: `${chatId}/${msg.message_id}`,
+                    error: err,
+                });
+            },
+        });
+    }
     clearGroupHistory();
 };