@n8n/instance-ai 1.0.0 → 1.2.0

package/dist/planned-tasks/planned-task-permissions.js

@@ -0,0 +1,27 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.applyPlannedTaskPermissions = applyPlannedTaskPermissions;
+const PLANNED_TASK_PERMISSION_OVERRIDES = {
+    'manage-data-tables': {
+        createDataTable: 'always_allow',
+        mutateDataTableSchema: 'always_allow',
+        mutateDataTableRows: 'always_allow',
+    },
+    'build-workflow': {
+        runWorkflow: 'always_allow',
+        publishWorkflow: 'always_allow',
+    },
+};
+function applyPlannedTaskPermissions(context, taskKind) {
+    const overrides = PLANNED_TASK_PERMISSION_OVERRIDES[taskKind];
+    if (!overrides)
+        return context;
+    return {
+        ...context,
+        permissions: {
+            ...context.permissions,
+            ...overrides,
+        },
+    };
+}
+//# sourceMappingURL=planned-task-permissions.js.map

package/dist/planned-tasks/planned-task-permissions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"planned-task-permissions.js","sourceRoot":"","sources":["../../src/planned-tasks/planned-task-permissions.ts"],"names":[],"mappings":";;AAgCA,kEAcC;AAjCD,MAAM,iCAAiC,GAEnC;IACH,oBAAoB,EAAE;QACrB,eAAe,EAAE,cAAc;QAC/B,qBAAqB,EAAE,cAAc;QACrC,mBAAmB,EAAE,cAAc;KACnC;IACD,gBAAgB,EAAE;QACjB,WAAW,EAAE,cAAc;QAC3B,eAAe,EAAE,cAAc;KAC/B;CACD,CAAC;AAOF,SAAgB,2BAA2B,CAC1C,OAA0B,EAC1B,QAAyB;IAEzB,MAAM,SAAS,GAAG,iCAAiC,CAAC,QAAQ,CAAC,CAAC;IAC9D,IAAI,CAAC,SAAS;QAAE,OAAO,OAAO,CAAC;IAE/B,OAAO;QACN,GAAG,OAAO;QACV,WAAW,EAAE;YACZ,GAAG,OAAO,CAAC,WAAW;YACtB,GAAG,SAAS;SACa;KAC1B,CAAC;AACH,CAAC"}

package/dist/planned-tasks/planned-task-service.js

@@ -127,6 +127,12 @@ class PlannedTaskCoordinator {
                 action = { type: 'synthesize', graph: nextGraph };
                 return nextGraph;
             }
