@poncho-ai/messaging 0.3.0 → 0.5.0

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @poncho-ai/messaging@0.3.0 build /Users/cesar/Dev/latitude/poncho-ai/packages/messaging
+> @poncho-ai/messaging@0.5.0 build /Users/cesar/Dev/latitude/poncho-ai/packages/messaging
 > tsup src/index.ts --format esm --dts
 
 CLI Building entry: src/index.ts
@@ -7,8 +7,8 @@
 CLI tsup v8.5.1
 CLI Target: es2022
 ESM Build start
-ESM dist/index.js 44.67 KB
-ESM ⚡️ Build success in 55ms
+ESM dist/index.js 45.94 KB
+ESM ⚡️ Build success in 60ms
 DTS Build start
-DTS ⚡️ Build success in 3450ms
-DTS dist/index.d.ts 9.98 KB
+DTS ⚡️ Build success in 2033ms
+DTS dist/index.d.ts 10.13 KB

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @poncho-ai/messaging
 
+## 0.5.0
+
+### Minor Changes
+
+- [`8ef9316`](https://github.com/cesr/poncho-ai/commit/8ef93165084b4df581e39a1581b1ead64b7b3f42) Thanks [@cesr](https://github.com/cesr)! - Add auto-continuation support for messaging adapters (Telegram, Slack, Resend) on serverless platforms. When `PONCHO_MAX_DURATION` is set, agent runs that hit the soft deadline now automatically resume with "Continue" messages, matching the web UI and client SDK behavior.
+
+## 0.4.0
+
+### Minor Changes
+
+- [`3216e80`](https://github.com/cesr/poncho-ai/commit/3216e8072027896dd1cc5f29b1a7b0eea9ee1ff5) Thanks [@cesr](https://github.com/cesr)! - Add `allowedUserIds` option to Telegram adapter for restricting bot access to specific users.
+
 ## 0.3.0
 
 ### Minor Changes

package/dist/index.d.ts

@@ -83,6 +83,9 @@ interface AgentRunner {
     }): Promise<{
         response: string;
         files?: FileAttachment[];
+        continuation?: boolean;
+        steps?: number;
+        maxSteps?: number;
     }>;
 }
 interface AgentBridgeOptions {
@@ -192,6 +195,7 @@ declare class ResendAdapter implements MessagingAdapter {
 interface TelegramAdapterOptions {
     botTokenEnv?: string;
     webhookSecretEnv?: string;
+    allowedUserIds?: number[];
 }
 declare class TelegramAdapter implements MessagingAdapter {
     readonly platform: "telegram";
@@ -203,6 +207,7 @@ declare class TelegramAdapter implements MessagingAdapter {
     private botId;
     private readonly botTokenEnv;
     private readonly webhookSecretEnv;
+    private readonly allowedUserIds;
     private handler;
     private readonly sessionCounters;
     private lastUpdateId;

package/dist/index.js

@@ -33,6 +33,7 @@ var AgentBridge = class {
     await this.adapter.initialize();
   }
   async handleMessage(message) {
+    const MAX_CONTINUATIONS = 10;
     let cleanup;
     this.adapter.resetRequestState?.();
     try {
@@ -46,33 +47,53 @@ var AgentBridge = class {
       const titleParts = [platformTag, senderLabel];
       if (message.subject) titleParts.push(message.subject);
       const title = titleParts.join(" ") || `${message.platform} thread`;
-      const conversation = await this.runner.getOrCreateConversation(
-        conversationId,
-        {
-          platform: message.platform,
-          ownerId: this.ownerIdOverride ?? message.sender.id,
-          title
-        }
-      );
       const senderLine = message.sender.name ? `From: ${message.sender.name} <${message.sender.id}>` : `From: ${message.sender.id}`;
       const subjectLine = message.subject ? `Subject: ${message.subject}` : "";
       const header = [senderLine, subjectLine].filter(Boolean).join("\n");
-      const task = `${header}
+      let currentTask = `${header}
 
 ${message.text}`;
-      const result = await this.runner.run(conversationId, {
-        task,
-        messages: conversation.messages,
-        files: message.files,
-        metadata: {
-          platform: message.platform,
-          sender: message.sender,
-          threadId: message.threadRef.platformThreadId
+      let currentFiles = message.files;
+      let accumulatedResponse = "";
+      let accumulatedFiles = [];
+      let totalSteps = 0;
+      let stepBudget = 0;
+      let continuationCount = 0;
+      while (true) {
+        const conversation = await this.runner.getOrCreateConversation(
+          conversationId,
+          {
+            platform: message.platform,
+            ownerId: this.ownerIdOverride ?? message.sender.id,
+            title
+          }
+        );
+        const result = await this.runner.run(conversationId, {
+          task: currentTask,
+          messages: conversation.messages,
+          files: currentFiles,
+          metadata: {
+            platform: message.platform,
+            sender: message.sender,
+            threadId: message.threadRef.platformThreadId
+          }
+        });
+        accumulatedResponse += result.response;
+        if (result.files) accumulatedFiles.push(...result.files);
+        totalSteps += result.steps ?? 0;
+        if (typeof result.maxSteps === "number") stepBudget = result.maxSteps;
+        if (result.continuation && continuationCount < MAX_CONTINUATIONS && (stepBudget <= 0 || totalSteps < stepBudget)) {
+          continuationCount++;
+          console.log(`[agent-bridge] continuation ${continuationCount}/${MAX_CONTINUATIONS} for ${conversationId}`);
+          currentTask = "Continue";
+          currentFiles = void 0;
+          continue;
         }
-      });
+        break;
+      }
       if (this.adapter.autoReply) {
-        await this.adapter.sendReply(message.threadRef, result.response, {
-          files: result.files
+        await this.adapter.sendReply(message.threadRef, accumulatedResponse, {
+          files: accumulatedFiles.length > 0 ? accumulatedFiles : void 0
         });
       } else if (!this.adapter.hasSentInCurrentRequest) {
         console.warn("[agent-bridge] tool mode completed without send_email being called; no reply sent");
@@ -1136,12 +1157,14 @@ var TelegramAdapter = class {
   botId = 0;
   botTokenEnv;
   webhookSecretEnv;
+  allowedUserIds;
   handler;
   sessionCounters = /* @__PURE__ */ new Map();
   lastUpdateId = 0;
   constructor(options = {}) {
     this.botTokenEnv = options.botTokenEnv ?? "TELEGRAM_BOT_TOKEN";
     this.webhookSecretEnv = options.webhookSecretEnv ?? "TELEGRAM_WEBHOOK_SECRET";
+    this.allowedUserIds = options.allowedUserIds && options.allowedUserIds.length > 0 ? options.allowedUserIds : void 0;
   }
   // -----------------------------------------------------------------------
   // MessagingAdapter implementation
@@ -1247,6 +1270,13 @@ var TelegramAdapter = class {
       res.end();
       return;
     }
+    if (this.allowedUserIds && message.from) {
+      if (!this.allowedUserIds.includes(message.from.id)) {
+        res.writeHead(200);
+        res.end();
+        return;
+      }
+    }
     const chatId = String(message.chat.id);
     const chatType = message.chat.type;
     const isGroup = chatType === "group" || chatType === "supergroup";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@poncho-ai/messaging",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "description": "Messaging platform adapters for Poncho agents (Slack, Telegram, etc.)",
   "repository": {
     "type": "git",

package/src/adapters/telegram/index.ts

@@ -29,6 +29,7 @@ const NEW_COMMAND_RE = /^\/new(?:@(\S+))?$/i;
 export interface TelegramAdapterOptions {
   botTokenEnv?: string;
   webhookSecretEnv?: string;
+  allowedUserIds?: number[];
 }
 
 const collectBody = (req: http.IncomingMessage): Promise<string> =>
@@ -50,6 +51,7 @@ export class TelegramAdapter implements MessagingAdapter {
   private botId = 0;
   private readonly botTokenEnv: string;
   private readonly webhookSecretEnv: string;
+  private readonly allowedUserIds: number[] | undefined;
   private handler: IncomingMessageHandler | undefined;
   private readonly sessionCounters = new Map<string, number>();
   private lastUpdateId = 0;
@@ -58,6 +60,10 @@ export class TelegramAdapter implements MessagingAdapter {
     this.botTokenEnv = options.botTokenEnv ?? "TELEGRAM_BOT_TOKEN";
     this.webhookSecretEnv =
       options.webhookSecretEnv ?? "TELEGRAM_WEBHOOK_SECRET";
+    this.allowedUserIds =
+      options.allowedUserIds && options.allowedUserIds.length > 0
+        ? options.allowedUserIds
+        : undefined;
   }
 
   // -----------------------------------------------------------------------
@@ -197,6 +203,15 @@ export class TelegramAdapter implements MessagingAdapter {
       return;
     }
 
+    // -- User allowlist -----------------------------------------------------
+    if (this.allowedUserIds && message.from) {
+      if (!this.allowedUserIds.includes(message.from.id)) {
+        res.writeHead(200);
+        res.end();
+        return;
+      }
+    }
+
     const chatId = String(message.chat.id);
     const chatType = message.chat.type;
     const isGroup = chatType === "group" || chatType === "supergroup";

package/src/bridge.ts

@@ -1,6 +1,7 @@
 import { createHash } from "node:crypto";
 import type {
   AgentBridgeOptions,
+  FileAttachment,
   IncomingMessage,
   MessagingAdapter,
   AgentRunner,
@@ -53,6 +54,7 @@ export class AgentBridge {
   }
 
   private async handleMessage(message: IncomingMessage): Promise<void> {
+    const MAX_CONTINUATIONS = 10;
     let cleanup: (() => Promise<void>) | undefined;
 
     this.adapter.resetRequestState?.();
@@ -71,36 +73,64 @@ export class AgentBridge {
       if (message.subject) titleParts.push(message.subject);
       const title = titleParts.join(" ") || `${message.platform} thread`;
 
-      const conversation = await this.runner.getOrCreateConversation(
-        conversationId,
-        {
-          platform: message.platform,
-          ownerId: this.ownerIdOverride ?? message.sender.id,
-          title,
-        },
-      );
-
       const senderLine = message.sender.name
         ? `From: ${message.sender.name} <${message.sender.id}>`
         : `From: ${message.sender.id}`;
       const subjectLine = message.subject ? `Subject: ${message.subject}` : "";
       const header = [senderLine, subjectLine].filter(Boolean).join("\n");
-      const task = `${header}\n\n${message.text}`;
-
-      const result = await this.runner.run(conversationId, {
-        task,
-        messages: conversation.messages,
-        files: message.files,
-        metadata: {
-          platform: message.platform,
-          sender: message.sender,
-          threadId: message.threadRef.platformThreadId,
-        },
-      });
+
+      let currentTask = `${header}\n\n${message.text}`;
+      let currentFiles = message.files;
+      let accumulatedResponse = "";
+      let accumulatedFiles: FileAttachment[] = [];
+      let totalSteps = 0;
+      let stepBudget = 0;
+      let continuationCount = 0;
+
+      while (true) {
+        const conversation = await this.runner.getOrCreateConversation(
+          conversationId,
+          {
+            platform: message.platform,
+            ownerId: this.ownerIdOverride ?? message.sender.id,
+            title,
+          },
+        );
+
+        const result = await this.runner.run(conversationId, {
+          task: currentTask,
+          messages: conversation.messages,
+          files: currentFiles,
+          metadata: {
+            platform: message.platform,
+            sender: message.sender,
+            threadId: message.threadRef.platformThreadId,
+          },
+        });
+
+        accumulatedResponse += result.response;
+        if (result.files) accumulatedFiles.push(...result.files);
+        totalSteps += result.steps ?? 0;
+        if (typeof result.maxSteps === "number") stepBudget = result.maxSteps;
+
+        if (
+          result.continuation &&
+          continuationCount < MAX_CONTINUATIONS &&
+          (stepBudget <= 0 || totalSteps < stepBudget)
+        ) {
+          continuationCount++;
+          console.log(`[agent-bridge] continuation ${continuationCount}/${MAX_CONTINUATIONS} for ${conversationId}`);
+          currentTask = "Continue";
+          currentFiles = undefined;
+          continue;
+        }
+
+        break;
+      }
 
       if (this.adapter.autoReply) {
-        await this.adapter.sendReply(message.threadRef, result.response, {
-          files: result.files,
+        await this.adapter.sendReply(message.threadRef, accumulatedResponse, {
+          files: accumulatedFiles.length > 0 ? accumulatedFiles : undefined,
         });
       } else if (!this.adapter.hasSentInCurrentRequest) {
         console.warn("[agent-bridge] tool mode completed without send_email being called; no reply sent");

package/src/types.ts

@@ -123,6 +123,9 @@ export interface AgentRunner {
   ): Promise<{
     response: string;
     files?: FileAttachment[];
+    continuation?: boolean;
+    steps?: number;
+    maxSteps?: number;
   }>;
 }