@bd7pil/ocrc 0.9.0

package/dist/transport/telegram/streaming/response-streamer.js

@@ -0,0 +1,400 @@
+import { createLogger } from "../../../utils/logger.js";
+const logger = createLogger("stream");
+import { chunkPlainText } from "../render/chunker.js";
+import { getTelegramRenderedPartSignature } from "../render/part-signature.js";
+import { resolveStreamThrottleMs, } from "./stream-throttle.js";
+function buildStateKey(sessionId, messageId) {
+    return `${sessionId}:${messageId}`;
+}
+function clonePart(part) {
+    return {
+        blocks: [...part.blocks],
+        fallbackText: part.fallbackText,
+        source: part.source,
+        ...(part.entities ? { entities: [...part.entities] } : {}),
+    };
+}
+function isEmptyPart(part) {
+    return part.blocks.length === 0 && part.fallbackText.length === 0;
+}
+/**
+ * Rebuilds a payload as plain text. Native parts may hold far more than a text
+ * message can carry, so the whole representation is re-chunked rather than
+ * swapped part by part; the resulting part count may differ.
+ */
+function toPlainPayload(payload) {
+    // Nothing native to degrade: re-chunking here would only invalidate the
+    // entity offsets a plain part carries.
+    if (payload.parts.every((part) => part.blocks.length === 0)) {
+        return payload;
+    }
+    const text = payload.parts
+        .map((part) => part.fallbackText)
+        .filter(Boolean)
+        .join("\n\n");
+    return {
+        ...payload,
+        parts: chunkPlainText(text),
+    };
+}
+function normalizePayload(payload) {
+    const normalizedParts = payload.parts.map(clonePart).filter((part) => !isEmptyPart(part));
+    if (normalizedParts.length === 0) {
+        logger.debug("[ResponseStreamer] Dropped empty streaming payload after normalization");
+        return null;
+    }
+    return {
+        parts: normalizedParts,
+        sendOptions: payload.sendOptions,
+        editOptions: payload.editOptions,
+    };
+}
+function getErrorMessage(error) {
+    if (error instanceof Error) {
+        return error.message;
+    }
+    return String(error);
+}
+function getRetryAfterMs(error) {
+    const message = getErrorMessage(error);
+    if (!/\b429\b/.test(message)) {
+        return null;
+    }
+    const retryMatch = message.match(/retry after\s+(\d+)/i);
+    if (!retryMatch) {
+        return null;
+    }
+    const secondsText = retryMatch[1];
+    if (!secondsText) {
+        return null;
+    }
+    const seconds = Number.parseInt(secondsText, 10);
+    if (!Number.isFinite(seconds) || seconds <= 0) {
+        return null;
+    }
+    return seconds * 1000;
+}
+function delay(ms) {
+    return new Promise((resolve) => {
+        setTimeout(resolve, ms);
+    });
+}
+function enableNotificationForOptions(options) {
+    if (!options) {
+        return options;
+    }
+    const nextOptions = {
+        ...options,
+    };
+    delete nextOptions.disable_notification;
+    return nextOptions;
+}
+export class ResponseStreamer {
+    throttleMs;
+    sendPart;
+    editPart;
+    deleteText;
+    completePart;
+    states = new Map();
+    constructor(options) {
+        this.throttleMs = options.throttleMs;
+        this.sendPart = options.sendPart;
+        this.editPart = options.editPart;
+        this.deleteText = options.deleteText;
+        this.completePart = options.completePart;
+    }
+    resolveThrottleMs(sessionId) {
+        return resolveStreamThrottleMs(this.throttleMs, sessionId);
+    }
+    enqueue(sessionId, messageId, payload) {
+        const normalizedPayload = normalizePayload(payload);
+        if (!normalizedPayload) {
+            return;
+        }
+        const state = this.getOrCreateState(sessionId, messageId);
+        state.latestPayload = normalizedPayload;
+        if (state.isBroken) {
+            return;
+        }
+        this.ensureTimer(state);
+    }
+    async complete(sessionId, messageId, payload, options) {
+        const notStreamed = { streamed: false, telegramMessageIds: [] };
+        const state = this.states.get(buildStateKey(sessionId, messageId));
+        if (!state) {
+            logger.debug(`[ResponseStreamer] Complete skipped, no active stream state: session=${sessionId}, message=${messageId}`);
+            return notStreamed;
+        }
+        if (payload) {
+            const normalizedPayload = normalizePayload(payload);
+            if (normalizedPayload) {
+                state.latestPayload = normalizedPayload;
+            }
+        }
+        this.clearTimer(state);
+        await state.task.catch(() => false);
+        if (state.isBroken) {
+            await this.cleanupBrokenStream(state, "complete_broken_stream");
+            this.cancelState(state);
+            this.states.delete(state.key);
+            return notStreamed;
+        }
+        if (state.telegramMessageIds.length === 0) {
+            logger.debug(`[ResponseStreamer] Complete returned not streamed: session=${sessionId}, message=${messageId}, reason=no_visible_partials`);
+            this.cancelState(state);
+            this.states.delete(state.key);
+            return notStreamed;
+        }
+        let synced = true;
+        if (options?.flushFinal !== false) {
+            synced = await this.enqueueTask(state, () => this.flushState(state, "complete"));
+        }
+        if (synced && this.completePart && state.latestPayload) {
+            try {
+                // The message this persists is the end of the same answer, so it
+                // inherits the plain-text degradation the stream already fell back to.
+                const completionPayload = state.plainOnly
+                    ? toPlainPayload(state.latestPayload)
+                    : state.latestPayload;
+                const realMessageIds = [];
+                let notifyNextCompletePart = options?.notifyFirstCompletePart === true;
+                for (const part of completionPayload.parts) {
+                    if (isEmptyPart(part)) {
+                        continue;
+                    }
+                    const completeOptions = notifyNextCompletePart
+                        ? enableNotificationForOptions(completionPayload.sendOptions)
+                        : completionPayload.sendOptions;
+                    notifyNextCompletePart = false;
+                    const result = await this.completePart(part, completeOptions, sessionId);
+                    realMessageIds.push(result.messageId);
+                }
+                state.telegramMessageIds = realMessageIds;
+            }
+            catch (error) {
+                logger.error(`[ResponseStreamer] Failed to persist draft message: session=${sessionId}, message=${messageId}`, error);
+                synced = false;
+            }
+        }
+        const messageIds = [...state.telegramMessageIds];
+        this.cancelState(state);
+        this.states.delete(state.key);
+        return { streamed: synced, telegramMessageIds: messageIds };
+    }
+    clearMessage(sessionId, messageId, reason) {
+        const key = buildStateKey(sessionId, messageId);
+        const state = this.states.get(key);
+        if (!state) {
+            return;
+        }
+        this.cancelState(state);
+        this.states.delete(key);
+        logger.debug(`[ResponseStreamer] Cleared message stream: session=${sessionId}, message=${messageId}, reason=${reason}`);
+    }
+    clearSession(sessionId, reason) {
+        for (const state of Array.from(this.states.values())) {
+            if (state.sessionId !== sessionId) {
+                continue;
+            }
+            this.cancelState(state);
+            this.states.delete(state.key);
+        }
+        logger.debug(`[ResponseStreamer] Cleared session streams: session=${sessionId}, reason=${reason}`);
+    }
+    clearAll(reason) {
+        for (const state of this.states.values()) {
+            this.cancelState(state);
+        }
+        const count = this.states.size;
+        this.states.clear();
+        if (count > 0) {
+            logger.debug(`[ResponseStreamer] Cleared all streams: count=${count}, reason=${reason}`);
+        }
+    }
+    hasActiveStream(sessionId) {
+        for (const state of this.states.values()) {
+            if (state.sessionId === sessionId && !state.cancelled) {
+                return true;
+            }
+        }
+        return false;
+    }
+    getOrCreateState(sessionId, messageId) {
+        const key = buildStateKey(sessionId, messageId);
+        const existing = this.states.get(key);
+        if (existing) {
+            return existing;
+        }
+        const state = {
+            key,
+            sessionId,
+            messageId,
+            latestPayload: null,
+            lastSentSignatures: [],
+            telegramMessageIds: [],
+            timer: null,
+            task: Promise.resolve(true),
+            cancelled: false,
+            isBroken: false,
+            plainOnly: false,
+            fatalErrorMessage: null,
+            fatalErrorLogged: false,
+        };
+        this.states.set(key, state);
+        return state;
+    }
+    ensureTimer(state) {
+        if (state.timer || state.cancelled) {
+            return;
+        }
+        const throttleMs = this.resolveThrottleMs(state.sessionId);
+        if (throttleMs === 0) {
+            void this.enqueueTask(state, () => this.flushState(state, "immediate")).catch((error) => {
+                logger.error(`[ResponseStreamer] Immediate stream sync failed: session=${state.sessionId}, message=${state.messageId}`, error);
+            });
+            return;
+        }
+        state.timer = setTimeout(() => {
+            state.timer = null;
+            void this.enqueueTask(state, () => this.flushState(state, "throttle_elapsed")).catch((error) => {
+                logger.error(`[ResponseStreamer] Throttled stream sync failed: session=${state.sessionId}, message=${state.messageId}`, error);
+            });
+        }, throttleMs);
+    }
+    clearTimer(state) {
+        if (!state.timer) {
+            return;
+        }
+        clearTimeout(state.timer);
+        state.timer = null;
+    }
+    cancelState(state) {
+        state.cancelled = true;
+        this.clearTimer(state);
+    }
+    enqueueTask(state, task) {
+        const nextTask = state.task.catch(() => false).then(task);
+        state.task = nextTask;
+        return nextTask;
+    }
+    async flushState(state, reason) {
+        if (state.cancelled) {
+            return false;
+        }
+        if (state.isBroken) {
+            return false;
+        }
+        while (!state.cancelled) {
+            const latestPayload = state.latestPayload;
+            if (!latestPayload) {
+                return state.telegramMessageIds.length > 0;
+            }
+            // Signatures must describe what actually goes out, otherwise the
+            // "unchanged" checks below compare two different representations.
+            const payload = state.plainOnly ? toPlainPayload(latestPayload) : latestPayload;
+            const targetSignatures = payload.parts.map((part) => getTelegramRenderedPartSignature(part));
+            const unchanged = targetSignatures.length === state.lastSentSignatures.length &&
+                targetSignatures.every((signature, index) => signature === state.lastSentSignatures[index]);
+            if (unchanged) {
+                logger.debug(`[ResponseStreamer] Skipped unchanged payload: session=${state.sessionId}, message=${state.messageId}, parts=${payload.parts.length}`);
+                return state.telegramMessageIds.length > 0;
+            }
+            try {
+                await this.syncMessages(state, payload, targetSignatures);
+                logger.debug(`[ResponseStreamer] Stream synced: session=${state.sessionId}, message=${state.messageId}, reason=${reason}, parts=${payload.parts.length}`);
+                return true;
+            }
+            catch (error) {
+                const retryAfterMs = getRetryAfterMs(error);
+                if (retryAfterMs !== null) {
+                    const delayMs = Math.max(this.resolveThrottleMs(state.sessionId), retryAfterMs);
+                    logger.warn(`[ResponseStreamer] Stream sync rate-limited, retrying in ${delayMs}ms: session=${state.sessionId}, message=${state.messageId}, reason=${reason}`, error);
+                    await delay(delayMs);
+                    continue;
+                }
+                // First native failure: fall back to plain text for the rest of this
+                // message instead of tearing the stream down. A payload that is already
+                // plain has nothing left to degrade to, so it breaks straight away.
+                if (!state.plainOnly && payload.parts.some((part) => part.blocks.length > 0)) {
+                    this.markStreamPlainOnly(state, error, reason);
+                    continue;
+                }
+                this.markStreamBroken(state, error, reason);
+                return false;
+            }
+        }
+        return false;
+    }
+    markStreamPlainOnly(state, error, reason) {
+        if (state.plainOnly) {
+            return;
+        }
+        state.plainOnly = true;
+        logger.warn(`[ResponseStreamer] Native rendering failed, continuing this message as plain text: session=${state.sessionId}, message=${state.messageId}, reason=${reason}`, error);
+    }
+    markStreamBroken(state, error, reason) {
+        state.isBroken = true;
+        state.fatalErrorMessage = getErrorMessage(error);
+        if (state.fatalErrorLogged) {
+            return;
+        }
+        state.fatalErrorLogged = true;
+        logger.error(`[ResponseStreamer] Stream marked as broken: session=${state.sessionId}, message=${state.messageId}, reason=${reason}, error=${state.fatalErrorMessage}`, error);
+    }
+    async cleanupBrokenStream(state, reason) {
+        if (state.telegramMessageIds.length === 0) {
+            return;
+        }
+        for (let index = state.telegramMessageIds.length - 1; index >= 0; index--) {
+            const messageId = state.telegramMessageIds[index];
+            if (!messageId) {
+                continue;
+            }
+            try {
+                await this.deleteText(messageId, state.sessionId);
+            }
+            catch (error) {
+                logger.warn(`[ResponseStreamer] Failed to delete broken stream message: session=${state.sessionId}, message=${state.messageId}, telegramMessageId=${messageId}, reason=${reason}`, error);
+            }
+        }
+        state.telegramMessageIds = [];
+        state.lastSentSignatures = [];
+        logger.debug(`[ResponseStreamer] Cleaned up broken stream messages: session=${state.sessionId}, message=${state.messageId}, reason=${reason}`);
+    }
+    async syncMessages(state, payload, targetSignatures) {
+        for (let index = 0; index < payload.parts.length; index++) {
+            const part = payload.parts[index];
+            if (!part) {
+                continue;
+            }
+            const nextSignature = targetSignatures[index];
+            const currentMessageId = state.telegramMessageIds[index];
+            if (currentMessageId) {
+                if (state.lastSentSignatures[index] === nextSignature) {
+                    continue;
+                }
+                const result = await this.editPart(currentMessageId, part, payload.editOptions, state.sessionId);
+                state.lastSentSignatures[index] = result.deliveredSignature;
+                if (result.degradedToPlain) {
+                    this.markStreamPlainOnly(state, null, "transport_degraded_to_plain");
+                }
+                continue;
+            }
+            const result = await this.sendPart(part, payload.sendOptions, state.sessionId);
+            state.telegramMessageIds[index] = result.messageId;
+            state.lastSentSignatures[index] = result.deliveredSignature;
+            if (result.degradedToPlain) {
+                this.markStreamPlainOnly(state, null, "transport_degraded_to_plain");
+            }
+        }
+        for (let index = state.telegramMessageIds.length - 1; index >= payload.parts.length; index--) {
+            const messageId = state.telegramMessageIds[index];
+            if (messageId) {
+                await this.deleteText(messageId, state.sessionId);
+            }
+            state.telegramMessageIds.pop();
+            state.lastSentSignatures.pop();
+        }
+    }
+}
+//# sourceMappingURL=response-streamer.js.map

