fix(telegram): clear split reasoning previews

hugenshen · hugenshen · commit f123c42f3790 · 2026-06-18T23:38:15.000+08:00
diff --git a/extensions/telegram/src/bot-message-dispatch.test.ts b/extensions/telegram/src/bot-message-dispatch.test.ts
@@ -3435,6 +3435,36 @@ describe("dispatchTelegramMessage draft streaming", () => {
     expect(updates.join("\n")).not.toContain("CheckingReading");
   });
 
+  it("clears the prior reasoning preview before splitting into a new reasoning message", async () => {
+    const answerDraftStream = createDraftStream(2001);
+    const reasoningDraftStream = createTestDraftStream({
+      messageId: 3001,
+      clearMessageIdOnForceNew: true,
+    });
+    createTelegramDraftStream
+      .mockImplementationOnce(() => answerDraftStream)
+      .mockImplementationOnce(() => reasoningDraftStream);
+    dispatchReplyWithBufferedBlockDispatcher.mockImplementation(async ({ replyOptions }) => {
+      await replyOptions?.onReasoningStream?.({ text: "<think>First thought</think>" });
+      await replyOptions?.onReasoningEnd?.();
+      await replyOptions?.onReasoningStream?.({ text: "<think>Second thought</think>" });
+      return { queuedFinal: false };
+    });
+
+    await dispatchWithContext({ context: createReasoningStreamContext() });
+
+    expect(reasoningDraftStream.update).toHaveBeenCalledWith("Thinking\n\n_First thought_");
+    expect(reasoningDraftStream.update).toHaveBeenCalledWith("Thinking\n\n_Second thought_");
+    expect(reasoningDraftStream.forceNewMessage).toHaveBeenCalledTimes(1);
+    expect(reasoningDraftStream.clear).toHaveBeenCalledTimes(2);
+    expect(reasoningDraftStream.clear.mock.invocationCallOrder[0]).toBeLessThan(
+      reasoningDraftStream.forceNewMessage.mock.invocationCallOrder[0],
+    );
+    expect(reasoningDraftStream.clear.mock.invocationCallOrder[0]).toBeLessThan(
+      reasoningDraftStream.update.mock.invocationCallOrder[1],
+    );
+  });
+
   it("streams reasoning from configured defaults", async () => {
     const { answerDraftStream, reasoningDraftStream } = setupDraftStreams({
       answerMessageId: 2001,
diff --git a/extensions/telegram/src/bot-message-dispatch.ts b/extensions/telegram/src/bot-message-dispatch.ts
@@ -1203,6 +1203,18 @@ export const dispatchTelegramMessage = async ({
     lane.stream?.forceNewMessage();
     resetDraftLaneState(lane);
   };
+  const clearReasoningLaneForNewMessage = async () => {
+    if (
+      !reasoningLane.hasStreamedMessage &&
+      typeof reasoningLane.stream?.messageId() !== "number"
+    ) {
+      resetDraftLaneState(reasoningLane);
+      return;
+    }
+    await reasoningLane.stream?.clear();
+    reasoningLane.stream?.forceNewMessage();
+    resetDraftLaneState(reasoningLane);
+  };
   const rotateAnswerLaneForNewMessage = async () => {
     if (materializeAnswerLaneBeforeRotation) {
       await materializeAnswerLaneBeforeRotation();
@@ -2358,8 +2370,7 @@ export const dispatchTelegramMessage = async ({
                     ? (payload) =>
                         enqueueDraftLaneEvent(async () => {
                           if (splitReasoningOnNextStream) {
-                            reasoningLane.stream?.forceNewMessage();
-                            resetDraftLaneState(reasoningLane);
+                            await clearReasoningLaneForNewMessage();
                             splitReasoningOnNextStream = false;
                           }
                           await ingestDraftLaneSegments(payload, true);
