@elsium-ai/workflows 0.3.0 → 0.4.1

package/dist/dag.d.ts

@@ -0,0 +1,4 @@
+import type { DagWorkflowConfig } from './types';
+import type { Workflow } from './workflow';
+export declare function defineDagWorkflow(config: DagWorkflowConfig): Workflow;
+//# sourceMappingURL=dag.d.ts.map

package/dist/dag.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dag.d.ts","sourceRoot":"","sources":["../src/dag.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAEX,iBAAiB,EAKjB,MAAM,SAAS,CAAA;AAChB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAkJ1C,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,iBAAiB,GAAG,QAAQ,CA+BrE"}

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 export { step, executeStep } from './step';
 export { defineWorkflow, defineParallelWorkflow, defineBranchWorkflow } from './workflow';
 export type { Workflow, ParallelWorkflowConfig, BranchConfig } from './workflow';
-export type { StepConfig, StepContext, StepResult, StepStatus, RetryConfig, WorkflowConfig, WorkflowResult, WorkflowStatus, WorkflowRunOptions, } from './types';
+export { defineDagWorkflow } from './dag';
+export type { StepConfig, StepContext, StepResult, StepStatus, RetryConfig, WorkflowConfig, WorkflowResult, WorkflowStatus, WorkflowRunOptions, DagStepConfig, DagWorkflowConfig, } from './types';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAA;AAG1C,OAAO,EAAE,cAAc,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AACzF,YAAY,EAAE,QAAQ,EAAE,sBAAsB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAGhF,YAAY,EACX,UAAU,EACV,WAAW,EACX,UAAU,EACV,UAAU,EACV,WAAW,EACX,cAAc,EACd,cAAc,EACd,cAAc,EACd,kBAAkB,GAClB,MAAM,SAAS,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAA;AAG1C,OAAO,EAAE,cAAc,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAA;AACzF,YAAY,EAAE,QAAQ,EAAE,sBAAsB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAA;AAGhF,OAAO,EAAE,iBAAiB,EAAE,MAAM,OAAO,CAAA;AAGzC,YAAY,EACX,UAAU,EACV,WAAW,EACX,UAAU,EACV,UAAU,EACV,WAAW,EACX,cAAc,EACd,cAAc,EACd,cAAc,EACd,kBAAkB,EAClB,aAAa,EACb,iBAAiB,GACjB,MAAM,SAAS,CAAA"}

package/dist/index.js

@@ -372,10 +372,128 @@ function defineBranchWorkflow(name, branches, fallback) {
     }
   };
 }
