fix telegram partial draft previews

Andy Ye · Andy Ye · commit 977c2fdbfbbc · 2026-05-09T00:25:46.000-07:00
diff --git a/extensions/telegram/src/bot-message-dispatch.test.ts b/extensions/telegram/src/bot-message-dispatch.test.ts
@@ -442,7 +442,7 @@ describe("dispatchTelegramMessage draft streaming", () => {
       expect.objectContaining({
         chatId: 123,
         thread: { id: 777, scope: "dm" },
-        minInitialChars: 30,
+        minInitialChars: 0,
       }),
     );
     expect(draftStream.update).toHaveBeenCalledWith("Hello");
@@ -468,6 +468,41 @@ describe("dispatchTelegramMessage draft streaming", () => {
     expect(draftStream.clear).toHaveBeenCalledTimes(1);
   });
 
+  it("sends direct model partials through the Telegram draft transport before final delivery", async () => {
+    const { createTelegramDraftStream: createRealTelegramDraftStream } =
+      await vi.importActual<typeof import("./draft-stream.js")>("./draft-stream.js");
+    const bot = createBot();
+    createTelegramDraftStream.mockImplementation((params) => createRealTelegramDraftStream(params));
+    dispatchReplyWithBufferedBlockDispatcher.mockImplementation(
+      async ({ dispatcherOptions, replyOptions }) => {
+        await replyOptions?.onPartialReply?.({ text: "Hi" });
+        await vi.waitFor(() =>
+          expect(bot.api.sendMessage).toHaveBeenCalledWith(
+            123,
+            "Hi",
+            expect.objectContaining({ message_thread_id: 777 }),
+          ),
+        );
+        expect(deliverReplies).not.toHaveBeenCalled();
+
+        await dispatcherOptions.deliver({ text: "Hi there" }, { kind: "final" });
+        return { queuedFinal: true };
+      },
+    );
+
+    await dispatchWithContext({ context: createContext(), bot });
+
+    await vi.waitFor(() =>
+      expect(bot.api.editMessageText).toHaveBeenCalledWith(
+        123,
+        777,
+        "Hi there",
+        expect.objectContaining({ parse_mode: "HTML" }),
+      ),
+    );
+    expect(deliverReplies).not.toHaveBeenCalled();
+  });
+
   it("keeps retained overflow draft previews", async () => {
     const draftStream = createDraftStream();
     const bot = createBot();
diff --git a/extensions/telegram/src/bot-message-dispatch.ts b/extensions/telegram/src/bot-message-dispatch.ts
@@ -428,7 +428,7 @@ export const dispatchTelegramMessage = async ({
     replyToMode !== "off" && typeof msg.message_id === "number"
       ? (replyQuoteMessageId ?? msg.message_id)
       : undefined;
-  const draftMinInitialChars = streamMode === "progress" ? 0 : DRAFT_MIN_INITIAL_CHARS;
+  const draftMinInitialChars = streamMode === "block" ? DRAFT_MIN_INITIAL_CHARS : 0;
   const progressSeed = `${route.accountId}:${chatId}:${threadSpec.id ?? ""}`;
   const mediaLocalRoots = getAgentScopedMediaLocalRoots(cfg, route.agentId);
   const createDraftLane = (laneName: LaneName, enabled: boolean): DraftLaneState => {
