@ai-setting/roy-agent-core 1.7.66 → 1.7.68

package/dist/shared/@ai-setting/{roy-agent-core-nr2m4xw1.js → roy-agent-core-1k83dq3a.js}

@@ -7,7 +7,7 @@ import {
 import {
   BackgroundTaskManager,
   createDelegateTool
-} from "./roy-agent-core-2wxbx36p.js";
+} from "./roy-agent-core-dnjm17sa.js";
 import {
   TaskEntityEventTypes
 } from "./roy-agent-core-8gxth0eh.js";
@@ -22,7 +22,7 @@ import {
   searchTasksTool,
   treeTasksTool,
   updateTaskTool
-} from "./roy-agent-core-t0ktaevq.js";
+} from "./roy-agent-core-h5vmff8w.js";
 import {
   createOperationTool,
   deleteOperationTool,
@@ -198,6 +198,8 @@ class TaskComponent extends BaseComponent {
   config;
   toolComponent;
   backgroundTaskManager;
+  skipAbortRunningTasksOnDispose = false;
+  waitForRunningTasksOnDispose = false;
   configComponent;
   tagService;
   plugins = new Map;
@@ -227,6 +229,8 @@ class TaskComponent extends BaseComponent {
       throw new Error("ConfigComponent is required for TaskComponent initialization");
     }
     this.configComponent = options.configComponent;
+    this.skipAbortRunningTasksOnDispose = options.skipAbortRunningTasksOnDispose === true;
+    this.waitForRunningTasksOnDispose = options.waitForRunningTasksOnDispose === true;
     await this.registerConfig(options);
     this.setStatus("running");
     logger2.info("[TaskComponent] Initialized");
@@ -350,7 +354,10 @@ class TaskComponent extends BaseComponent {
       }
     }
     if (this.backgroundTaskManager) {
-      this.backgroundTaskManager.dispose();
+      await this.backgroundTaskManager.dispose(this.skipAbortRunningTasksOnDispose ? {
+        skipAbortRunningTasks: true,
+        waitForRunningTasks: this.waitForRunningTasksOnDispose
+      } : undefined);
       this.backgroundTaskManager = undefined;
     }
     if (this.store) {
@@ -539,6 +546,9 @@ class TaskComponent extends BaseComponent {
     }
     return task;
   }
+  async updateTaskContext(id, context) {
+    return this.store.updateTask(id, { context });
+  }
   async createOperation(options) {
     const createCtx = {
       options: { ...options }