@llblab/pi-telegram 0.2.9 → 0.2.10

package/tests/turns.test.ts

@@ -11,6 +11,7 @@ import {
   buildTelegramTurnPrompt,
   formatTelegramTurnStatusSummary,
   truncateTelegramQueueSummary,
+  updateTelegramPromptTurnText,
 } from "../lib/turns.ts";
 
 test("Turn helpers truncate queue summaries predictably", () => {
@@ -69,6 +70,120 @@ test("Turn helpers summarize text and attachment-only turns", () => {
   );
 });
 
+test("Turn helpers update queued prompt text for edited Telegram messages", () => {
+  const turn = {
+    kind: "prompt" as const,
+    chatId: 99,
+    replyToMessageId: 10,
+    sourceMessageIds: [10],
+    queueOrder: 1,
+    queueLane: "default" as const,
+    laneOrder: 1,
+    queuedAttachments: [],
+    content: [{ type: "text" as const, text: "[telegram] old" }],
+    historyText: "old",
+    statusSummary: "old",
+  };
+  const updated = updateTelegramPromptTurnText({
+    turn,
+    telegramPrefix: "[telegram]",
+    rawText: "new edited message",
+  });
+  assert.equal(updated.content[0]?.type, "text");
+  assert.equal(
+    (updated.content[0] as { type: "text"; text: string }).text,
+    "[telegram] new edited message",
+  );
+  assert.equal(updated.historyText, "new edited message");
+  assert.equal(updated.statusSummary, "new edited message");
+  assert.notEqual(updated, turn);
+});
+
+test("Turn helpers preserve queued prompt attachments when captions are edited", () => {
+  const turn = {
+    kind: "prompt" as const,
+    chatId: 99,
+    replyToMessageId: 10,
+    sourceMessageIds: [10],
+    queueOrder: 1,
+    queueLane: "default" as const,
+    laneOrder: 1,
+    queuedAttachments: [],
+    content: [
+      {
+        type: "text" as const,
+        text:
+          "[telegram] old caption\n\n" +
+          "Telegram attachments were saved locally:\n" +
+          "- /tmp/demo.png\n" +
+          "- /tmp/report.txt",
+      },
+      { type: "image" as const, data: "abc", mimeType: "image/png" },
+    ],
+    historyText: "old caption\nAttachments:\n- /tmp/demo.png\n- /tmp/report.txt",
+    statusSummary: "old caption",
+  };
+  const updated = updateTelegramPromptTurnText({
+    turn,
+    telegramPrefix: "[telegram]",
+    rawText: "new caption",
+  });
+  assert.equal(
+    (updated.content[0] as { type: "text"; text: string }).text,
+    "[telegram] new caption\n\n" +
+      "Telegram attachments were saved locally:\n" +
+      "- /tmp/demo.png\n" +
+      "- /tmp/report.txt",
+  );
+  assert.deepEqual(updated.content[1], turn.content[1]);
+  assert.equal(
+    updated.historyText,
+    "new caption\nAttachments:\n- /tmp/demo.png\n- /tmp/report.txt",
+  );
+  assert.equal(updated.statusSummary, "new caption");
+});
+
+test("Turn helpers preserve abort-history prompt context when queued turns are edited", () => {
+  const turn = {
+    kind: "prompt" as const,
+    chatId: 99,
+    replyToMessageId: 10,
+    sourceMessageIds: [10],
+    queueOrder: 1,
+    queueLane: "default" as const,
+    laneOrder: 1,
+    queuedAttachments: [],
+    content: [
+      {
+        type: "text" as const,
+        text:
+          "[telegram]\n\n" +
+          "Earlier Telegram messages arrived after an aborted turn. " +
+          "Treat them as prior user messages, in order:\n\n" +
+          "1. older Current Telegram message: quote\n\n" +
+          "Current Telegram message:\nold current",
+      },
+    ],
+    historyText: "old current",
+    statusSummary: "old current",
+  };
+  const updated = updateTelegramPromptTurnText({
+    turn,
+    telegramPrefix: "[telegram]",
+    rawText: "new current",
+  });
+  assert.equal(
+    (updated.content[0] as { type: "text"; text: string }).text,
+    "[telegram]\n\n" +
+      "Earlier Telegram messages arrived after an aborted turn. " +
+      "Treat them as prior user messages, in order:\n\n" +
+      "1. older Current Telegram message: quote\n\n" +
+      "Current Telegram message:\nnew current",
+  );
+  assert.equal(updated.historyText, "new current");
+  assert.equal(updated.statusSummary, "new current");
+});
+
 test("Turn helpers assemble prompt turns with text, ids, history, and image payloads", async () => {
   const turn = await buildTelegramPromptTurn({
     telegramPrefix: "[telegram]",

package/tests/updates.test.ts

@@ -15,6 +15,7 @@ import {
   executeTelegramUpdatePlan,
   extractDeletedTelegramMessageIds,
   getAuthorizedTelegramCallbackQuery,
+  getAuthorizedTelegramEditedMessage,
   getAuthorizedTelegramMessage,
   getTelegramAuthorizationState,
   normalizeTelegramReactionEmoji,
@@ -74,7 +75,7 @@ test("Update routing extracts only private human callback queries", () => {
   assert.ok(query);
 });
 
-test("Update routing extracts private human messages from message or edited_message", () => {
+test("Update routing extracts private human messages and edited messages separately", () => {
   assert.equal(
     getAuthorizedTelegramMessage({
       message: {
@@ -85,12 +86,19 @@ test("Update routing extracts private human messages from message or edited_mess
     undefined,
   );
   const directMessage = getAuthorizedTelegramMessage({
-    edited_message: {
+    message: {
       chat: { type: "private" },
       from: { id: 1, is_bot: false },
     },
   });
   assert.ok(directMessage);
+  const editedMessage = getAuthorizedTelegramEditedMessage({
+    edited_message: {
+      chat: { type: "private" },
+      from: { id: 1, is_bot: false },
+    },
+  });
+  assert.ok(editedMessage);
 });
 
 test("Update flow prioritizes deleted business-message handling over other update kinds", () => {
@@ -107,7 +115,7 @@ test("Update flow prioritizes deleted business-message handling over other updat
   assert.deepEqual(action, { kind: "deleted", messageIds: [1, 2] });
 });
 
-test("Update flow returns authorized callback and message actions", () => {
+test("Update flow returns authorized callback, message, and edit actions", () => {
   const callbackAction = buildTelegramUpdateFlowAction(
     {
       callback_query: {
@@ -135,6 +143,16 @@ test("Update flow returns authorized callback and message actions", () => {
     messageAction.kind === "message" ? messageAction.authorization : undefined,
     { kind: "pair", userId: 9 },
   );
+  const editAction = buildTelegramUpdateFlowAction(
+    {
+      edited_message: {
+        chat: { type: "private" },
+        from: { id: 9, is_bot: false },
+      },
+    },
+    9,
+  );
+  assert.equal(editAction.kind, "edited-message");
 });
 
 test("Update flow ignores unauthorized transport shapes and preserves reaction events", () => {
@@ -239,6 +257,7 @@ test("Update runtime executes delete and reaction plans through the right side e
       handleAuthorizedTelegramCallbackQuery: async () => {},
       sendTextReply: async () => undefined,
       handleAuthorizedTelegramMessage: async () => {},
+      handleAuthorizedTelegramEditedMessage: async () => {},
     },
   );
   assert.deepEqual(events, ["media:1,2", "queue:1,2"]);
@@ -273,6 +292,9 @@ test("Update runtime can execute directly from raw updates", async () => {
       handleAuthorizedTelegramMessage: async () => {
         events.push("message");
       },
+      handleAuthorizedTelegramEditedMessage: async () => {
+        events.push("edited-message");
+      },
     },
   );
   assert.deepEqual(events, [
@@ -282,6 +304,37 @@ test("Update runtime can execute directly from raw updates", async () => {
   ]);
 });
 
+test("Update runtime routes edited messages without creating normal message turns", async () => {
+  const events: string[] = [];
+  await executeTelegramUpdate(
+    {
+      edited_message: {
+        chat: { id: 10, type: "private" },
+        message_id: 20,
+        from: { id: 7, is_bot: false },
+      },
+    },
+    7,
+    {
+      ctx: {} as never,
+      removePendingMediaGroupMessages: () => {},
+      removeQueuedTelegramTurnsByMessageIds: () => 0,
+      handleAuthorizedTelegramReactionUpdate: async () => {},
+      pairTelegramUserIfNeeded: async () => false,
+      answerCallbackQuery: async () => {},
+      handleAuthorizedTelegramCallbackQuery: async () => {},
+      sendTextReply: async () => undefined,
+      handleAuthorizedTelegramMessage: async () => {
+        events.push("message");
+      },
+      handleAuthorizedTelegramEditedMessage: async () => {
+        events.push("edited-message");
+      },
+    },
+  );
+  assert.deepEqual(events, ["edited-message"]);
+});
+
 test("Update runtime handles callback deny and message pair flows", async () => {
   const events: string[] = [];
   await executeTelegramUpdatePlan(
@@ -317,6 +370,9 @@ test("Update runtime handles callback deny and message pair flows", async () =>
       handleAuthorizedTelegramMessage: async () => {
         events.push("message");
       },
+      handleAuthorizedTelegramEditedMessage: async () => {
+        events.push("edited-message");
+      },
     },
   );
   await executeTelegramUpdatePlan(
@@ -346,6 +402,9 @@ test("Update runtime handles callback deny and message pair flows", async () =>
       handleAuthorizedTelegramMessage: async () => {
         events.push("message");
       },
+      handleAuthorizedTelegramEditedMessage: async () => {
+        events.push("edited-message");
+      },
     },
   );
   assert.deepEqual(events, [