aamp-openclaw-plugin 0.1.36 → 0.1.37

package/README.md

@@ -38,6 +38,7 @@ npm run build
         "enabled": true,
         "config": {
           "aampHost": "https://meshmail.ai",
+          "taskDispatchConcurrency": 10,
           "slug": "openclaw-agent",
           "credentialsFile": "~/.openclaw/extensions/aamp-openclaw-plugin/.credentials.json",
           "senderPolicies": [
@@ -57,6 +58,7 @@ npm run build
 ```
 
 If `senderPolicies` is omitted, all senders are accepted. If set, the dispatch sender must match one policy and all configured dispatch-context rules for that sender must pass.
+`taskDispatchConcurrency` is optional and defaults to `10`.
 
 The plugin also understands:
 

package/dist/index.js

@@ -421,6 +421,17 @@ var TinyEmitter = class {
     }
     return true;
   }
+  async emitAsync(event, ...args) {
+    const bucket = this.listeners.get(event);
+    if (!bucket || bucket.size === 0)
+      return false;
+    const settled = await Promise.allSettled([...bucket].map((listener) => Promise.resolve(listener(...args))));
+    const rejected = settled.find((result) => result.status === "rejected");
+    if (rejected) {
+      throw rejected.reason;
+    }
+    return true;
+  }
 };
 
 // ../sdk/src/jmap-push.js
@@ -1897,14 +1908,27 @@ function buildRegisteredCommandDispatchPayload(opts) {
     stream: { mode: opts.streamMode ?? "full" }
   };
 }
+var DEFAULT_TASK_DISPATCH_CONCURRENCY = 10;
+function normalizeTaskDispatchConcurrency(value) {
+  if (value == null)
+    return DEFAULT_TASK_DISPATCH_CONCURRENCY;
+  if (!Number.isFinite(value) || !Number.isInteger(value) || value < 1) {
+    throw new Error("taskDispatchConcurrency must be a positive integer");
+  }
+  return value;
+}
 var AampClient = class _AampClient extends TinyEmitter {
   jmapClient;
   smtpSender;
   config;
+  taskDispatchConcurrency;
+  pendingTaskDispatches = [];
+  activeTaskDispatchCount = 0;
   streamAppendQueues = /* @__PURE__ */ new Map();
   constructor(config) {
     super();
     this.config = config;
+    this.taskDispatchConcurrency = normalizeTaskDispatchConcurrency(config.taskDispatchConcurrency);
     const mailboxToken = config.mailboxToken;
     const resolvedBaseUrl = config.baseUrl;
     const derived = deriveMailboxServiceDefaults(config.email, resolvedBaseUrl);
@@ -1940,7 +1964,7 @@ var AampClient = class _AampClient extends TinyEmitter {
       rejectUnauthorized: config.rejectUnauthorized
     });
     this.jmapClient.on("task.dispatch", (task) => {
-      this.emit("task.dispatch", task);
+      this.enqueueTaskDispatch(task);
     });
     this.jmapClient.on("task.cancel", (task) => {
       this.emit("task.cancel", task);
@@ -1993,6 +2017,7 @@ var AampClient = class _AampClient extends TinyEmitter {
       smtpPort: config.smtpPort ?? 587,
       smtpPassword: config.smtpPassword,
       reconnectInterval: config.reconnectInterval,
+      taskDispatchConcurrency: config.taskDispatchConcurrency,
       rejectUnauthorized: config.rejectUnauthorized
     });
   }
@@ -2246,6 +2271,30 @@ var AampClient = class _AampClient extends TinyEmitter {
     }
     return res.json();
   }
+  enqueueTaskDispatch(task) {
+    this.pendingTaskDispatches.push(task);
+    this.drainTaskDispatchQueue();
+  }
+  drainTaskDispatchQueue() {
+    while (this.activeTaskDispatchCount < this.taskDispatchConcurrency && this.pendingTaskDispatches.length > 0) {
+      const nextTask = this.pendingTaskDispatches.shift();
+      if (!nextTask)
+        return;
+      this.activeTaskDispatchCount += 1;
+      void this.runTaskDispatch(nextTask);
+    }
+  }
+  async runTaskDispatch(task) {
+    try {
+      await this.emitAsync("task.dispatch", task);
+    } catch (err) {
+      const error = err instanceof Error ? err : new Error(String(err));
+      this.emit("error", error);
+    } finally {
+      this.activeTaskDispatchCount = Math.max(0, this.activeTaskDispatchCount - 1);
+      this.drainTaskDispatchQueue();
+    }
+  }
   getStreamAppendQueue(streamId) {
     let queue = this.streamAppendQueues.get(streamId);
     if (!queue) {
@@ -2706,6 +2755,19 @@ function isSyntheticPendingKey(taskKey) {
 function isTaskAwaitingHelpReply(task) {
   return task.awaitingHelpReply === true;
 }
+function isConversationalTask(task) {
+  return task.dispatchContext?.source === "feishu";
+}
+function firstDispatchContextValue(context, keys) {
+  if (!context)
+    return void 0;
+  for (const key of keys) {
+    const value = context[key]?.trim();
+    if (value)
+      return value;
+  }
+  return void 0;
+}
 function threadAlreadyTerminal(events) {
   return (events ?? []).some(
     (event) => event.intent === "task.result" || event.intent === "task.cancel"
@@ -2752,12 +2814,36 @@ function buildOpenClawMainSessionKey(mainKey, config) {
 function buildAampConversationSessionKey(value, config) {
   return buildOpenClawMainSessionKey(`${AAMP_SESSION_PREFIX}default:${value}`, config);
 }
+function buildAampStickySessionKey(dispatchContext, config) {
+  const stickyValue = firstDispatchContextValue(dispatchContext, ["session_key", "conversation_key", "thread_key"]);
+  if (!stickyValue)
+    return void 0;
+  return buildAampConversationSessionKey(`session:${stickyValue}`, config);
+}
 function buildAampTaskSessionKey(taskId, config) {
   return buildAampConversationSessionKey(`task:${taskId}`, config);
 }
 function buildAampWakeSessionKey(kind, id) {
   return `${AAMP_SESSION_PREFIX}wake:${kind}:${id}`;
 }
+function buildSessionKeyForPendingTask(task, config) {
+  return buildAampStickySessionKey(task.dispatchContext, config) ?? buildAampTaskSessionKey(task.taskId, config);
+}
+function buildWakeSessionKeyForPendingTask(task, config) {
+  return buildAampStickySessionKey(task.dispatchContext, config) ?? buildAampWakeSessionKey("task", task.taskId);
+}
+function findPendingEntryForSession(sessionKey, config) {
+  if (typeof sessionKey !== "string" || !isAampSessionKey(sessionKey))
+    return void 0;
+  const requested = buildOpenClawMainSessionKey(stripOpenClawAgentScope(sessionKey), config);
+  const entries = [...pendingTasks.entries()].filter(([key, task]) => isActionablePendingTask(key, task)).filter(([, task]) => buildSessionKeyForPendingTask(task, config) === requested).sort((a, b) => {
+    const rankDiff = priorityRank(a[1].priority) - priorityRank(b[1].priority);
+    if (rankDiff !== 0)
+      return rankDiff;
+    return new Date(a[1].receivedAt).getTime() - new Date(b[1].receivedAt).getTime();
+  });
+  return entries[0];
+}
 function resolvePendingKeyFromSessionKey(sessionKey) {
   if (typeof sessionKey !== "string")
     return void 0;
@@ -2848,6 +2934,7 @@ function queuePendingTask(task) {
     from: task.from,
     title: task.title,
     bodyText: task.bodyText ?? "",
+    dispatchContext: task.dispatchContext,
     threadHistory: task.threadHistory ?? [],
     threadContextText: task.threadContextText ?? "",
     priority: task.priority ?? "normal",
@@ -3030,8 +3117,8 @@ var src_default = {
       api.logger.info(`[AAMP] Directory profile synced${cardText ? " (card text registered)" : ""}`);
     }
     function wakeAgentForPendingTask(task) {
-      const fallbackSessionKey = buildAampWakeSessionKey("task", task.taskId);
-      const openClawSessionKey = buildAampTaskSessionKey(task.taskId, api.config);
+      const fallbackSessionKey = buildWakeSessionKeyForPendingTask(task, api.config);
+      const openClawSessionKey = buildSessionKeyForPendingTask(task, api.config);
       const fallback = () => triggerHeartbeatWake(fallbackSessionKey, `task ${task.taskId}`);
       const dispatcher = channelRuntime?.reply?.dispatchReplyWithBufferedBlockDispatcher;
       api.logger.info(
@@ -3117,13 +3204,14 @@ var src_default = {
         email: identity.email,
         smtpPassword: identity.smtpPassword,
         baseUrl: base,
+        taskDispatchConcurrency: cfg.taskDispatchConcurrency,
         // Local/dev: management-service proxy uses plain HTTP, no TLS cert to verify.
         // Production: set to true when using wss:// with valid certs.
         rejectUnauthorized: false
       });
       aampClient.on("task.dispatch", (task) => {
         api.logger.info(`[AAMP] \u2190 task.dispatch  ${task.taskId}  "${task.title}"  from=${task.from}`);
-        void (async () => {
+        return (async () => {
           try {
             if (terminalTaskIds.has(task.taskId)) {
               api.logger.info(`[AAMP] Skipping already-terminal task ${task.taskId}`);
@@ -3167,7 +3255,7 @@ var src_default = {
           } catch (err) {
             api.logger.error(`[AAMP] task.dispatch handler failed for ${task.taskId}: ${err.message}`);
             if (pendingTasks.has(task.taskId)) {
-              triggerHeartbeatWake(buildAampWakeSessionKey("task", task.taskId), `task ${task.taskId}`);
+              triggerHeartbeatWake(buildWakeSessionKeyForPendingTask(pendingTasks.get(task.taskId), api.config), `task ${task.taskId}`);
             }
           }
         })();
@@ -3567,7 +3655,8 @@ ${notifyBody?.bodyText ?? help.question}`;
           }
           return [targetedPendingKey, targetedTask];
         })() : void 0;
