Flush buffered Telegram finals after segmented delivery

R2 · R2 · commit f1b425034912 · 2026-05-18T13:42:56.000Z
diff --git a/extensions/telegram/src/bot-message-dispatch.test.ts b/extensions/telegram/src/bot-message-dispatch.test.ts
@@ -1587,6 +1587,46 @@ describe("dispatchTelegramMessage draft streaming", () => {
     expect(answerDraftStream.update).toHaveBeenCalledWith("Answer");
   });
 
+  it("flushes buffered final answers after a skipped reasoning segment", async () => {
+    deliverInboundReplyWithMessageSendContext
+      .mockResolvedValueOnce({ status: "handled_no_send" })
+      .mockResolvedValueOnce({
+        status: "handled_visible",
+        delivery: { messageIds: ["1001"], visibleReplySent: true },
+      });
+    dispatchReplyWithBufferedBlockDispatcher.mockImplementation(async ({ dispatcherOptions }) => {
+      await dispatcherOptions.deliver(
+        { text: "<think>Hidden reasoning</think>Visible answer" },
+        { kind: "final" },
+      );
+      return { queuedFinal: true };
+    });
+
+    await dispatchWithContext({
+      context: createContext({
+        ctxPayload: {
+          SessionKey: "agent:main:telegram:direct:123",
+          ChatType: "direct",
+        } as unknown as TelegramMessageContext["ctxPayload"],
+      }),
+      cfg: { agents: { defaults: { reasoningDefault: "on" } } },
+      streamMode: "off",
+      telegramDeps: telegramDepsForTest,
+    });
+
+    expect(deliverInboundReplyWithMessageSendContext).toHaveBeenCalledTimes(2);
+    expectRecordFields(mockCallArg(deliverInboundReplyWithMessageSendContext, 0), {
+      info: { kind: "final" },
+    });
+    expectRecordFields(mockCallArg(deliverInboundReplyWithMessageSendContext, 0).payload, {
+      text: "Reasoning:\n_Hidden reasoning_",
+    });
+    expectRecordFields(mockCallArg(deliverInboundReplyWithMessageSendContext, 1).payload, {
+      text: "Visible answer",
+    });
+    expect(deliverReplies).not.toHaveBeenCalled();
+  });
+
   it("suppresses reasoning-only finals without raw text fallback", async () => {
     setupDraftStreams({ answerMessageId: 2001, reasoningMessageId: 3001 });
     dispatchReplyWithBufferedBlockDispatcher.mockImplementation(async ({ dispatcherOptions }) => {
diff --git a/extensions/telegram/src/bot-message-dispatch.ts b/extensions/telegram/src/bot-message-dispatch.ts
@@ -1383,6 +1383,9 @@ export const dispatchTelegramMessage = async ({
                     };
 
                     if (segments.length > 0) {
+                      if (info.kind === "final") {
+                        await flushBufferedFinalAnswer();
+                      }
                       trackBlockMedia(blockDelivered);
                       return;
                     }

Original file line number	Diff line number	Diff line change
`@@ -1383,6 +1383,9 @@ export const dispatchTelegramMessage = async ({`
`1383`	`1383`	`};`
`1384`	`1384`
`1385`	`1385`	`if (segments.length > 0) {`
	`1386`	`+ if (info.kind === "final") {`
	`1387`	`+ await flushBufferedFinalAnswer();`
	`1388`	`+ }`
`1386`	`1389`	`trackBlockMedia(blockDelivered);`
`1387`	`1390`	`return;`
`1388`	`1391`	`}`