@automagik/omni 2.260911.1 → 2.260911.3

package/dist/server/index.js

@@ -38426,7 +38426,7 @@ var init_debounce = __esm(() => {
 class AutomationEngine {
   config;
   subscriptions = new Map;
-  instanceQueues = new Map;
+  queues = new Map;
   debounceManagers = new Map;
   automations = [];
   eventBus = null;
@@ -38751,29 +38751,38 @@ class AutomationEngine {
     const manager = new DebounceManager(automation.debounce, callback);
     this.debounceManagers.set(automation.id, manager);
   }
+  resolveQueueKey(automation, context2, instanceId) {
+    if (automation.maxConcurrency == null && automation.concurrencyKey == null)
+      return instanceId;
+    const base = `${instanceId}:${automation.id}`;
+    if (!automation.concurrencyKey)
+      return base;
+    const partition = substituteTemplate(automation.concurrencyKey, context2).trim();
+    return partition ? `${base}:${partition}` : base;
+  }
   async queueExecution(automation, event, context2, instanceId) {
-    let queue = this.instanceQueues.get(instanceId);
-    if (!queue) {
-      const maxConcurrency = this.config.instanceConcurrencyOverrides?.[instanceId] ?? this.config.defaultConcurrency;
-      queue = {
-        instanceId,
-        activeCount: 0,
-        maxConcurrency,
-        pending: []
-      };
-      this.instanceQueues.set(instanceId, queue);
-    }
+    const key = this.resolveQueueKey(automation, context2, instanceId);
+    const maxConcurrency = automation.maxConcurrency ?? this.config.instanceConcurrencyOverrides?.[instanceId] ?? this.config.defaultConcurrency;
+    const queue = this.queues.get(key) ?? {
+      key,
+      evictWhenIdle: key !== instanceId,
+      activeCount: 0,
+      maxConcurrency,
+      pending: []
+    };
+    queue.maxConcurrency = maxConcurrency;
+    this.queues.set(key, queue);
     if (queue.activeCount < queue.maxConcurrency) {
       return this.executeAutomation(automation, event, context2, queue);
     }
     const maxQueueDepth = this.config.maxQueueDepth ?? 100;
     if (queue.pending.length >= maxQueueDepth) {
       logger4.warn("Queue full, applying backpressure", {
-        instanceId,
+        queueKey: key,
         queueDepth: queue.pending.length,
         maxDepth: maxQueueDepth
       });
-      throw new QueueFullError(instanceId, queue.pending.length, maxQueueDepth);
+      throw new QueueFullError(key, queue.pending.length, maxQueueDepth);
     }
     const queueRef = queue;
     return new Promise((resolve) => {
@@ -38785,7 +38794,7 @@ class AutomationEngine {
       });
       logger4.debug("Execution queued", {
         automationId: automation.id,
-        instanceId,
+        queueKey: key,
         queueLength: queueRef.pending.length
       });
     });
@@ -38892,6 +38901,10 @@ class AutomationEngine {
           next.resolve(result);
         }
       }
+      if (queue.evictWhenIdle && queue.activeCount === 0 && queue.pending.length === 0) {
+        if (this.queues.get(queue.key) === queue)
+          this.queues.delete(queue.key);
+      }
     }
   }
   async logExecution(result, trustedTenantId = null) {
@@ -38917,8 +38930,8 @@ class AutomationEngine {
     });
   }
   getMetrics() {
-    const instanceQueues = Array.from(this.instanceQueues.values()).map((q) => ({
-      instanceId: q.instanceId,
+    const instanceQueues = Array.from(this.queues.values()).map((q) => ({
+      instanceId: q.key,
       activeCount: q.activeCount,
       pendingCount: q.pending.length
     }));
@@ -38961,12 +38974,12 @@ var init_engine = __esm(() => {
   init_templates();
   logger4 = createLogger("automations:engine");
   QueueFullError = class QueueFullError extends Error {
-    instanceId;
+    queueKey;
     queueDepth;
     maxDepth;
-    constructor(instanceId, queueDepth, maxDepth) {
-      super(`Queue full for instance ${instanceId}: ${queueDepth}/${maxDepth} pending`);
-      this.instanceId = instanceId;
+    constructor(queueKey, queueDepth, maxDepth) {
+      super(`Queue full for ${queueKey}: ${queueDepth}/${maxDepth} pending`);
+      this.queueKey = queueKey;
       this.queueDepth = queueDepth;
       this.maxDepth = maxDepth;
       this.name = "QueueFullError";
@@ -174411,6 +174424,8 @@ var init_schema2 = __esm(() => {
     enabled: boolean2("enabled").notNull().default(true),
     priority: integer("priority").notNull().default(0),
     transactionalEmissions: boolean2("transactional_emissions").notNull().default(false),
+    maxConcurrency: integer("max_concurrency"),
+    concurrencyKey: text("concurrency_key"),
     managedByAgentId: uuid("managed_by_agent_id").references(() => agents.id, { onDelete: "cascade" }),
     createdAt: timestamp("created_at", { withTimezone: true }).notNull().defaultNow(),
     updatedAt: timestamp("updated_at", { withTimezone: true }).notNull().defaultNow(),
@@ -251657,7 +251672,7 @@ var init_sentry_scrub = __esm(() => {
 var require_package7 = __commonJS((exports, module) => {
   module.exports = {
     name: "@omni/api",
-    version: "2.260911.1",
+    version: "2.260911.3",
     private: true,
     type: "module",
     exports: {
@@ -379382,6 +379397,12 @@ var init_automations3 = __esm(() => {
     transactionalEmissions: exports_external.boolean().openapi({
       description: "Transactional publication (G5, #988): buffer the run's emit_event publishes and flush them in order " + "only when every action succeeded; a failed run publishes zero"
     }),
+    maxConcurrency: exports_external.number().int().nullable().openapi({
+      description: "Per-automation concurrency limit (#1108). null = queued per instance with the engine default; " + "set = a queue private to this automation, 1 being strict single-flight"
+    }),
+    concurrencyKey: exports_external.string().nullable().openapi({
+      description: "Template over the event payload partitioning this automation\u2019s queue (#1108), e.g. " + '"{{payload.from.id}}" to serialize per chat; null = one queue for the whole automation'
+    }),
     managedByAgentId: exports_external.string().uuid().nullable().openapi({
       description: "Set when this automation was compiled from an agent event manifest (RFC #925 G4b, #986); " + "null = hand-made. Managed automations reject manual mutation \u2014 edit the owning agent\u2019s manifest instead"
     }),
@@ -379400,6 +379421,12 @@ var init_automations3 = __esm(() => {
     priority: exports_external.number().int().default(0).openapi({ description: "Priority" }),
     transactionalEmissions: exports_external.boolean().default(false).openapi({
       description: "Transactional publication (G5, #988): buffer the run's emit_event publishes and flush them in order " + "only when every action succeeded; a failed run publishes zero. Default false = immediate publishing"
+    }),
+    maxConcurrency: exports_external.number().int().min(1).nullable().optional().openapi({
+      description: "Per-automation concurrency limit (#1108). Omit/null = today\u2019s per-instance queueing with the engine " + "default; 1 = strict single-flight, which is what a read-before-write action needs"
+    }),
+    concurrencyKey: exports_external.string().min(1).nullable().optional().openapi({
+      description: "Template over the event payload partitioning this automation\u2019s queue (#1108), e.g. " + '"{{payload.from.id}}" to serialize per chat; omit/null = one queue for the whole automation'
     })
   });
   AutomationLogSchema = exports_external.object({
@@ -379445,7 +379472,9 @@ var init_automations3 = __esm(() => {
   AutomationMetricsSchema = exports_external.object({
     running: exports_external.boolean().openapi({ description: "Engine running" }),
     instanceQueues: exports_external.array(exports_external.object({
-      instanceId: exports_external.string().openapi({ description: "Instance ID" }),
+      instanceId: exports_external.string().openapi({
+        description: 'Queue key: the instance ID, or "<instanceId>:<automationId>[:<partition>]" for an automation ' + "running on its own queue (#1108)"
+      }),
       activeCount: exports_external.number().int().openapi({ description: "Active jobs" }),
       pendingCount: exports_external.number().int().openapi({ description: "Pending jobs" })
     })).optional().openapi({ description: "Instance queue stats" }),
@@ -385197,7 +385226,9 @@ var init_automations4 = __esm(() => {
     debounce: debounceSchema.optional().describe("Message debounce configuration"),
     enabled: exports_external.boolean().default(true).describe("Whether automation is enabled"),
     priority: exports_external.number().int().default(0).describe("Priority (higher runs first)"),
-    transactionalEmissions: exports_external.boolean().default(false).describe("Transactional publication (G5, #988): buffer the run's emit_event publishes and flush them in order " + "only when every action succeeded; a failed run publishes zero. Default false = immediate publishing")
+    transactionalEmissions: exports_external.boolean().default(false).describe("Transactional publication (G5, #988): buffer the run's emit_event publishes and flush them in order " + "only when every action succeeded; a failed run publishes zero. Default false = immediate publishing"),
+    maxConcurrency: exports_external.number().int().min(1).nullable().optional().describe("Per-automation concurrency limit (#1108). Omit/null = today\u2019s per-instance queueing with the engine " + "default; 1 = strict single-flight, which is what a read-before-write action needs"),
+    concurrencyKey: exports_external.string().min(1).nullable().optional().describe("Template over the event payload partitioning this automation\u2019s queue (#1108), e.g. " + '"{{payload.from.id}}" to serialize per chat; omit/null = one queue for the whole automation')
   });
   updateAutomationSchema = createAutomationSchema.partial();
   listQuerySchema5 = exports_external.object({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@automagik/omni",
-  "version": "2.260911.1",
+  "version": "2.260911.3",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/automagik-dev/omni.git",