yaml-flow 3.0.0 → 3.1.1

package/dist/event-graph/index.cjs

@@ -6566,7 +6566,7 @@ function addDynamicTask(graph, taskName, taskConfig) {
     }
   };
 }
-function createDefaultTaskState() {
+function createDefaultGraphEngineStore() {
   return {
     status: "not-started",
     executionCount: 0,
@@ -6579,7 +6579,7 @@ function createDefaultTaskState() {
 function createInitialExecutionState(graph, executionId) {
   const tasks = {};
   for (const taskName of Object.keys(graph.tasks)) {
-    tasks[taskName] = createDefaultTaskState();
+    tasks[taskName] = createDefaultGraphEngineStore();
   }
   return {
     status: "running",
@@ -7100,7 +7100,7 @@ function selectOptimalTasks(candidates, graph, state, conflictStrategy) {
 
 // src/event-graph/task-transitions.ts
 function applyTaskStart(state, taskName) {
-  const existingTask = state.tasks[taskName] ?? createDefaultTaskState2();
+  const existingTask = state.tasks[taskName] ?? createDefaultGraphEngineStore2();
   const updatedTask = {
     ...existingTask,
     status: "running",
@@ -7115,8 +7115,8 @@ function applyTaskStart(state, taskName) {
     lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
   };
 }
-function applyTaskCompletion(state, graph, taskName, result, dataHash) {
-  const existingTask = state.tasks[taskName] ?? createDefaultTaskState2();
+function applyTaskCompletion(state, graph, taskName, result, dataHash, data) {
+  const existingTask = state.tasks[taskName] ?? createDefaultGraphEngineStore2();
   const taskConfig = graph.tasks[taskName];
   if (!taskConfig) {
     throw new Error(`Task "${taskName}" not found in graph`);
@@ -7148,6 +7148,7 @@ function applyTaskCompletion(state, graph, taskName, result, dataHash) {
     executionCount: existingTask.executionCount + 1,
     lastEpoch: existingTask.executionCount + 1,
     lastDataHash: dataHash,
+    data,
     lastConsumedHashes,
     error: void 0
   };
@@ -7160,7 +7161,7 @@ function applyTaskCompletion(state, graph, taskName, result, dataHash) {
   };
 }
 function applyTaskFailure(state, graph, taskName, error) {
-  const existingTask = state.tasks[taskName] ?? createDefaultTaskState2();
+  const existingTask = state.tasks[taskName] ?? createDefaultGraphEngineStore2();
   const taskConfig = graph.tasks[taskName];
   if (taskConfig?.retry) {
     const retryCount = existingTask.retryCount + 1;
@@ -7203,7 +7204,7 @@ function applyTaskFailure(state, graph, taskName, error) {
   };
 }
 function applyTaskProgress(state, taskName, message, progress) {
-  const existingTask = state.tasks[taskName] ?? createDefaultTaskState2();
+  const existingTask = state.tasks[taskName] ?? createDefaultGraphEngineStore2();
   const updatedTask = {
     ...existingTask,
     progress: typeof progress === "number" ? progress : existingTask.progress,
@@ -7219,7 +7220,27 @@ function applyTaskProgress(state, taskName, message, progress) {
     lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
   };
 }
-function createDefaultTaskState2() {
+function applyTaskRestart(state, taskName) {
+  const existingTask = state.tasks[taskName];
+  if (!existingTask) return state;
+  const updatedTask = {
+    ...existingTask,
+    status: "not-started",
+    startedAt: void 0,
+    completedAt: void 0,
+    failedAt: void 0,
+    error: void 0,
+    data: void 0,
+    progress: null,
+    lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  return {
+    ...state,
+    tasks: { ...state.tasks, [taskName]: updatedTask },
+    lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
+  };
+}
+function createDefaultGraphEngineStore2() {
   return {
     status: "not-started",
     executionCount: 0,
@@ -7239,11 +7260,13 @@ function apply(state, event, graph) {
     case "task-started":
       return applyTaskStart(state, event.taskName);
     case "task-completed":
-      return applyTaskCompletion(state, graph, event.taskName, event.result, event.dataHash);
+      return applyTaskCompletion(state, graph, event.taskName, event.result, event.dataHash, event.data);
     case "task-failed":
       return applyTaskFailure(state, graph, event.taskName, event.error);
     case "task-progress":
       return applyTaskProgress(state, event.taskName, event.message, event.progress);
+    case "task-restart":
+      return applyTaskRestart(state, event.taskName);
     case "inject-tokens":
       return applyInjectTokens(state, event.tokens);
     case "agent-action":
@@ -7314,7 +7337,7 @@ function applyTaskCreation(state, taskName, taskConfig) {
     ...state,
     tasks: {
       ...state.tasks,
-      [taskName]: createDefaultTaskState()
+      [taskName]: createDefaultGraphEngineStore()
     },
     lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
   };
@@ -8213,9 +8236,10 @@ exports.applyAll = applyAll;
 exports.applyTaskCompletion = applyTaskCompletion;
 exports.applyTaskFailure = applyTaskFailure;
 exports.applyTaskProgress = applyTaskProgress;
+exports.applyTaskRestart = applyTaskRestart;
 exports.applyTaskStart = applyTaskStart;
 exports.computeAvailableOutputs = computeAvailableOutputs;
-exports.createDefaultTaskState = createDefaultTaskState;
+exports.createDefaultGraphEngineStore = createDefaultGraphEngineStore;
 exports.createInitialExecutionState = createInitialExecutionState;
 exports.detectStuckState = detectStuckState;
 exports.exportGraphConfig = exportGraphConfig;