-        const nextEntry = targetedPendingKey ? targetedEntry : nextPendingEntry();
+        const sessionScopedEntry = targetedPendingKey ? void 0 : findPendingEntryForSession(ctx?.sessionKey, api.config);
+        const nextEntry = targetedPendingKey ? targetedEntry : sessionScopedEntry ?? nextPendingEntry();
         if (!nextEntry)
           return {};
         const [taskKey, task] = nextEntry;
@@ -3602,19 +3691,45 @@ ${notifyBody?.bodyText ?? help.question}`;
             `   Example: attachments: [{ filename: "file.html", path: "/tmp/aamp-files/file.html" }]`
           ] : []
         ].join("\n") : "";
-        const lines = isNotification ? [
-          `## Sub-task Update`,
+        const dispatchContextLines = task.dispatchContext && Object.keys(task.dispatchContext).length > 0 ? `Dispatch Context:
+${Object.entries(task.dispatchContext).map(([key, value]) => `  - ${key}: ${value}`).join("\n")}` : "";
+        const taskPromptLines = isConversationalTask(task) ? [
+          `## Pending AAMP Conversation Turn`,
           ``,
-          `A sub-task you dispatched has returned a result. Review the information below.`,
-          `If the sub-task included attachments, use aamp_download_attachment to fetch them.`,
+          `This AAMP task came from a chat surface (${task.dispatchContext?.source ?? "unknown"}).`,
+          `Treat it as an ongoing conversation turn, not a one-off work order.`,
+          `Your job is to reply naturally to the user's latest message and keep the conversation moving.`,
+          ``,
+          `### Tool selection rules for chat turns:`,
+          ``,
+          `Use aamp_send_result for normal conversation replies, including:`,
+          `  - greetings, acknowledgements, and small talk ("hi", "hello", "thanks", "got it")`,
+          `  - short follow-up questions that help narrow the user's intent`,
+          `  - direct answers, suggestions, or next-step guidance`,
+          ``,
+          `Use aamp_send_help ONLY when you are truly blocked and cannot produce a meaningful`,
+          `reply without waiting for specific missing information from the human.`,
+          `Do NOT use aamp_send_help just because the message is brief or casual.`,
+          ``,
+          `IMPORTANT: For conversational traffic, replying to "hi" with a natural greeting and an`,
+          `offer to help is CORRECT. Do not reject greetings as invalid tasks.`,
+          ``,
+          `### Sub-task dispatch rules:`,
+          `If you delegate work to another agent via aamp_dispatch_task, you MUST pass`,
+          `parentTaskId: "${task.taskId}" to establish the parent-child relationship.`,
+          `If you need to find a suitable agent first, call aamp_directory_search.`,
           ``,
           `Task ID:  ${task.taskId}`,
-          `Priority: ${task.priority}`,
           `From:     ${task.from}`,
           `Title:    ${task.title}`,
-          task.bodyText ? `
+          dispatchContextLines,
+          task.threadContextText ? `${task.threadContextText}` : "",
+          task.bodyText ? `Latest user message:
 ${task.bodyText}` : "",
-          actionRequiredSection,
+          task.contextLinks.length ? `Context Links:
+${task.contextLinks.map((l) => `  - ${l}`).join("\n")}` : "",
+          task.expiresAt ? `Expires: ${task.expiresAt}` : `Expires: none`,
+          `Received: ${task.receivedAt}`,
           otherActionableTasks.length > 0 ? `
 (+${otherActionableTasks.length} more tasks queued)` : ""
         ] : [
@@ -3649,6 +3764,7 @@ ${task.bodyText}` : "",
           `Task ID:  ${task.taskId}`,
           `From:     ${task.from}`,
           `Title:    ${task.title}`,
+          dispatchContextLines,
           task.threadContextText ? `${task.threadContextText}` : "",
           task.bodyText ? `Description:
 ${task.bodyText}` : "",
@@ -3658,7 +3774,23 @@ ${task.contextLinks.map((l) => `  - ${l}`).join("\n")}` : "",
           `Received: ${task.receivedAt}`,
           otherActionableTasks.length > 0 ? `
 (+${otherActionableTasks.length} more tasks queued)` : ""
-        ].filter(Boolean).join("\n");
+        ];
+        const lines = isNotification ? [
+          `## Sub-task Update`,
+          ``,
+          `A sub-task you dispatched has returned a result. Review the information below.`,
+          `If the sub-task included attachments, use aamp_download_attachment to fetch them.`,
+          ``,
+          `Task ID:  ${task.taskId}`,
+          `Priority: ${task.priority}`,
+          `From:     ${task.from}`,
+          `Title:    ${task.title}`,
+          task.bodyText ? `
+${task.bodyText}` : "",
+          actionRequiredSection,
+          otherActionableTasks.length > 0 ? `
+(+${otherActionableTasks.length} more tasks queued)` : ""
+        ] : taskPromptLines.filter(Boolean).join("\n");
         return { prependContext: lines };
       },
       { priority: 5 }