package/dist/transport/telegram/streaming/response-streamer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"response-streamer.js","sourceRoot":"","sources":["../../../../src/transport/telegram/streaming/response-streamer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,gCAAgC,EAAE,MAAM,6BAA6B,CAAC;AAE/E,OAAO,EACL,uBAAuB,GAExB,MAAM,sBAAsB,CAAC;AA8D9B,SAAS,aAAa,CAAC,SAAiB,EAAE,SAAiB;IACzD,OAAO,GAAG,SAAS,IAAI,SAAS,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,SAAS,CAAC,IAA0B;IAC3C,OAAO;QACL,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACxB,YAAY,EAAE,IAAI,CAAC,YAAY;QAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC3D,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,IAA0B;IAC7C,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,CAAC;AACpE,CAAC;AAED;;;;GAIG;AACH,SAAS,cAAc,CAAC,OAAgC;IACtD,wEAAwE;IACxE,uCAAuC;IACvC,IAAI,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;QAC5D,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK;SACvB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC;SAChC,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,MAAM,CAAC,CAAC;IAEhB,OAAO;QACL,GAAG,OAAO;QACV,KAAK,EAAE,cAAc,CAAC,IAAI,CAAC;KAC5B,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAgC;IACxD,MAAM,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1F,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,wEAAwE,CAAC,CAAC;QACvF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QACL,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,WAAW,EAAE,OAAO,CAAC,WAAW;KACjC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IAED,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACvC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;IACzD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,WAAW,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;IAClC,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IACjD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;QAC9C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,OAAO,GAAG,IAAI,CAAC;AACxB,CAAC;AAED,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,4BAA4B,CACnC,OAA+C;IAE/C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,WAAW,GAAG;QAClB,GAAG,OAAO;KACqE,CAAC;IAClF,OAAO,WAAW,CAAC,oBAAoB,CAAC;IAExC,OAAO,WAAyC,CAAC;AACnD,CAAC;AAED,MAAM,OAAO,gBAAgB;IACV,UAAU,CAAmB;IAC7B,QAAQ,CAAsC;IAC9C,QAAQ,CAAsC;IAC9C,UAAU,CAAwC;IAClD,YAAY,CAA0C;IACtD,MAAM,GAA6B,IAAI,GAAG,EAAE,CAAC;IAE9D,YAAY,OAAgC;QAC1C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC3C,CAAC;IAEO,iBAAiB,CAAC,SAAiB;QACzC,OAAO,uBAAuB,CAAC,IAAI,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IAC7D,CAAC;IAED,OAAO,CAAC,SAAiB,EAAE,SAAiB,EAAE,OAAgC;QAC5E,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACpD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC1D,KAAK,CAAC,aAAa,GAAG,iBAAiB,CAAC;QACxC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,SAAiB,EACjB,SAAiB,EACjB,OAAiC,EACjC,OAAyC;QAEzC,MAAM,WAAW,GAAyB,EAAE,QAAQ,EAAE,KAAK,EAAE,kBAAkB,EAAE,EAAE,EAAE,CAAC;QAEtF,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QACnE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,CAAC,KAAK,CACV,wEAAwE,SAAS,aAAa,SAAS,EAAE,CAC1G,CAAC;YACF,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;YACpD,IAAI,iBAAiB,EAAE,CAAC;gBACtB,KAAK,CAAC,aAAa,GAAG,iBAAiB,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAEvB,MAAM,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;QAEpC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,wBAAwB,CAAC,CAAC;YAChE,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,KAAK,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,MAAM,CAAC,KAAK,CACV,8DAA8D,SAAS,aAAa,SAAS,8BAA8B,CAC5H,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC9B,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,MAAM,GAAG,IAAI,CAAC;QAClB,IAAI,OAAO,EAAE,UAAU,KAAK,KAAK,EAAE,CAAC;YAClC,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;QACnF,CAAC;QAED,IAAI,MAAM,IAAI,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,iEAAiE;gBACjE,uEAAuE;gBACvE,MAAM,iBAAiB,GAAG,KAAK,CAAC,SAAS;oBACvC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,aAAa,CAAC;oBACrC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;gBACxB,MAAM,cAAc,GAAa,EAAE,CAAC;gBACpC,IAAI,sBAAsB,GAAG,OAAO,EAAE,uBAAuB,KAAK,IAAI,CAAC;gBACvE,KAAK,MAAM,IAAI,IAAI,iBAAiB,CAAC,KAAK,EAAE,CAAC;oBAC3C,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;wBACtB,SAAS;oBACX,CAAC;oBACD,MAAM,eAAe,GAAG,sBAAsB;wBAC5C,CAAC,CAAC,4BAA4B,CAAC,iBAAiB,CAAC,WAAW,CAAC;wBAC7D,CAAC,CAAC,iBAAiB,CAAC,WAAW,CAAC;oBAClC,sBAAsB,GAAG,KAAK,CAAC;oBAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC;oBACzE,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACxC,CAAC;gBACD,KAAK,CAAC,kBAAkB,GAAG,cAAc,CAAC;YAC5C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CACV,+DAA+D,SAAS,aAAa,SAAS,EAAE,EAChG,KAAK,CACN,CAAC;gBACF,MAAM,GAAG,KAAK,CAAC;YACjB,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,GAAG,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACjD,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC9B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC;IAC9D,CAAC;IAED,YAAY,CAAC,SAAiB,EAAE,SAAiB,EAAE,MAAc;QAC/D,MAAM,GAAG,GAAG,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACxB,MAAM,CAAC,KAAK,CACV,sDAAsD,SAAS,aAAa,SAAS,YAAY,MAAM,EAAE,CAC1G,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,SAAiB,EAAE,MAAc;QAC5C,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YACrD,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBAClC,SAAS;YACX,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,CAAC,KAAK,CACV,uDAAuD,SAAS,YAAY,MAAM,EAAE,CACrF,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,MAAc;QACrB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAEpB,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,MAAM,CAAC,KAAK,CAAC,iDAAiD,KAAK,YAAY,MAAM,EAAE,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IAED,eAAe,CAAC,SAAiB;QAC/B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;gBACtD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,gBAAgB,CAAC,SAAiB,EAAE,SAAiB;QAC3D,MAAM,GAAG,GAAG,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACtC,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,KAAK,GAAgB;YACzB,GAAG;YACH,SAAS;YACT,SAAS;YACT,aAAa,EAAE,IAAI;YACnB,kBAAkB,EAAE,EAAE;YACtB,kBAAkB,EAAE,EAAE;YACtB,KAAK,EAAE,IAAI;YACX,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;YAC3B,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,KAAK;YACf,SAAS,EAAE,KAAK;YAChB,iBAAiB,EAAE,IAAI;YACvB,gBAAgB,EAAE,KAAK;SACxB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC5B,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,WAAW,CAAC,KAAkB;QACpC,IAAI,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACnC,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;YACrB,KAAK,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACtF,MAAM,CAAC,KAAK,CACV,4DAA4D,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,EAAE,EACzG,KAAK,CACN,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,KAAK,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;YACnB,KAAK,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC,CAAC,KAAK,CAClF,CAAC,KAAK,EAAE,EAAE;gBACR,MAAM,CAAC,KAAK,CACV,4DAA4D,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,EAAE,EACzG,KAAK,CACN,CAAC;YACJ,CAAC,CACF,CAAC;QACJ,CAAC,EAAE,UAAU,CAAC,CAAC;IACjB,CAAC;IAEO,UAAU,CAAC,KAAkB;QACnC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1B,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC;IACrB,CAAC;IAEO,WAAW,CAAC,KAAkB;QACpC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;IAEO,WAAW,CAAC,KAAkB,EAAE,IAA4B;QAClE,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,KAAK,CAAC,IAAI,GAAG,QAAQ,CAAC;QACtB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAkB,EAAE,MAAc;QACzD,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YACxB,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;YAC1C,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,OAAO,KAAK,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;YAC7C,CAAC;YAED,iEAAiE;YACjE,kEAAkE;YAClE,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAChF,MAAM,gBAAgB,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAClD,gCAAgC,CAAC,IAAI,CAAC,CACvC,CAAC;YACF,MAAM,SAAS,GACb,gBAAgB,CAAC,MAAM,KAAK,KAAK,CAAC,kBAAkB,CAAC,MAAM;gBAC3D,gBAAgB,CAAC,KAAK,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,KAAK,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC;YAE9F,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,CAAC,KAAK,CACV,yDAAyD,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,WAAW,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CACtI,CAAC;gBACF,OAAO,KAAK,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;YAC7C,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC;gBAC1D,MAAM,CAAC,KAAK,CACV,6CAA6C,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,YAAY,MAAM,WAAW,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAC5I,CAAC;gBACF,OAAO,IAAI,CAAC;YACd,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC5C,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;oBAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,YAAY,CAAC,CAAC;oBAChF,MAAM,CAAC,IAAI,CACT,4DAA4D,OAAO,eAAe,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,YAAY,MAAM,EAAE,EACjJ,KAAK,CACN,CAAC;oBACF,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC;oBACrB,SAAS;gBACX,CAAC;gBAED,qEAAqE;gBACrE,wEAAwE;gBACxE,oEAAoE;gBACpE,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC;oBAC7E,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;oBAC/C,SAAS;gBACX,CAAC;gBAED,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC5C,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,mBAAmB,CAAC,KAAkB,EAAE,KAAc,EAAE,MAAc;QAC5E,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,MAAM,CAAC,IAAI,CACT,8FAA8F,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,YAAY,MAAM,EAAE,EAC7J,KAAK,CACN,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,KAAkB,EAAE,KAAc,EAAE,MAAc;QACzE,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC;QACtB,KAAK,CAAC,iBAAiB,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAEjD,IAAI,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC9B,MAAM,CAAC,KAAK,CACV,uDAAuD,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,YAAY,MAAM,WAAW,KAAK,CAAC,iBAAiB,EAAE,EACxJ,KAAK,CACN,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,KAAkB,EAAE,MAAc;QAClE,IAAI,KAAK,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,KAAK,IAAI,KAAK,GAAG,KAAK,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;YAC1E,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;YACpD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CACT,sEAAsE,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,uBAAuB,SAAS,YAAY,MAAM,EAAE,EACrK,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC;QAC9B,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC;QAC9B,MAAM,CAAC,KAAK,CACV,iEAAiE,KAAK,CAAC,SAAS,aAAa,KAAK,CAAC,SAAS,YAAY,MAAM,EAAE,CACjI,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,KAAkB,EAClB,OAAgC,EAChC,gBAA0B;QAE1B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;YAC1D,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YACD,MAAM,aAAa,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,gBAAgB,GAAG,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAEzD,IAAI,gBAAgB,EAAE,CAAC;gBACrB,IAAI,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,KAAK,aAAa,EAAE,CAAC;oBACtD,SAAS;gBACX,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,IAAI,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;gBACjG,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,kBAAkB,CAAC;gBAC5D,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;oBAC3B,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,IAAI,EAAE,6BAA6B,CAAC,CAAC;gBACvE,CAAC;gBACD,SAAS;YACX,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;YAC/E,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC;YACnD,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,kBAAkB,CAAC;YAC5D,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;gBAC3B,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,IAAI,EAAE,6BAA6B,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;QAED,KAAK,IAAI,KAAK,GAAG,KAAK,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;YAC7F,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;YACpD,CAAC;YACD,KAAK,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC;YAC/B,KAAK,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;CACF"}

