@action-llama/action-llama 0.28.0 → 0.29.0

package/dist/execution/waiting-registry.js

@@ -0,0 +1,185 @@
+/**
+ * WaitingRegistry — tracks agent instances that are suspended mid-conversation,
+ * waiting for a specific trigger (webhook or agent-trigger) to resume.
+ *
+ * Instances are matched FIFO within each agent. The registry is the single source
+ * of truth for all waiting instances and handles matching, timeout cleanup, and
+ * persistence to the KV store for restart recovery.
+ */
+// ── Dot-path matching helper ─────────────────────────────────
+/**
+ * Check if a dot-path key matches a value in an object.
+ * e.g. matchDotPath({ a: { b: "c" } }, "a.b", "c") → true
+ */
+export function matchDotPath(obj, key, value) {
+    const parts = key.split(".");
+    let current = obj;
+    for (const part of parts) {
+        if (current == null || typeof current !== "object")
+            return false;
+        current = current[part];
+    }
+    return String(current) === value;
+}
+// ── Registry ─────────────────────────────────────────────────
+const KV_NAMESPACE = "waiting-instances";
+export class WaitingRegistry {
+    entries = new Map();
+    /**
+     * Register a waiting instance. Starts its timeout timer.
+     */
+    register(entry) {
+        // Start timeout timer
+        const remaining = entry.deadline - Date.now();
+        if (remaining <= 0) {
+            // Already expired — reject immediately
+            entry.reject?.(new Error("Wait timeout expired before registration"));
+            return;
+        }
+        entry.timeoutTimer = setTimeout(() => {
+            const removed = this.remove(entry.instanceId);
+            if (removed) {
+                removed.reject?.(new Error("Wait timeout expired"));
+            }
+        }, remaining);
+        entry.timeoutTimer.unref?.();
+        this.entries.set(entry.instanceId, entry);
+    }
+    /**
+     * Find and remove a matching waiting instance for a webhook trigger (FIFO).
+     * Returns null if no match found.
+     */
+    matchWebhook(agentName, context) {
+        // Find all entries for this agent, sorted by registeredAt (FIFO)
+        const candidates = this.getByAgent(agentName)
+            .filter((e) => e.filter.type === "webhook")
+            .sort((a, b) => a.registeredAt - b.registeredAt);
+        for (const entry of candidates) {
+            const filter = entry.filter;
+            // Check source filter
+            if (filter.source && filter.source !== context.source)
+                continue;
+            // Check event filter
+            if (filter.event && filter.event !== context.event)
+                continue;
+            // Check dot-path match predicates on the context
+            if (filter.match) {
+                let allMatch = true;
+                for (const [key, value] of Object.entries(filter.match)) {
+                    if (!matchDotPath(context, key, value)) {
+                        allMatch = false;
+                        break;
+                    }
+                }
+                if (!allMatch)
+                    continue;
+            }
+            // Match found — remove from registry and clear timeout
+            return this._removeAndCleanup(entry.instanceId);
+        }
+        return null;
+    }
+    /**
+     * Find and remove a matching waiting instance for an agent trigger (FIFO).
+     * Returns null if no match found.
+     */
+    matchAgentTrigger(agentName, sourceAgent) {
+        const candidates = this.getByAgent(agentName)
+            .filter((e) => e.filter.type === "agent-trigger")
+            .sort((a, b) => a.registeredAt - b.registeredAt);
+        for (const entry of candidates) {
+            const filter = entry.filter;
+            // If sourceAgent is specified, it must match
+            if (filter.sourceAgent && filter.sourceAgent !== sourceAgent)
+                continue;
+            return this._removeAndCleanup(entry.instanceId);
+        }
+        return null;
+    }
+    /**
+     * Remove a waiting instance by ID (for timeout/kill).
+     */
+    remove(instanceId) {
+        return this._removeAndCleanup(instanceId);
+    }
+    /**
+     * Get all waiting instances for a specific agent.
+     */
+    getByAgent(agentName) {
+        return Array.from(this.entries.values()).filter((e) => e.agentName === agentName);
+    }
+    /**
+     * Get all waiting instances.
+     */
+    getAll() {
+        return Array.from(this.entries.values());
+    }
+    /**
+     * Get the total number of waiting instances.
+     */
+    count() {
+        return this.entries.size;
+    }
+    /**
+     * Persist all waiting instances to the KV store.
+     */
+    async persist(store) {
+        // Clear existing entries
+        await store.kv.deleteAll(KV_NAMESPACE);
+        for (const entry of this.entries.values()) {
+            const persisted = {
+                instanceId: entry.instanceId,
+                agentName: entry.agentName,
+                filter: entry.filter,
+                deadline: entry.deadline,
+                registeredAt: entry.registeredAt,
+                runId: entry.runId,
+                runtimeType: entry.runtimeType,
+                cwd: entry.cwd,
+            };
+            await store.kv.set(KV_NAMESPACE, entry.instanceId, persisted);
+        }
+    }
+    /**
+     * Rehydrate waiting instances from the KV store.
+     * Returns a new registry with loaded entries (no in-memory resolve/reject callbacks).
+     */
+    static async rehydrate(store) {
+        const registry = new WaitingRegistry();
+        const entries = await store.kv.list(KV_NAMESPACE);
+        const now = Date.now();
+        for (const { value } of entries) {
+            // Skip expired entries
+            if (value.deadline <= now) {
+                await store.kv.delete(KV_NAMESPACE, value.instanceId);
+                continue;
+            }
+            const entry = {
+                ...value,
+                // No resolve/reject — cold restart instances get new runners on match
+            };
+            // Start timeout timer for remaining time
+            entry.timeoutTimer = setTimeout(() => {
+                registry.remove(entry.instanceId);
+            }, entry.deadline - now);
+            entry.timeoutTimer.unref?.();
+            registry.entries.set(entry.instanceId, entry);
+        }
+        return registry;
+    }
+    /**
+     * Remove an entry and clean up its timeout timer.
+     */
+    _removeAndCleanup(instanceId) {
+        const entry = this.entries.get(instanceId);
+        if (!entry)
+            return null;
+        this.entries.delete(instanceId);
+        if (entry.timeoutTimer) {
+            clearTimeout(entry.timeoutTimer);
+            entry.timeoutTimer = undefined;
+        }
+        return entry;
+    }
+}
+//# sourceMappingURL=waiting-registry.js.map