+// src/dag.ts
+function buildDependencyMaps(steps) {
+  const stepMap = new Map(steps.map((s) => [s.name, s]));
+  const inDegree = new Map;
+  const dependents = new Map;
+  for (const step2 of steps) {
+    inDegree.set(step2.name, 0);
+    dependents.set(step2.name, []);
+  }
+  for (const step2 of steps) {
+    validateAndRegisterDeps(step2, stepMap, inDegree, dependents);
+  }
+  return { inDegree, dependents };
+}
+function validateAndRegisterDeps(step2, stepMap, inDegree, dependents) {
+  for (const dep of step2.dependsOn ?? []) {
+    if (!stepMap.has(dep)) {
+      throw new Error(`Step "${step2.name}" depends on unknown step "${dep}"`);
+    }
+    inDegree.set(step2.name, (inDegree.get(step2.name) ?? 0) + 1);
+    dependents.get(dep)?.push(step2.name);
+  }
+}
+function findNextWave(inDegree, resolved, steps) {
+  const wave = [];
+  for (const [name, degree] of inDegree) {
+    if (degree === 0 && !resolved.has(name)) {
+      wave.push(name);
+    }
+  }
+  if (wave.length === 0) {
+    const remaining = steps.filter((s) => !resolved.has(s.name)).map((s) => s.name);
+    throw new Error(`Cycle detected in DAG workflow. Remaining steps: ${remaining.join(", ")}`);
+  }
+  return wave;
+}
+function resolveWave(wave, resolved, inDegree, dependents) {
+  for (const name of wave) {
+    resolved.add(name);
+    for (const dep of dependents.get(name) ?? []) {
+      inDegree.set(dep, (inDegree.get(dep) ?? 0) - 1);
+    }
+  }
+}
+function topologicalSort(steps) {
+  const { inDegree, dependents } = buildDependencyMaps(steps);
+  const waves = [];
+  const resolved = new Set;
+  while (resolved.size < steps.length) {
+    const wave = findNextWave(inDegree, resolved, steps);
+    resolveWave(wave, resolved, inDegree, dependents);
+    waves.push(wave);
+  }
+  return waves;
+}
+function resolveStepInput(stepConfig, outputs, input) {
+  const deps = stepConfig.dependsOn ?? [];
+  return deps.length > 0 ? outputs[deps[0]] ?? input : input;
+}
+function executeWaveSteps(wave, stepMap, config, input, outputs, options) {
+  const wavePromises = wave.map((stepName) => stepMap.get(stepName)).filter((s) => s !== undefined).map((stepConfig) => {
+    const stepInput = resolveStepInput(stepConfig, outputs, input);
+    const context = {
+      workflowName: config.name,
+      stepIndex: config.steps.indexOf(stepConfig),
+      previousOutputs: { ...outputs },
+      signal: options?.signal
+    };
+    return executeStep(stepConfig, stepInput, context);
+  });
+  return Promise.all(wavePromises);
+}
+async function processWaveResults(waveResults, wave, stepResults, outputs, config) {
+  let failed = false;
+  for (let i = 0;i < waveResults.length; i++) {
+    const result = waveResults[i];
+    const stepName = wave[i];
+    stepResults.push(result);
+    if (result.status === "completed" && result.data !== undefined) {
+      outputs[stepName] = result.data;
+    }
+    await config.onStepComplete?.(result);
+    if (result.status === "failed") {
+      failed = true;
+      await config.onStepError?.(new Error(result.error ?? "Step failed"), stepName);
+    }
+  }
+  return failed;
+}
+function defineDagWorkflow(config) {
+  const stepMap = new Map(config.steps.map((s) => [s.name, s]));
+  return {
+    name: config.name,
+    async run(input, options) {
+      const startTime = performance.now();
+      const waves = topologicalSort(config.steps);
+      const stepResults = [];
+      const outputs = {};
+      let failed = false;
+      for (const wave of waves) {
+        const waveResults = await executeWaveSteps(wave, stepMap, config, input, outputs, options);
+        failed = await processWaveResults(waveResults, wave, stepResults, outputs, config);
+        if (failed)
+          break;
+      }
+      const workflowResult = {
+        name: config.name,
+        status: failed ? "failed" : "completed",
+        steps: stepResults,
+        totalDurationMs: Math.round(performance.now() - startTime),
+        outputs
+      };
+      await config.onComplete?.(workflowResult);
+      return workflowResult;
+    }
+  };
+}
 export {
   step,
   executeStep,
   defineWorkflow,
   defineParallelWorkflow,
+  defineDagWorkflow,
   defineBranchWorkflow
 };

package/dist/types.d.ts