package/dist/transport/telegram/streaming/running-tool-tracker.js

@@ -0,0 +1,150 @@
+import { bucketElapsedMs } from "./duration-formatter.js";
+import { createLogger } from "../../../utils/logger.js";
+const logger = createLogger("stream");
+/**
+ * Drives elapsed-time updates for tool calls.
+ *
+ * OpenCode emits `running` tool events when the tool output changes, not on a
+ * schedule: a tool that blocks without printing anything produces no events at
+ * all. Elapsed time therefore has to come from an own interval, never from
+ * incoming events.
+ */
+export class RunningToolTracker {
+    thresholdMs;
+    tickIntervalMs;
+    maxTrackingMs;
+    onTick;
+    onHeartbeat;
+    calls = new Map();
+    heartbeatSessions = new Set();
+    timer = null;
+    constructor(options) {
+        this.thresholdMs = options.thresholdMs;
+        this.tickIntervalMs = options.tickIntervalMs;
+        this.maxTrackingMs = options.maxTrackingMs;
+        this.onTick = options.onTick;
+        this.onHeartbeat = options.onHeartbeat;
+    }
+    track(sessionId, callId) {
+        if (!sessionId || !callId || this.calls.has(callId)) {
+            return;
+        }
+        this.calls.set(callId, { sessionId, startedAt: Date.now(), stopped: false });
+        this.ensureTimer();
+    }
+    /**
+     * Forgets the call and returns how long it ran, but only if it lived past the
+     * threshold. Fast calls return undefined so their output stays as it was.
+     */
+    release(callId) {
+        const call = this.calls.get(callId);
+        if (!call) {
+            return undefined;
+        }
+        this.calls.delete(callId);
+        this.stopTimerWhenIdle();
+        if (call.lastBucketMs === undefined) {
+            return undefined;
+        }
+        return Date.now() - call.startedAt;
+    }
+    setHeartbeatActive(sessionId, active) {
+        if (!sessionId) {
+            return;
+        }
+        if (active) {
+            this.heartbeatSessions.add(sessionId);
+            this.ensureTimer();
+            return;
+        }
+        this.heartbeatSessions.delete(sessionId);
+        this.stopTimerWhenIdle();
+    }
+    clearSession(sessionId, reason) {
+        let clearedAny = this.heartbeatSessions.delete(sessionId);
+        for (const [callId, call] of Array.from(this.calls.entries())) {
+            if (call.sessionId !== sessionId) {
+                continue;
+            }
+            this.calls.delete(callId);
+            clearedAny = true;
+        }
+        this.stopTimerWhenIdle();
+        if (clearedAny) {
+            logger.debug(`[RunningToolTracker] Cleared session: session=${sessionId}, reason=${reason}`);
+        }
+    }
+    clearAll(reason) {
+        const count = this.calls.size;
+        this.calls.clear();
+        this.heartbeatSessions.clear();
+        this.stopTimerWhenIdle();
+        if (count > 0) {
+            logger.debug(`[RunningToolTracker] Cleared all calls: count=${count}, reason=${reason}`);
+        }
+    }
+    trackedCallIds(sessionId) {
+        const matches = [];
+        for (const [callId, call] of this.calls) {
+            if (call.sessionId === sessionId) {
+                matches.push({ callId, startedAt: call.startedAt });
+            }
+        }
+        matches.reverse();
+        matches.sort((left, right) => right.startedAt - left.startedAt);
+        return matches.map((entry) => entry.callId);
+    }
+    newestCallId(sessionId) {
+        return this.trackedCallIds(sessionId)[0];
+    }
+    displayTick(callId, now = Date.now()) {
+        const call = this.calls.get(callId);
+        if (!call) {
+            return undefined;
+        }
+        const elapsedMs = now - call.startedAt;
+        if (elapsedMs < this.thresholdMs) {
+            return undefined;
+        }
+        const isFinal = elapsedMs >= this.maxTrackingMs || call.stopped;
+        const bucketMs = isFinal ? this.maxTrackingMs : bucketElapsedMs(elapsedMs);
+        return { sessionId: call.sessionId, callId, elapsedMs: bucketMs, isFinal };
+    }
+    ensureTimer() {
+        if (this.timer) {
+            return;
+        }
+        this.timer = setInterval(() => this.tick(), this.tickIntervalMs);
+    }
+    stopTimerWhenIdle() {
+        if (!this.timer || this.calls.size > 0 || this.heartbeatSessions.size > 0) {
+            return;
+        }
+        clearInterval(this.timer);
+        this.timer = null;
+    }
+    tick() {
+        const now = Date.now();
+        for (const [callId, call] of this.calls) {
+            if (call.stopped) {
+                continue;
+            }
+            const elapsedMs = now - call.startedAt;
+            if (elapsedMs < this.thresholdMs) {
+                continue;
+            }
+            const isFinal = elapsedMs >= this.maxTrackingMs;
+            const bucketMs = isFinal ? this.maxTrackingMs : bucketElapsedMs(elapsedMs);
+            if (!isFinal && call.lastBucketMs === bucketMs) {
+                continue;
+            }
+            call.lastBucketMs = bucketMs;
+            call.stopped = isFinal;
+            this.onTick({ sessionId: call.sessionId, callId, elapsedMs: bucketMs, isFinal });
+        }
+        for (const sessionId of this.heartbeatSessions) {
+            this.onHeartbeat(sessionId);
+        }
+    }
+}
+//# sourceMappingURL=running-tool-tracker.js.map

