Show Telegram progress placeholder for suppressed drafts

gbarros-dev · gbarros-dev · commit ea471ba3724e · 2026-06-20T16:28:42.000+02:00
diff --git a/extensions/telegram/src/bot-message-dispatch.test.ts b/extensions/telegram/src/bot-message-dispatch.test.ts
@@ -2600,6 +2600,30 @@ describe("dispatchTelegramMessage draft streaming", () => {
     expect(editMessageTelegram).not.toHaveBeenCalled();
   });
 
+  it("shows a progress placeholder when progress mode suppresses answer partial text", async () => {
+    const draftStream = createSequencedDraftStream(2001);
+    createTelegramDraftStream.mockReturnValue(draftStream);
+    dispatchReplyWithBufferedBlockDispatcher.mockImplementation(async ({ replyOptions }) => {
+      await replyOptions?.onPartialReply?.({ text: "Short draft" });
+      return { queuedFinal: false };
+    });
+
+    await dispatchWithContext({
+      context: createContext(),
+      streamMode: "progress",
+      telegramCfg: {
+        streaming: {
+          mode: "progress",
+          progress: { label: "Working", toolProgress: false },
+        },
+      },
+    });
+
+    expect(draftStream.updatePreview).toHaveBeenCalledWith(telegramHtmlPreview("<b>Working</b>"));
+    expect(draftStream.update).not.toHaveBeenCalledWith("Short draft");
+    expect(draftStream.flush).toHaveBeenCalled();
+  });
+
   it("replaces Telegram command progress items with matching command output", async () => {
     const { answerDraftStream } = setupDraftStreams({ answerMessageId: 2001 });
     dispatchReplyWithBufferedBlockDispatcher.mockImplementation(async ({ replyOptions }) => {
diff --git a/extensions/telegram/src/bot-message-dispatch.ts b/extensions/telegram/src/bot-message-dispatch.ts
@@ -1364,7 +1364,7 @@ export const dispatchTelegramMessage = async ({
       recomputeQueuedAnswerBlockRotations();
     }
   };
-  const updateDraftFromPartial = (lane: DraftLaneState, update: DraftPartialTextUpdate) => {
+  const updateDraftFromPartial = async (lane: DraftLaneState, update: DraftPartialTextUpdate) => {
     const laneStream = lane.stream;
     if (!laneStream || !update.text) {
       return;
@@ -1376,6 +1376,7 @@ export const dispatchTelegramMessage = async ({
     }
     if (lane === answerLane) {
       if (streamMode === "progress") {
+        await progressDraft.start();
         return;
       }
       resetAnswerToolProgressDraft();
@@ -1402,7 +1403,7 @@ export const dispatchTelegramMessage = async ({
         reasoningStepState.noteReasoningHint();
         reasoningStepState.noteReasoningDelivered();
       }
-      updateDraftFromPartial(lanes[segment.lane], segment.update);
+      await updateDraftFromPartial(lanes[segment.lane], segment.update);
     }
   };
   const flushDraftLane = async (lane: DraftLaneState) => {

Original file line number	Diff line number	Diff line change
`@@ -1364,7 +1364,7 @@ export const dispatchTelegramMessage = async ({`
`1364`	`1364`	`recomputeQueuedAnswerBlockRotations();`
`1365`	`1365`	`}`
`1366`	`1366`	`};`
`1367`		`- const updateDraftFromPartial = (lane: DraftLaneState, update: DraftPartialTextUpdate) => {`
	`1367`	`+ const updateDraftFromPartial = async (lane: DraftLaneState, update: DraftPartialTextUpdate) => {`
`1368`	`1368`	`const laneStream = lane.stream;`
`1369`	`1369`	`if (!laneStream \|\| !update.text) {`
`1370`	`1370`	`return;`
`@@ -1376,6 +1376,7 @@ export const dispatchTelegramMessage = async ({`
`1376`	`1376`	`}`
`1377`	`1377`	`if (lane === answerLane) {`
`1378`	`1378`	`if (streamMode === "progress") {`
	`1379`	`+ await progressDraft.start();`
`1379`	`1380`	`return;`
`1380`	`1381`	`}`
`1381`	`1382`	`resetAnswerToolProgressDraft();`
`@@ -1402,7 +1403,7 @@ export const dispatchTelegramMessage = async ({`
`1402`	`1403`	`reasoningStepState.noteReasoningHint();`
`1403`	`1404`	`reasoningStepState.noteReasoningDelivered();`
`1404`	`1405`	`}`
`1405`		`- updateDraftFromPartial(lanes[segment.lane], segment.update);`
	`1406`	`+ await updateDraftFromPartial(lanes[segment.lane], segment.update);`
`1406`	`1407`	`}`
`1407`	`1408`	`};`
`1408`	`1409`	`const flushDraftLane = async (lane: DraftLaneState) => {`