fix(telegram): preserve buttons on deduped preview finals

Elarwei001 · Elarwei001 · commit 630ac2712b77 · 2026-05-19T21:08:50.000+08:00
diff --git a/extensions/telegram/src/bot-message-dispatch.test.ts b/extensions/telegram/src/bot-message-dispatch.test.ts
@@ -565,6 +565,31 @@ describe("dispatchTelegramMessage draft streaming", () => {
     expect(answerDraftStream.clear).not.toHaveBeenCalled();
   });
 
+  it("attaches buttons before deduping a preview-streamed final", async () => {
+    const { answerDraftStream } = setupDraftStreams({ answerMessageId: 2001 });
+    const buttons = [[{ text: "OK", callback_data: "ok" }]];
+    dispatchReplyWithBufferedBlockDispatcher.mockImplementation(
+      async ({ dispatcherOptions, replyOptions }) => {
+        await replyOptions?.onPartialReply?.({ text: "Choose" });
+        await dispatcherOptions.deliver(
+          { text: "Choose", channelData: { telegram: { buttons } } },
+          { kind: "final" },
+        );
+        return { queuedFinal: true };
+      },
+    );
+
+    await dispatchWithContext({ context: createContext() });
+
+    expect(answerDraftStream.update).toHaveBeenCalledWith("Choose");
+    expect(mockCallArg(editMessageTelegram)).toBe(123);
+    expect(mockCallArg(editMessageTelegram, 0, 1)).toBe(2001);
+    expect(mockCallArg(editMessageTelegram, 0, 2)).toBe("Choose");
+    expectRecordFields(mockCallArg(editMessageTelegram, 0, 3), { buttons });
+    expect(deliverReplies).not.toHaveBeenCalled();
+    expect(answerDraftStream.clear).not.toHaveBeenCalled();
+  });
+
   it("dedups each per-block final against a multi-block preview", async () => {
     // Preview streams "First block\n\nSecond block" once; the reply pipeline
     // then emits two separate final payloads ("First block" and "Second block")
diff --git a/extensions/telegram/src/bot-message-dispatch.ts b/extensions/telegram/src/bot-message-dispatch.ts
@@ -1345,6 +1345,7 @@ export const dispatchTelegramMessage = async ({
                       return;
                     }
                     const effectivePayload = deduped;
+                    const telegramButtons = resolvePayloadTelegramInlineButtons(effectivePayload);
 
                     // Preview-streamed final dedup (channel-layer replacement
                     // for the previous core-layer suppression introduced by
@@ -1371,6 +1372,26 @@ export const dispatchTelegramMessage = async ({
                           answerLane.lastPartialText,
                         );
                         if (isPreviewStreamedText(effectivePayload.text, previewDedupeText)) {
+                          if (telegramButtons) {
+                            try {
+                              await (telegramDeps.editMessageTelegram ?? editMessageTelegram)(
+                                chatId,
+                                previewMessageId,
+                                effectivePayload.text ?? answerLane.lastPartialText,
+                                {
+                                  api: bot.api,
+                                  cfg,
+                                  accountId: route.accountId,
+                                  linkPreview: telegramCfg.linkPreview,
+                                  buttons: telegramButtons,
+                                },
+                              );
+                            } catch (err) {
+                              logVerbose(
+                                `telegram: preview-finalized button edit failed: ${formatErrorMessage(err)}`,
+                              );
+                            }
+                          }
                           answerLane.finalized = true;
                           deliveryState.markDelivered();
                           emitPreviewFinalizedHook({
@@ -1399,7 +1420,6 @@ export const dispatchTelegramMessage = async ({
                       queuedFinal = true;
                       return;
                     }
-                    const telegramButtons = resolvePayloadTelegramInlineButtons(effectivePayload);
                     const split = splitTextIntoLaneSegments(
                       { text: effectivePayload.text },
                       payload.isReasoning,