package/dist/transport/telegram/streaming/running-tool-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"running-tool-tracker.js","sourceRoot":"","sources":["../../../../src/transport/telegram/streaming/running-tool-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;AAwBtC;;;;;;;GAOG;AACH,MAAM,OAAO,kBAAkB;IACZ,WAAW,CAAS;IACpB,cAAc,CAAS;IACvB,aAAa,CAAS;IACtB,MAAM,CAAsC;IAC5C,WAAW,CAA2C;IACtD,KAAK,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC5C,iBAAiB,GAAgB,IAAI,GAAG,EAAE,CAAC;IACpD,KAAK,GAA0C,IAAI,CAAC;IAE5D,YAAY,OAAkC;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,SAAiB,EAAE,MAAc;QACrC,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACpD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7E,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED;;;OAGG;IACH,OAAO,CAAC,MAAc;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,kBAAkB,CAAC,SAAiB,EAAE,MAAe;QACnD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACtC,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAED,YAAY,CAAC,SAAiB,EAAE,MAAc;QAC5C,IAAI,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAE1D,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAC9D,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC1B,UAAU,GAAG,IAAI,CAAC;QACpB,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,iDAAiD,SAAS,YAAY,MAAM,EAAE,CAAC,CAAC;QAC/F,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,MAAc;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,MAAM,CAAC,KAAK,CAAC,iDAAiD,KAAK,YAAY,MAAM,EAAE,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;IAED,cAAc,CAAC,SAAiB;QAC9B,MAAM,OAAO,GAA4C,EAAE,CAAC;QAE5D,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACxC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACjC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAED,OAAO,CAAC,OAAO,EAAE,CAAC;QAClB,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,YAAY,CAAC,SAAiB;QAC5B,OAAO,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED,WAAW,CAAC,MAAc,EAAE,MAAc,IAAI,CAAC,GAAG,EAAE;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,SAAS,GAAG,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC;QACvC,IAAI,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YACjC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,OAAO,GAAG,SAAS,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,OAAO,CAAC;QAChE,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAC3E,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAC7E,CAAC;IAEO,WAAW;QACjB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1E,OAAO;QACT,CAAC;QAED,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAEO,IAAI;QACV,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACxC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,MAAM,SAAS,GAAG,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC;YACvC,IAAI,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,SAAS,IAAI,IAAI,CAAC,aAAa,CAAC;YAChD,MAAM,QAAQ,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAC3E,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;gBAC/C,SAAS;YACX,CAAC;YAED,IAAI,CAAC,YAAY,GAAG,QAAQ,CAAC;YAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QACnF,CAAC;QAED,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC/C,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;CACF"}

