Function markInterrupted

packages/bot-telegram/src/event-renderer.ts:317–357

()

      pendingInterrupt = undefined;
    },
    async markInterrupted() {
      // Idempotent. Mark BEFORE any await so subsequent subscriber callbacks
      // (including the RUN_ERROR that AG-UI fires when we abort) bail.
      if (aborted) return;
      aborted = true;
      // Settle EVERY in-flight text stream. `ChunkedEditStream.append`
      // schedules its placeholder post asynchronously (on setupPromise), so a
      // stream that just received content may still report chunkCount === 0
      // until that post resolves. If we deleted such a stream without
      // finishing it, its in-flight postPlaceholder would still resolve and
      // leave a stray "…" message orphaned forever. `finish()` awaits
      // setupPromise, so it deterministically creates-and-flushes any pending
      // placeholder rather than abandoning it.
      const tasks: Promise<void>[] = [];
      for (const [id, stream] of Array.from(streams.entries())) {
        const buf = buffers.get(id) ?? "";
        if (buf.trim().length > 0) {
          // Real content: surface the interrupted marker on the partial reply
          // before finishing.
          stream.append(buf + INTERRUPTED_SUFFIX);
        }
        // Empty/whitespace buffer: just finish(). If nothing was ever posted
        // it's a no-op; if a placeholder post is in flight it gets flushed/
        // edited rather than left as a stray "…".
        tasks.push(
          stream
            .finish()
            .catch((e) =>
              console.error("[bot-telegram] stream finalize failed:", e),
            ),
        );
        streams.delete(id);
        finalised.add(id);
      }
      buffers.clear();
      await Promise.all(tasks);
      // Clean up any tool-status placeholders whose END never arrived before
      // the interrupt — mark them cancelled rather than leaving the stale
      // "🔧 using <tool>…" line orphaned in the chat.
      await drainToolStatuses((tool) => `⚠️ ${tool} (cancelled)`);
    },
  };
}