package/dist/execution/waiting-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"waiting-registry.js","sourceRoot":"","sources":["../../src/execution/waiting-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAwCH,gEAAgE;AAEhE;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,GAAQ,EAAE,GAAW,EAAE,KAAa;IAC/D,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC7B,IAAI,OAAO,GAAG,GAAG,CAAC;IAClB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO,KAAK,CAAC;QACjE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IACD,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,KAAK,CAAC;AACnC,CAAC;AAED,gEAAgE;AAEhE,MAAM,YAAY,GAAG,mBAAmB,CAAC;AAEzC,MAAM,OAAO,eAAe;IAClB,OAAO,GAAG,IAAI,GAAG,EAA2B,CAAC;IAErD;;OAEG;IACH,QAAQ,CAAC,KAAsB;QAC7B,sBAAsB;QACtB,MAAM,SAAS,GAAG,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9C,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;YACnB,uCAAuC;YACvC,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC,CAAC;YACtE,OAAO;QACT,CAAC;QAED,KAAK,CAAC,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;YACnC,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC9C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAC;YACtD,CAAC;QACH,CAAC,EAAE,SAAS,CAAC,CAAC;QACd,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,EAAE,CAAC;QAE7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IAED;;;OAGG;IACH,YAAY,CAAC,SAAiB,EAAE,OAAuB;QACrD,iEAAiE;QACjE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;aAC1C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC;aAC1C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;QAEnD,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAkD,CAAC;YAExE,sBAAsB;YACtB,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM;gBAAE,SAAS;YAEhE,qBAAqB;YACrB,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK;gBAAE,SAAS;YAE7D,iDAAiD;YACjD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,IAAI,QAAQ,GAAG,IAAI,CAAC;gBACpB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;oBACxD,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC;wBACvC,QAAQ,GAAG,KAAK,CAAC;wBACjB,MAAM;oBACR,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,QAAQ;oBAAE,SAAS;YAC1B,CAAC;YAED,uDAAuD;YACvD,OAAO,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;OAGG;IACH,iBAAiB,CAAC,SAAiB,EAAE,WAAmB;QACtD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;aAC1C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,eAAe,CAAC;aAChD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC;QAEnD,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAwD,CAAC;YAE9E,6CAA6C;YAC7C,IAAI,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,KAAK,WAAW;gBAAE,SAAS;YAEvE,OAAO,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,UAAkB;QACvB,OAAO,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,SAAiB;QAC1B,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;IACpF,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK;QACH,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,KAAuB;QACnC,yBAAyB;QACzB,MAAM,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAEvC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,MAAM,SAAS,GAA6B;gBAC1C,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,GAAG,EAAE,KAAK,CAAC,GAAG;aACf,CAAC;YACF,MAAM,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,YAAY,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,KAAuB;QAC5C,MAAM,QAAQ,GAAG,IAAI,eAAe,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,EAAE,CAAC,IAAI,CAA2B,YAAY,CAAC,CAAC;QAC5E,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,MAAM,EAAE,KAAK,EAAE,IAAI,OAAO,EAAE,CAAC;YAChC,uBAAuB;YACvB,IAAI,KAAK,CAAC,QAAQ,IAAI,GAAG,EAAE,CAAC;gBAC1B,MAAM,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;gBACtD,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAAoB;gBAC7B,GAAG,KAAK;gBACR,sEAAsE;aACvE,CAAC;YAEF,yCAAyC;YACzC,KAAK,CAAC,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;gBACnC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACpC,CAAC,EAAE,KAAK,CAAC,QAAQ,GAAG,GAAG,CAAC,CAAC;YACzB,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,EAAE,CAAC;YAE7B,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,UAAkB;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAChC,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACjC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC;QACjC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}