package/dist/transport/telegram/streaming/stream-throttle.js

@@ -0,0 +1,57 @@
+const SECOND_MS = 1000;
+const MINUTE_MS = 60 * SECOND_MS;
+export const STREAM_THROTTLE_BASE_MS = SECOND_MS;
+export const STREAM_THROTTLE_MAX_MS = 10 * SECOND_MS;
+const TIER_TWO_AFTER_MS = MINUTE_MS;
+const TIER_FIVE_AFTER_MS = 5 * MINUTE_MS;
+const TIER_TEN_AFTER_MS = 15 * MINUTE_MS;
+const startedAtBySession = new Map();
+export function resolveProgressiveThrottleMs(elapsedMs) {
+    if (elapsedMs < TIER_TWO_AFTER_MS) {
+        return STREAM_THROTTLE_BASE_MS;
+    }
+    if (elapsedMs < TIER_FIVE_AFTER_MS) {
+        return 2 * SECOND_MS;
+    }
+    if (elapsedMs < TIER_TEN_AFTER_MS) {
+        return 5 * SECOND_MS;
+    }
+    return STREAM_THROTTLE_MAX_MS;
+}
+export function noteStreamActivity(sessionId, now = Date.now()) {
+    if (!sessionId || startedAtBySession.has(sessionId)) {
+        return;
+    }
+    startedAtBySession.set(sessionId, now);
+}
+export function getStreamThrottleMs(sessionId, now = Date.now()) {
+    if (!sessionId) {
+        return STREAM_THROTTLE_BASE_MS;
+    }
+    const startedAt = startedAtBySession.get(sessionId);
+    if (startedAt === undefined) {
+        return STREAM_THROTTLE_BASE_MS;
+    }
+    return resolveProgressiveThrottleMs(now - startedAt);
+}
+export function getSessionStreamThrottleMs(sessionId, now = Date.now()) {
+    noteStreamActivity(sessionId, now);
+    return getStreamThrottleMs(sessionId, now);
+}
+export function resolveStreamThrottleMs(throttleMs, sessionId) {
+    const value = typeof throttleMs === "function" ? throttleMs(sessionId) : throttleMs;
+    return Math.max(0, Math.floor(value));
+}
+export function resetStreamThrottle(sessionId) {
+    if (!sessionId) {
+        return;
+    }
+    startedAtBySession.delete(sessionId);
+}
+export function resetAllStreamThrottles() {
+    startedAtBySession.clear();
+}
+export function __resetStreamThrottleForTests() {
+    startedAtBySession.clear();
+}
+//# sourceMappingURL=stream-throttle.js.map