@@ -47,4 +47,14 @@ export interface WorkflowResult {
 export interface WorkflowRunOptions {
     signal?: AbortSignal;
 }
+export interface DagStepConfig<TInput = unknown, TOutput = unknown> extends StepConfig<TInput, TOutput> {
+    dependsOn?: string[];
+}
+export interface DagWorkflowConfig {
+    name: string;
+    steps: DagStepConfig[];
+    onStepComplete?: (result: StepResult) => void | Promise<void>;
+    onStepError?: (error: Error, stepName: string) => void | Promise<void>;
+    onComplete?: (result: WorkflowResult) => void | Promise<void>;
+}
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAI5B,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAA;AAEnF,MAAM,WAAW,UAAU,CAAC,MAAM,GAAG,OAAO,EAAE,OAAO,GAAG,OAAO;IAC9D,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACzB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;IAClE,KAAK,CAAC,EAAE,WAAW,CAAA;IACnB,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,KAAK,OAAO,CAAA;IAC5D,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;IAC5D,SAAS,CAAC,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,WAAW,WAAW;IAC3B,YAAY,EAAE,MAAM,CAAA;IACpB,SAAS,EAAE,MAAM,CAAA;IACjB,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACxC,MAAM,CAAC,EAAE,WAAW,CAAA;CACpB;AAED,MAAM,WAAW,UAAU,CAAC,CAAC,GAAG,OAAO;IACtC,IAAI,EAAE,MAAM,CAAA;IACZ,MAAM,EAAE,UAAU,CAAA;IAClB,IAAI,CAAC,EAAE,CAAC,CAAA;IACR,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,UAAU,EAAE,MAAM,CAAA;IAClB,UAAU,EAAE,MAAM,CAAA;CAClB;AAID,MAAM,WAAW,WAAW;IAC3B,UAAU,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAA;CACvC;AAID,MAAM,MAAM,cAAc,GAAG,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAA;AAE3E,MAAM,WAAW,cAAc;IAC9B,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,UAAU,EAAE,CAAA;IACnB,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC7D,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACtE,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CAC7D;AAED,MAAM,WAAW,cAAc;IAC9B,IAAI,EAAE,MAAM,CAAA;IACZ,MAAM,EAAE,cAAc,CAAA;IACtB,KAAK,EAAE,UAAU,EAAE,CAAA;IACnB,eAAe,EAAE,MAAM,CAAA;IACvB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAChC;AAED,MAAM,WAAW,kBAAkB;IAClC,MAAM,CAAC,EAAE,WAAW,CAAA;CACpB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAI5B,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAA;AAEnF,MAAM,WAAW,UAAU,CAAC,MAAM,GAAG,OAAO,EAAE,OAAO,GAAG,OAAO;IAC9D,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACzB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;IAClE,KAAK,CAAC,EAAE,WAAW,CAAA;IACnB,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,KAAK,OAAO,CAAA;IAC5D,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;IAC5D,SAAS,CAAC,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,WAAW,WAAW;IAC3B,YAAY,EAAE,MAAM,CAAA;IACpB,SAAS,EAAE,MAAM,CAAA;IACjB,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACxC,MAAM,CAAC,EAAE,WAAW,CAAA;CACpB;AAED,MAAM,WAAW,UAAU,CAAC,CAAC,GAAG,OAAO;IACtC,IAAI,EAAE,MAAM,CAAA;IACZ,MAAM,EAAE,UAAU,CAAA;IAClB,IAAI,CAAC,EAAE,CAAC,CAAA;IACR,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,UAAU,EAAE,MAAM,CAAA;IAClB,UAAU,EAAE,MAAM,CAAA;CAClB;AAID,MAAM,WAAW,WAAW;IAC3B,UAAU,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAA;CACvC;AAID,MAAM,MAAM,cAAc,GAAG,SAAS,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAA;AAE3E,MAAM,WAAW,cAAc;IAC9B,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,UAAU,EAAE,CAAA;IACnB,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC7D,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACtE,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CAC7D;AAED,MAAM,WAAW,cAAc;IAC9B,IAAI,EAAE,MAAM,CAAA;IACZ,MAAM,EAAE,cAAc,CAAA;IACtB,KAAK,EAAE,UAAU,EAAE,CAAA;IACnB,eAAe,EAAE,MAAM,CAAA;IACvB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAChC;AAED,MAAM,WAAW,kBAAkB;IAClC,MAAM,CAAC,EAAE,WAAW,CAAA;CACpB;AAID,MAAM,WAAW,aAAa,CAAC,MAAM,GAAG,OAAO,EAAE,OAAO,GAAG,OAAO,CACjE,SAAQ,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC;IACnC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAA;CACpB;AAED,MAAM,WAAW,iBAAiB;IACjC,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,aAAa,EAAE,CAAA;IACtB,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAC7D,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACtE,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,cAAc,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CAC7D"}

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@elsium-ai/workflows",
-	"version": "0.3.0",
+	"version": "0.4.1",
 	"description": "Multi-step workflow pipelines and DAG execution for ElsiumAI",
 	"license": "MIT",
 	"author": "Eric Utrera <ebutrera9103@gmail.com>",
@@ -26,7 +26,7 @@
 		"dev": "bun --watch src/index.ts"
 	},
 	"dependencies": {
-		"@elsium-ai/core": "^0.3.0",
+		"@elsium-ai/core": "^0.4.1",
 		"zod": "^3.24.0"
 	},
 	"devDependencies": {