github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-27 00:33:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(telegram): restore DM draft streaming

Browse Source
This commit is contained in:
Ayaan Zaidi
2026-03-08 08:18:01 +05:30
committed by Ayaan Zaidi
parent 56cd0084d9
commit e45fcc57ed
2 changed files with 34 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/telegram/bot-message-dispatch.test.ts
View File

@@ -1171,7 +1171,7 @@ describe("dispatchTelegramMessage draft streaming", () => {
},
);
it("uses message preview transport for all DM lanes when streaming is active", async () => {
it("uses message preview transport for DM reasoning lane when answer preview lane is active", async () => {
setupDraftStreams({ answerMessageId: 999, reasoningMessageId: 111 });
dispatchReplyWithBufferedBlockDispatcher.mockImplementation(
async ({ dispatcherOptions, replyOptions }) => {
@@ -1187,12 +1187,10 @@ describe("dispatchTelegramMessage draft streaming", () => {
await dispatchWithContext({ context: createReasoningStreamContext(), streamMode: "partial" });
expect(createTelegramDraftStream).toHaveBeenCalledTimes(2);
// Both answer (call[0]) and reasoning (call[1]) lanes should use message
// transport in DMs to prevent duplicate messages. (Fixes #33453)
expect(createTelegramDraftStream.mock.calls[0]?.[0]).toEqual(
expect.objectContaining({
thread: { id: 777, scope: "dm" },
previewTransport: "message",
previewTransport: "auto",
}),
);
expect(createTelegramDraftStream.mock.calls[1]?.[0]).toEqual(
@@ -1203,6 +1201,35 @@ describe("dispatchTelegramMessage draft streaming", () => {
);
});
it("materializes DM answer draft final without sending a duplicate final message", async () => {
const answerDraftStream = createTestDraftStream({ previewMode: "draft" });
answerDraftStream.materialize.mockResolvedValue(321);
const reasoningDraftStream = createDraftStream(111);
createTelegramDraftStream
.mockImplementationOnce(() => answerDraftStream)
.mockImplementationOnce(() => reasoningDraftStream);
dispatchReplyWithBufferedBlockDispatcher.mockImplementation(
async ({ dispatcherOptions, replyOptions }) => {
await replyOptions?.onPartialReply?.({ text: "Checking the directory..." });
await dispatcherOptions.deliver({ text: "Checking the directory..." }, { kind: "final" });
return { queuedFinal: true };
},
);
deliverReplies.mockResolvedValue({ delivered: true });
await dispatchWithContext({ context: createContext(), streamMode: "partial" });
expect(createTelegramDraftStream.mock.calls[0]?.[0]).toEqual(
expect.objectContaining({
thread: { id: 777, scope: "dm" },
previewTransport: "auto",
}),
);
expect(answerDraftStream.materialize).toHaveBeenCalledTimes(1);
expect(deliverReplies).not.toHaveBeenCalled();
expect(editMessageTelegram).not.toHaveBeenCalled();
});
it("keeps reasoning and answer streaming in separate preview lanes", async () => {
const { answerDraftStream, reasoningDraftStream } = setupDraftStreams({
answerMessageId: 999,