package/dist/transport/telegram/streaming/stream-throttle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-throttle.js","sourceRoot":"","sources":["../../../../src/transport/telegram/streaming/stream-throttle.ts"],"names":[],"mappings":"AAAA,MAAM,SAAS,GAAG,IAAI,CAAC;AACvB,MAAM,SAAS,GAAG,EAAE,GAAG,SAAS,CAAC;AAEjC,MAAM,CAAC,MAAM,uBAAuB,GAAG,SAAS,CAAC;AACjD,MAAM,CAAC,MAAM,sBAAsB,GAAG,EAAE,GAAG,SAAS,CAAC;AAErD,MAAM,iBAAiB,GAAG,SAAS,CAAC;AACpC,MAAM,kBAAkB,GAAG,CAAC,GAAG,SAAS,CAAC;AACzC,MAAM,iBAAiB,GAAG,EAAE,GAAG,SAAS,CAAC;AAIzC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAkB,CAAC;AAErD,MAAM,UAAU,4BAA4B,CAAC,SAAiB;IAC5D,IAAI,SAAS,GAAG,iBAAiB,EAAE,CAAC;QAClC,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,IAAI,SAAS,GAAG,kBAAkB,EAAE,CAAC;QACnC,OAAO,CAAC,GAAG,SAAS,CAAC;IACvB,CAAC;IAED,IAAI,SAAS,GAAG,iBAAiB,EAAE,CAAC;QAClC,OAAO,CAAC,GAAG,SAAS,CAAC;IACvB,CAAC;IAED,OAAO,sBAAsB,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB,EAAE,MAAc,IAAI,CAAC,GAAG,EAAE;IAC5E,IAAI,CAAC,SAAS,IAAI,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QACpD,OAAO;IACT,CAAC;IAED,kBAAkB,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,SAAiB,EAAE,MAAc,IAAI,CAAC,GAAG,EAAE;IAC7E,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,MAAM,SAAS,GAAG,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACpD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED,OAAO,4BAA4B,CAAC,GAAG,GAAG,SAAS,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,SAAiB,EAAE,MAAc,IAAI,CAAC,GAAG,EAAE;IACpF,kBAAkB,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACnC,OAAO,mBAAmB,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,UAA4B,EAAE,SAAiB;IACrF,MAAM,KAAK,GAAG,OAAO,UAAU,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;IACpF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,SAAiB;IACnD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO;IACT,CAAC;IAED,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,uBAAuB;IACrC,kBAAkB,CAAC,KAAK,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,6BAA6B;IAC3C,kBAAkB,CAAC,KAAK,EAAE,CAAC;AAC7B,CAAC"}