+            const hasWorkLeft = graph.tasks.some((t) => t.status === 'planned' || t.status === 'running');
+            if (graph.tasks.length > 0 && !hasWorkLeft) {
+                const nextGraph = { ...graph, status: 'completed' };
+                action = { type: 'synthesize', graph: nextGraph };
+                return nextGraph;
+            }
             const availableSlots = options.availableSlots ?? graph.tasks.length;
             if (availableSlots <= 0) {
                 action = { type: 'none', graph };

package/dist/planned-tasks/planned-task-service.js.map

@@ -1 +1 @@
-{"version":3,"file":"planned-task-service.js","sourceRoot":"","sources":["../../src/planned-tasks/planned-task-service.ts"],"names":[],"mappings":";;;AASA,SAAS,eAAe,CAAC,KAAoB;IAC5C,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,MAAM,CAAC;AACpE,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAoB;IACjD,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,EAAE,8BAA8B,KAAK,GAAG,CAAC,CAAC;YACzE,CAAC;QACF,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;YAC1E,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,EAAE,kCAAkC,CAAC,CAAC;QAC9E,CAAC;IACF,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IAE3D,MAAM,KAAK,GAAG,CAAC,MAAc,EAAE,EAAE;QAChC,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO;QAChC,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,+CAA+C,MAAM,GAAG,CAAC,CAAC;QAC3E,CAAC;QAED,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9B,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;YACtC,KAAK,CAAC,KAAK,CAAC,CAAC;QACd,CAAC;QACD,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACrB,CAAC,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;AACF,CAAC;AAED,SAAS,SAAS,CAAC,IAAuB;IACzC,OAAO,IAAI,CAAC,MAAM,KAAK,WAAW,CAAC;AACpC,CAAC;AAED,SAAS,gBAAgB,CACxB,KAAuB,EACvB,MAAc,EACd,OAAuD;IAEvD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;IAClE,IAAI,KAAK,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAE3B,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;IAC/B,KAAK,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACrC,OAAO,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,CAAC;AAC5B,CAAC;AAED,MAAa,sBAAsB;IACL;IAA7B,YAA6B,OAA2B;QAA3B,YAAO,GAAP,OAAO,CAAoB;IAAG,CAAC;IAE5D,KAAK,CAAC,UAAU,CACf,QAAgB,EAChB,KAAoB,EACpB,QAAwD;QAExD,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE5B,MAAM,KAAK,GAAqB;YAC/B,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,cAAc,EAAE,QAAQ,CAAC,cAAc;YACvC,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAoB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC9C,GAAG,IAAI;gBACP,MAAM,EAAE,SAAS;aACjB,CAAC,CAAC;SACH,CAAC;QAEF,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACzC,OAAO,KAAK,CAAC;IACd,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB;QAC9B,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,WAAW,CAChB,QAAgB,EAChB,MAAc,EACd,MAA2E;QAE3E,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;YACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB;YAClE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;YAC3D,KAAK,EAAE,SAAS;SAChB,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAClB,QAAgB,EAChB,MAAc,EACd,MAAmF;QAEnF,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM;YACpC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;YACvC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE;YAC3C,KAAK,EAAE,SAAS;SAChB,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CACf,QAAgB,EAChB,MAAc,EACd,MAA+C;QAE/C,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,eAAe;YACpD,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE;SAC3C,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAClB,QAAgB,EAChB,MAAc,EACd,MAAgD;QAEhD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,WAAW;YACnB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,IAAI,CAAC,KAAK;YAClC,UAAU,EAAE,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE;SAC5C,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CACT,QAAgB,EAChB,UAAuC,EAAE;QAIzC,IAAI,MAAM,GAA+B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAEvE,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;YAC7C,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBACjC,OAAO,KAAK,CAAC;YACd,CAAC;YAED,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC;YACxE,IAAI,UAAU,EAAE,CAAC;gBAChB,MAAM,SAAS,GAAqB,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;gBAC5E,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;gBAC1D,OAAO,SAAS,CAAC;YAClB,CAAC;YAED,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5D,MAAM,SAAS,GAAqB,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;gBACtE,MAAM,GAAG,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;gBAClD,OAAO,SAAS,CAAC;YAClB,CAAC;YAED,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;YACpE,IAAI,cAAc,IAAI,CAAC,EAAE,CAAC;gBACzB,MAAM,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBACjC,OAAO,KAAK,CAAC;YACd,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,GAAG,CAC5B,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAChF,CAAC;YACF,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CACpC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAC3F,CAAC;YAEF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBACjC,OAAO,KAAK,CAAC;YACd,CAAC;YAED,MAAM,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,EAAE,CAAC;YACjF,OAAO,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QAGH,OAAO,MAAM,CAAC;IACf,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAgB;QAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;CACD;AApJD,wDAoJC"}
+{"version":3,"file":"planned-task-service.js","sourceRoot":"","sources":["../../src/planned-tasks/planned-task-service.ts"],"names":[],"mappings":";;;AASA,SAAS,eAAe,CAAC,KAAoB;IAC5C,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,MAAM,CAAC;AACpE,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAoB;IACjD,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,EAAE,8BAA8B,KAAK,GAAG,CAAC,CAAC;YACzE,CAAC;QACF,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;YAC1E,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,EAAE,kCAAkC,CAAC,CAAC;QAC9E,CAAC;IACF,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IAE3D,MAAM,KAAK,GAAG,CAAC,MAAc,EAAE,EAAE;QAChC,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO;QAChC,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,+CAA+C,MAAM,GAAG,CAAC,CAAC;QAC3E,CAAC;QAED,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9B,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;YACtC,KAAK,CAAC,KAAK,CAAC,CAAC;QACd,CAAC;QACD,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACrB,CAAC,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC;AACF,CAAC;AAED,SAAS,SAAS,CAAC,IAAuB;IACzC,OAAO,IAAI,CAAC,MAAM,KAAK,WAAW,CAAC;AACpC,CAAC;AAED,SAAS,gBAAgB,CACxB,KAAuB,EACvB,MAAc,EACd,OAAuD;IAEvD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;IAClE,IAAI,KAAK,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAE3B,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;IAC/B,KAAK,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACrC,OAAO,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,CAAC;AAC5B,CAAC;AAED,MAAa,sBAAsB;IACL;IAA7B,YAA6B,OAA2B;QAA3B,YAAO,GAAP,OAAO,CAAoB;IAAG,CAAC;IAE5D,KAAK,CAAC,UAAU,CACf,QAAgB,EAChB,KAAoB,EACpB,QAAwD;QAExD,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE5B,MAAM,KAAK,GAAqB;YAC/B,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,cAAc,EAAE,QAAQ,CAAC,cAAc;YACvC,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAoB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC9C,GAAG,IAAI;gBACP,MAAM,EAAE,SAAS;aACjB,CAAC,CAAC;SACH,CAAC;QAEF,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACzC,OAAO,KAAK,CAAC;IACd,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB;QAC9B,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,WAAW,CAChB,QAAgB,EAChB,MAAc,EACd,MAA2E;QAE3E,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;YACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB;YAClE,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;YAC3D,KAAK,EAAE,SAAS;SAChB,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAClB,QAAgB,EAChB,MAAc,EACd,MAAmF;QAEnF,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM;YACpC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;YACvC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE;YAC3C,KAAK,EAAE,SAAS;SAChB,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CACf,QAAgB,EAChB,MAAc,EACd,MAA+C;QAE/C,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,eAAe;YACpD,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE;SAC3C,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAClB,QAAgB,EAChB,MAAc,EACd,MAAgD;QAEhD,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CACpD,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,IAAI;YACP,MAAM,EAAE,WAAW;YACnB,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,IAAI,CAAC,KAAK;YAClC,UAAU,EAAE,MAAM,EAAE,UAAU,IAAI,IAAI,CAAC,GAAG,EAAE;SAC5C,CAAC,CAAC,CACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CACT,QAAgB,EAChB,UAAuC,EAAE;QAIzC,IAAI,MAAM,GAA+B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAEvE,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;YAC7C,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,MAAM,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBACjC,OAAO,KAAK,CAAC;YACd,CAAC;YAED,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC;YACxE,IAAI,UAAU,EAAE,CAAC;gBAChB,MAAM,SAAS,GAAqB,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;gBAC5E,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;gBAC1D,OAAO,SAAS,CAAC;YAClB,CAAC;YAED,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5D,MAAM,SAAS,GAAqB,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;gBACtE,MAAM,GAAG,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;gBAClD,OAAO,SAAS,CAAC;YAClB,CAAC;YAID,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;YAC9F,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBAC5C,MAAM,SAAS,GAAqB,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;gBACtE,MAAM,GAAG,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;gBAClD,OAAO,SAAS,CAAC;YAClB,CAAC;YAED,MAAM,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;YACpE,IAAI,cAAc,IAAI,CAAC,EAAE,CAAC;gBACzB,MAAM,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBACjC,OAAO,KAAK,CAAC;YACd,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,GAAG,CAC5B,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAChF,CAAC;YACF,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CACpC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAC3F,CAAC;YAEF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;gBACjC,OAAO,KAAK,CAAC;YACd,CAAC;YAED,MAAM,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,EAAE,CAAC;YACjF,OAAO,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QAGH,OAAO,MAAM,CAAC;IACf,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAgB;QAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;CACD;AA7JD,wDA6JC"}

package/dist/runtime/background-task-manager.d.ts

@@ -12,6 +12,7 @@ export interface ManagedBackgroundTask {
     startedAt: number;
     abortController: AbortController;
     corrections: string[];
+    onCorrectionQueued?: () => void;
     messageGroupId?: string;
     outcome?: Record<string, unknown>;
     plannedTaskId?: string;
@@ -28,7 +29,7 @@ export interface SpawnManagedBackgroundTaskOptions {
     plannedTaskId?: string;
     workItemId?: string;
     traceContext?: InstanceAiTraceContext;
-    run: (signal: AbortSignal, drainCorrections: () => string[]) => Promise<string | BackgroundTaskResult>;
+    run: (signal: AbortSignal, drainCorrections: () => string[], waitForCorrection: () => Promise<void>) => Promise<string | BackgroundTaskResult>;
     onLimitReached?: (errorMessage: string) => void;
     onCompleted?: (task: ManagedBackgroundTask) => void | Promise<void>;
     onFailed?: (task: ManagedBackgroundTask) => void | Promise<void>;

package/dist/runtime/background-task-manager.js

@@ -21,6 +21,11 @@ class BackgroundTaskManager {
         if (task.status !== 'running')
             return 'task-completed';
         task.corrections.push(correction);
+        if (task.onCorrectionQueued) {
+            const notify = task.onCorrectionQueued;
+            task.onCorrectionQueued = undefined;
+            notify();
+        }
         return 'queued';
     }
     cancelTask(threadId, taskId) {
@@ -84,8 +89,15 @@ class BackgroundTaskManager {
             task.corrections.length = 0;
             return pending;
         };
+        const waitForCorrection = async () => await new Promise((resolve) => {
+            if (task.corrections.length > 0) {
+                resolve();
+                return;
+            }
+            task.onCorrectionQueued = resolve;
+        });
         try {
-            const raw = await options.run(task.abortController.signal, drainCorrections);
+            const raw = await options.run(task.abortController.signal, drainCorrections, waitForCorrection);
             task.status = 'completed';
             task.result = typeof raw === 'string' ? raw : raw.text;
             task.outcome = typeof raw === 'string' ? undefined : raw.outcome;

package/dist/runtime/background-task-manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"background-task-manager.js","sourceRoot":"","sources":["../../src/runtime/background-task-manager.ts"],"names":[],"mappings":";;;AA2KA,sEAqBC;AA/ID,MAAa,qBAAqB;IAGJ;IAFZ,KAAK,GAAG,IAAI,GAAG,EAAiC,CAAC;IAElE,YAA6B,yBAAyB,CAAC;QAA1B,2BAAsB,GAAtB,sBAAsB,CAAI;IAAG,CAAC;IAE3D,gBAAgB,CAAC,QAAgB;QAChC,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;IAC9E,CAAC;IAED,eAAe,CAAC,QAAgB;QAC/B,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACrC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,CACjE,CAAC;IACH,CAAC;IAED,eAAe,CACd,QAAgB,EAChB,MAAc,EACd,UAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ;YAAE,OAAO,gBAAgB,CAAC;QACjE,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,gBAAgB,CAAC;QACvD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,MAAc;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAEvF,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,YAAY,CAAC,QAAgB;QAC5B,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACzC,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;gBAAE,SAAS;YACtE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,SAAS;QACR,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACzC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC7B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,OAA0C;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QACnE,IAAI,YAAY,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjD,OAAO,CAAC,cAAc,EAAE,CACvB,0CAA0C,IAAI,CAAC,sBAAsB,iEAAiE,CACtI,CAAC;YACF,OAAO,KAAK,CAAC;QACd,CAAC;QAED,MAAM,IAAI,GAA0B;YACnC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,eAAe,EAAE,IAAI,eAAe,EAAE;YACtC,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,YAAY,EAAE,OAAO,CAAC,YAAY;SAClC,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACrC,KAAK,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACb,CAAC;IAEO,KAAK,CAAC,WAAW,CACxB,IAA2B,EAC3B,OAA0C;QAE1C,MAAM,gBAAgB,GAAG,GAAa,EAAE;YACvC,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;YACtC,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC;YAC5B,OAAO,OAAO,CAAC;QAChB,CAAC,CAAC;QAEF,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;YAC7E,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;YACvD,IAAI,CAAC,OAAO,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;YACjE,MAAM,OAAO,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO;YAChD,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;YACvB,IAAI,CAAC,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACpE,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC;gBACJ,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAC1C,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC;gBACjC,CAAC;YACF,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChC,CAAC;QACF,CAAC;IACF,CAAC;CACD;AAxHD,sDAwHC;AAEM,KAAK,UAAU,6BAA6B,CAGlD,OAAe,EACf,YAAqB,EACrB,UAA+C,EAAE;IAEjD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAE9C,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QACjC,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3C,SAAS;QACV,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,mBAAmB,IAAI,CAAC,IAAI,WAAW,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,oBAAoB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,yBAAyB,OAAO,EAAE,CAAC;AAC/E,CAAC"}
+{"version":3,"file":"background-task-manager.js","sourceRoot":"","sources":["../../src/runtime/background-task-manager.ts"],"names":[],"mappings":";;;AAgMA,sEAqBC;AAjKD,MAAa,qBAAqB;IAGJ;IAFZ,KAAK,GAAG,IAAI,GAAG,EAAiC,CAAC;IAElE,YAA6B,yBAAyB,CAAC;QAA1B,2BAAsB,GAAtB,sBAAsB,CAAI;IAAG,CAAC;IAE3D,gBAAgB,CAAC,QAAgB;QAChC,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;IAC9E,CAAC;IAED,eAAe,CAAC,QAAgB;QAC/B,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACrC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,CACjE,CAAC;IACH,CAAC;IAED,eAAe,CACd,QAAgB,EAChB,MAAc,EACd,UAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ;YAAE,OAAO,gBAAgB,CAAC;QACjE,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,gBAAgB,CAAC;QACvD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC;YACvC,IAAI,CAAC,kBAAkB,GAAG,SAAS,CAAC;YACpC,MAAM,EAAE,CAAC;QACV,CAAC;QACD,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,MAAc;QAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAEvF,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,YAAY,CAAC,QAAgB;QAC5B,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACzC,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;gBAAE,SAAS;YACtE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,SAAS;QACR,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACzC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;YAC7B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,OAA0C;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QACnE,IAAI,YAAY,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjD,OAAO,CAAC,cAAc,EAAE,CACvB,0CAA0C,IAAI,CAAC,sBAAsB,iEAAiE,CACtI,CAAC;YACF,OAAO,KAAK,CAAC;QACd,CAAC;QAED,MAAM,IAAI,GAA0B;YACnC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,eAAe,EAAE,IAAI,eAAe,EAAE;YACtC,WAAW,EAAE,EAAE;YACf,cAAc,EAAE,OAAO,CAAC,cAAc;YACtC,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,YAAY,EAAE,OAAO,CAAC,YAAY;SAClC,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACrC,KAAK,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACb,CAAC;IAEO,KAAK,CAAC,WAAW,CACxB,IAA2B,EAC3B,OAA0C;QAE1C,MAAM,gBAAgB,GAAG,GAAa,EAAE;YACvC,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;YACtC,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC;YAC5B,OAAO,OAAO,CAAC;QAChB,CAAC,CAAC;QAEF,MAAM,iBAAiB,GAAG,KAAK,IAAmB,EAAE,CACnD,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACnC,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjC,OAAO,EAAE,CAAC;gBACV,OAAO;YACR,CAAC;YACD,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC;QACnC,CAAC,CAAC,CAAC;QAEJ,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAC5B,IAAI,CAAC,eAAe,CAAC,MAAM,EAC3B,gBAAgB,EAChB,iBAAiB,CACjB,CAAC;YACF,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;YACvD,IAAI,CAAC,OAAO,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;YACjE,MAAM,OAAO,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO;YAChD,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;YACvB,IAAI,CAAC,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACpE,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC;gBACJ,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAC1C,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC;gBACjC,CAAC;YACF,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChC,CAAC;QACF,CAAC;IACF,CAAC;CACD;AA1ID,sDA0IC;AAEM,KAAK,UAAU,6BAA6B,CAGlD,OAAe,EACf,YAAqB,EACrB,UAA+C,EAAE;IAEjD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAE9C,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QACjC,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3C,SAAS;QACV,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,mBAAmB,IAAI,CAAC,IAAI,WAAW,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,oBAAoB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,yBAAyB,OAAO,EAAE,CAAC;AAC/E,CAAC"}

package/dist/runtime/resumable-stream-executor.d.ts

@@ -1,6 +1,8 @@
 import type { InstanceAiEvent } from '@n8n/api-types';
 import type { RunTree } from 'langsmith';
 import type { InstanceAiEventBus } from '../event-bus';
+import type { Logger } from '../logger';
+import { type WorkSummary } from '../stream/work-summary-accumulator';
 import type { SuspensionInfo } from '../utils/stream-helpers';
 type ConfirmationRequestEvent = Extract<InstanceAiEvent, {
     type: 'confirmation-request';
@@ -22,6 +24,7 @@ export interface ResumableStreamContext {
     agentId: string;
     eventBus: InstanceAiEventBus;
     signal: AbortSignal;
+    logger: Logger;
 }
 export interface ManualSuspensionControl {
     mode: 'manual';
@@ -30,6 +33,7 @@ export interface AutoResumeControl {
     mode: 'auto';
     waitForConfirmation: (requestId: string) => Promise<Record<string, unknown>>;
     drainCorrections?: () => string[];
+    waitForCorrection?: () => Promise<void>;
     onSuspension?: (suspension: SuspensionInfo) => void;
     buildResumeOptions?: (input: {
         mastraRunId: string;
@@ -44,14 +48,15 @@ export interface ExecuteResumableStreamOptions {
     control: ResumableStreamControl;
     initialMastraRunId?: string;
     llmStepTraceHooks?: LlmStepTraceHooks;
-    workingMemoryEnabled?: boolean;
 }
+export type TraceStatus = 'completed' | 'cancelled' | 'suspended' | 'errored';
 export interface ExecuteResumableStreamResult {
-    status: 'completed' | 'cancelled' | 'suspended' | 'errored';
+    status: TraceStatus;
     mastraRunId: string;
     text?: Promise<string>;
     suspension?: SuspensionInfo;
     confirmationEvent?: ConfirmationRequestEvent;
+    workSummary: WorkSummary;
 }
 export interface LlmStepTraceHooks {
     executionOptions: {
@@ -66,7 +71,7 @@ export interface LlmStepTraceHooks {
     onStreamChunk: (chunk: unknown) => void;
     startSegment: () => void;
     finalize: (source: ResumableStreamSource, options?: {
-        status?: 'completed' | 'cancelled' | 'suspended';
+        status?: TraceStatus;
         error?: string;
     }) => Promise<void>;
 }

package/dist/runtime/resumable-stream-executor.js

@@ -2,14 +2,14 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createLlmStepTraceHooks = createLlmStepTraceHooks;
 exports.executeResumableStream = executeResumableStream;
-const working_memory_tracing_1 = require("./working-memory-tracing");
 const map_chunk_1 = require("../stream/map-chunk");
+const work_summary_accumulator_1 = require("../stream/work-summary-accumulator");
 const langsmith_tracing_1 = require("../tracing/langsmith-tracing");
 const stream_helpers_1 = require("../utils/stream-helpers");
 const MAX_TRACE_STRING_LENGTH = 2_000;
 const MAX_TRACE_ARRAY_ITEMS = 20;
 const MAX_TRACE_OBJECT_KEYS = 20;
-const SYNTHETIC_TOOL_TRACE_NAMES = new Set(['updateWorkingMemory']);
+const SYNTHETIC_TOOL_TRACE_NAMES = new Set();
 function isRecord(value) {
     return value !== null && typeof value === 'object' && !Array.isArray(value);
 }
@@ -639,23 +639,7 @@ function getChunkPayload(chunk) {
     }
     return isRecord(chunk.payload) ? chunk.payload : chunk;
 }
-function isMemoryToolTrace(toolName) {
-    return toolName === 'updateWorkingMemory';
-}
-function summarizeWorkingMemoryInput(memory) {
-    return {
-        memory_chars: memory.length,
-        memory_lines: memory.length > 0 ? memory.split('\n').length : 0,
-        memory_preview: memory.length > 0 ? truncateTraceString(memory.slice(0, 400)) : '',
-    };
-}
-function buildSyntheticToolInputs(toolCallId, toolName, args) {
-    if (isMemoryToolTrace(toolName) && isRecord(args) && typeof args.memory === 'string') {
-        return sanitizeTracePayload({
-            toolCallId,
-            args: summarizeWorkingMemoryInput(args.memory),
-        });
-    }
+function buildSyntheticToolInputs(toolCallId, _toolName, args) {
     return sanitizeTracePayload({
         toolCallId,
         args,
@@ -668,38 +652,6 @@ function shouldCreateSyntheticToolTrace(payload) {
         payload.providerExecuted === true ||
         payload.dynamic === true);
 }
-function resolveActorParentRun(parentRun) {
-    let current = parentRun;
-    while (current) {
-        if (current.run_type !== 'llm' && current.run_type !== 'tool') {
-            return current;
-        }
-        const next = Reflect.get(current, 'parent_run');
-        current = next instanceof Object ? next : undefined;
-    }
-    return parentRun;
-}
-async function startSyntheticToolGroupRun(parentRun, toolName) {
-    if (!isMemoryToolTrace(toolName)) {
-        return undefined;
-    }
-    const actorParentRun = resolveActorParentRun(parentRun);
-    const groupRunTree = actorParentRun.createChild({
-        name: 'internal_state',
-        run_type: 'chain',
-        tags: dedupeTags([...(actorParentRun.tags ?? []), 'internal', 'memory']),
-        metadata: {
-            ...(actorParentRun.metadata ?? {}),
-            internal_state: true,
-            tool_name: toolName,
-        },
-        inputs: sanitizeTracePayload({
-            tool_name: toolName,
-        }),
-    });
-    await groupRunTree.postRun();
-    return groupRunTree;
-}
 async function startSyntheticToolTrace(chunk, records) {
     if (!isRecord(chunk) || chunk.type !== 'tool-call') {
         return;
@@ -717,22 +669,18 @@ async function startSyntheticToolTrace(chunk, records) {
     if (!parentRun) {
         return;
     }
-    const groupRunTree = await startSyntheticToolGroupRun(parentRun, toolName);
-    const toolParentRun = groupRunTree ?? parentRun;
-    const runTree = toolParentRun.createChild({
+    const runTree = parentRun.createChild({
         name: `tool:${toolName}`,
         run_type: 'tool',
         tags: dedupeTags([
-            ...(toolParentRun.tags ?? []),
+            ...(parentRun.tags ?? []),
             'tool',
             ...(toolName.startsWith('mastra_') ? ['native-tool'] : []),
-            ...(isMemoryToolTrace(toolName) ? ['memory', 'internal'] : []),
         ]),
         metadata: {
-            ...(toolParentRun.metadata ?? {}),
+            ...(parentRun.metadata ?? {}),
             tool_name: toolName,
             synthetic_tool_trace: true,
-            ...(isMemoryToolTrace(toolName) ? { memory_tool: true } : {}),
             ...(payload.providerExecuted === true ? { provider_executed: true } : {}),
             ...(payload.dynamic === true ? { dynamic_tool: true } : {}),
         },
@@ -742,7 +690,6 @@ async function startSyntheticToolTrace(chunk, records) {
     records.set(toolCallId, {
         toolCallId,
         toolName,
-        groupRunTree,
         runTree,
         finished: false,
     });
@@ -790,26 +737,6 @@ async function finishSyntheticToolTrace(chunk, records) {
             final_status: payload.isError === true ? 'error' : 'completed',
         },
     });
-    if (record.groupRunTree) {
-        await finishRunTree(record.groupRunTree, {
-            outputs: sanitizeTracePayload({
-                tool_name: record.toolName,
-            }),
-            ...(payload.isError === true
-                ? {
-                    error: typeof payload.result === 'string'
-                        ? payload.result
-                        : typeof payload.error === 'string'
-                            ? payload.error
-                            : 'Tool execution failed',
-                }
-                : {}),
-            metadata: {
-                final_status: payload.isError === true ? 'error' : 'completed',
-                internal_state: true,
-            },
-        });
-    }
 }
 async function finalizeSyntheticToolTraces(records, options) {
     for (const record of records.values()) {
@@ -826,19 +753,6 @@ async function finalizeSyntheticToolTraces(records, options) {
                 final_status: options?.status ?? 'completed',
             },
         });
-        if (record.groupRunTree) {
-            await finishRunTree(record.groupRunTree, {
-                outputs: sanitizeTracePayload({
-                    tool_name: record.toolName,
-                    status: options?.status ?? 'completed',
-                }),
-                ...(options?.error ? { error: options.error } : {}),
-                metadata: {
-                    final_status: options?.status ?? 'completed',
-                    internal_state: true,
-                },
-            });
-        }
     }
 }
 async function startLlmStepTrace(parentRun, messageId, request, stepNumber) {
@@ -1351,6 +1265,8 @@ async function executeResumableStream(options) {
     let activeStream = options.stream.fullStream;
     let activeMastraRunId = options.stream.runId ?? options.initialMastraRunId ?? '';
     let text = options.stream.text;
+    const workSummaryAccumulator = new work_summary_accumulator_1.WorkSummaryAccumulator();
+    let currentResponseId;
     while (true) {
         let suspension;
         let hasError = false;
@@ -1378,11 +1294,21 @@ async function executeResumableStream(options) {
                     status: 'cancelled',
                     error: 'Run cancelled while streaming',
                 });
-                return { status: 'cancelled', mastraRunId: activeMastraRunId, text };
+                return {
+                    status: 'cancelled',
+                    mastraRunId: activeMastraRunId,
+                    text,
+                    workSummary: workSummaryAccumulator.toSummary(),
+                };
             }
             await startSyntheticToolTrace(chunk, syntheticToolRecords);
             await finishSyntheticToolTrace(chunk, syntheticToolRecords);
             options.llmStepTraceHooks?.onStreamChunk(chunk);
+            if (isRecord(chunk) && chunk.type === 'step-start') {
+                const stepPayload = getChunkPayload(chunk);
+                const stepMessageId = typeof stepPayload?.messageId === 'string' ? stepPayload.messageId : undefined;
+                currentResponseId = stepMessageId;
+            }
             if (options.llmStepTraceHooks) {
             }
             else if (isRecord(chunk) && chunk.type === 'step-start') {
@@ -1410,7 +1336,7 @@ async function executeResumableStream(options) {
                     }
                 }
                 else if (!isSameSuspension(parsedSuspension, suspension)) {
-                    console.warn('[HITL] additional suspension encountered before resume; deferring', {
+                    options.context.logger.warn('Additional HITL suspension encountered before resume; deferring', {
                         threadId: options.context.threadId,
                         runId: options.context.runId,
                         activeRequestId: suspension.requestId,
@@ -1423,8 +1349,9 @@ async function executeResumableStream(options) {
             if (isErrorChunk(chunk)) {
                 hasError = true;
             }
-            const event = (0, map_chunk_1.mapMastraChunkToEvent)(options.context.runId, options.context.agentId, chunk);
+            const event = (0, map_chunk_1.mapMastraChunkToEvent)(options.context.runId, options.context.agentId, chunk, currentResponseId);
             if (event) {
+                workSummaryAccumulator.observe(event);
                 let shouldPublishEvent = true;
                 if (event.type === 'confirmation-request') {
                     const isPrimarySuspension = suspension !== undefined &&
@@ -1449,24 +1376,35 @@ async function executeResumableStream(options) {
                 publishCorrections(options.context, options.control.drainCorrections());
             }
         }
+        const traceStatus = suspension ? 'suspended' : hasError ? 'errored' : 'completed';
         if (options.llmStepTraceHooks) {
             await options.llmStepTraceHooks.finalize(activeSource, {
-                status: suspension ? 'suspended' : 'completed',
+                status: traceStatus,
             });
         }
         else {
             await finalizeLlmStepTraces(activeSource, llmStepRecords, {
-                status: suspension ? 'suspended' : 'completed',
+                status: traceStatus,
             });
         }
         await finalizeSyntheticToolTraces(syntheticToolRecords, {
-            status: suspension ? 'suspended' : 'completed',
+            status: traceStatus,
         });
         if (options.context.signal.aborted) {
-            return { status: 'cancelled', mastraRunId: activeMastraRunId, text };
+            return {
+                status: 'cancelled',
+                mastraRunId: activeMastraRunId,
+                text,
+                workSummary: workSummaryAccumulator.toSummary(),
+            };
         }
         if (!suspension) {
-            return { status: hasError ? 'errored' : 'completed', mastraRunId: activeMastraRunId, text };
+            return {
+                status: hasError ? 'errored' : 'completed',
+                mastraRunId: activeMastraRunId,
+                text,
+                workSummary: workSummaryAccumulator.toSummary(),
+            };
         }
         if (options.control.mode === 'manual') {
             return {
@@ -1474,10 +1412,12 @@ async function executeResumableStream(options) {
                 mastraRunId: activeMastraRunId,
                 text,
                 suspension,
+                workSummary: workSummaryAccumulator.toSummary(),
                 ...(confirmationEvent ? { confirmationEvent } : {}),
             };
         }
-        const resumeData = await waitForConfirmation(options.context.signal, pendingConfirmation ?? options.control.waitForConfirmation(suspension.requestId));
+        const confirmationPromise = pendingConfirmation ?? options.control.waitForConfirmation(suspension.requestId);
+        const resumeData = await waitForConfirmationOrCorrection(options.context.signal, confirmationPromise, options.control, options.context);
         const resumeOptions = options.control.buildResumeOptions?.({
             mastraRunId: activeMastraRunId,
             suspension,
@@ -1485,25 +1425,10 @@ async function executeResumableStream(options) {
             runId: activeMastraRunId,
             toolCallId: suspension.toolCallId,
         };
-        const resumed = options.workingMemoryEnabled
-            ? await (0, working_memory_tracing_1.traceWorkingMemoryContext)({
-                phase: 'resume',
-                agentId: options.context.agentId,
-                threadId: options.context.threadId,
-                resumeData: {
-                    requestId: suspension.requestId,
-                    toolCallId: suspension.toolCallId,
-                    ...(typeof resumeOptions.runId === 'string' ? { runId: resumeOptions.runId } : {}),
-                },
-                enabled: true,
-            }, async () => await (0, stream_helpers_1.asResumable)(options.agent).resumeStream(resumeData, {
-                ...resumeOptions,
-                ...(options.llmStepTraceHooks?.executionOptions ?? {}),
-            }))
-            : await (0, stream_helpers_1.asResumable)(options.agent).resumeStream(resumeData, {
-                ...resumeOptions,
-                ...(options.llmStepTraceHooks?.executionOptions ?? {}),
-            });
+        const resumed = await (0, stream_helpers_1.asResumable)(options.agent).resumeStream(resumeData, {
+            ...resumeOptions,
+            ...(options.llmStepTraceHooks?.executionOptions ?? {}),
+        });
         activeMastraRunId =
             (typeof resumed.runId === 'string' ? resumed.runId : '') || activeMastraRunId;
         activeSource = resumed;
@@ -1546,6 +1471,28 @@ async function waitForConfirmation(signal, confirmationPromise) {
         }
     }
 }
+async function waitForConfirmationOrCorrection(signal, confirmationPromise, control, context) {
+    if (!control.waitForCorrection) {
+        return await waitForConfirmation(signal, confirmationPromise);
+    }
+    const correctionSentinel = Object.freeze({ __correctionOverride: true });
+    const raced = Promise.race([
+        confirmationPromise,
+        control.waitForCorrection().then(() => correctionSentinel),
+    ]);
+    const result = await waitForConfirmation(signal, raced);
+    if (result === correctionSentinel) {
+        const corrections = control.drainCorrections?.() ?? [];
+        publishCorrections(context, corrections);
+        return {
+            __correctionOverride: true,
+            message: 'The user sent a correction while this tool was waiting for confirmation. ' +
+                'The confirmation has been skipped. Apply the correction and continue.',
+            corrections,
+        };
+    }
+    return result;
+}
 function isSameSuspension(left, right) {
     return left.requestId === right.requestId && left.toolCallId === right.toolCallId;
 }