@bratsos/workflow-engine 0.0.11 → 0.2.0

package/dist/index.js

@@ -0,0 +1,399 @@
+import { ModelKey } from './chunk-P4KMGCT3.js';
+export { AVAILABLE_MODELS, AnthropicBatchProvider, DEFAULT_MODEL_KEY, GoogleBatchProvider, ModelKey, ModelStatsTracker, NoInputSchema, OpenAIBatchProvider, calculateCost, createAIHelper, defineAsyncBatchStage, defineStage, getBestProviderForModel, getDefaultModel, getModel, getModelById, getRegisteredModel, listModels, listRegisteredModels, modelSupportsBatch, printAvailableModels, registerModels, requireStageOutput, resolveModelForProvider } from './chunk-P4KMGCT3.js';
+import './chunk-D7RVRRM2.js';
+export { PrismaAICallLogger, PrismaJobQueue, PrismaWorkflowPersistence, createPrismaAICallLogger, createPrismaJobQueue, createPrismaWorkflowPersistence } from './chunk-NYKMT46J.js';
+import './chunk-MUWP5SF2.js';
+export { StaleVersionError } from './chunk-SPXBCZLB.js';
+export { IdempotencyInProgressError, createKernel, createPluginRunner, definePlugin } from './chunk-HL3OJG7W.js';
+import { z } from 'zod';
+
+var Workflow = class {
+  constructor(id, name, description, inputSchema, outputSchema, stages, contextType) {
+    this.id = id;
+    this.name = name;
+    this.description = description;
+    this.inputSchema = inputSchema;
+    this.outputSchema = outputSchema;
+    this.stages = stages;
+    this.contextType = contextType;
+  }
+  /**
+   * Get execution plan as groups of stages
+   * Stages in the same group can be executed in parallel
+   */
+  getExecutionPlan() {
+    const groups = /* @__PURE__ */ new Map();
+    for (const node of this.stages) {
+      const group = groups.get(node.executionGroup) || [];
+      group.push(node);
+      groups.set(node.executionGroup, group);
+    }
+    const sortedGroups = Array.from(groups.keys()).sort((a, b) => a - b);
+    return sortedGroups.map((groupNum) => {
+      const group = groups.get(groupNum);
+      if (!group) throw new Error(`Group ${groupNum} not found`);
+      return group;
+    });
+  }
+  /**
+   * Get a specific stage by ID
+   */
+  getStage(stageId) {
+    const node = this.stages.find((n) => n.stage.id === stageId);
+    return node?.stage;
+  }
+  /**
+   * Get all stages in order
+   */
+  getAllStages() {
+    return [...this.stages];
+  }
+  /**
+   * Get a visual representation of the workflow execution order
+   */
+  getExecutionOrder() {
+    const executionPlan = this.getExecutionPlan();
+    const lines = [];
+    lines.push(`Workflow: ${this.name} (${this.id})`);
+    lines.push(`Total stages: ${this.stages.length}`);
+    lines.push(`Execution groups: ${executionPlan.length}`);
+    lines.push("");
+    lines.push("Execution Order:");
+    lines.push("================");
+    for (let i = 0; i < executionPlan.length; i++) {
+      const group = executionPlan[i];
+      const groupNumber = i + 1;
+      if (group.length === 1) {
+        const stage = group[0].stage;
+        lines.push(`${groupNumber}. ${stage.name} (${stage.id})`);
+        if (stage.description) {
+          lines.push(`   ${stage.description}`);
+        }
+      } else {
+        lines.push(`${groupNumber}. [PARALLEL]`);
+        for (const node of group) {
+          const stage = node.stage;
+          lines.push(`   - ${stage.name} (${stage.id})`);
+          if (stage.description) {
+            lines.push(`     ${stage.description}`);
+          }
+        }
+      }
+      lines.push("");
+    }
+    return lines.join("\n");
+  }
+  /**
+   * Get all stage IDs in execution order
+   *
+   * @returns Array of stage IDs
+   *
+   * @example
+   * ```typescript
+   * const ids = workflow.getStageIds();
+   * // ["data-extraction", "guidelines", "generator"]
+   * ```
+   */
+  getStageIds() {
+    return this.stages.map((node) => node.stage.id);
+  }
+  /**
+   * Check if a stage ID exists in this workflow
+   *
+   * @param stageId - The stage ID to check
+   * @returns true if the stage exists
+   */
+  hasStage(stageId) {
+    return this.stages.some((node) => node.stage.id === stageId);
+  }
+  /**
+   * Validate workflow configuration before execution
+   * Checks that all stage configs match their schemas
+   *
+   * @param config - Configuration object with keys matching stage IDs
+   * @returns Validation result with any errors
+   */
+  validateConfig(config) {
+    const errors = [];
+    for (const node of this.stages) {
+      const stage = node.stage;
+      const stageConfig = config[stage.id] || {};
+      try {
+        stage.configSchema.parse(stageConfig);
+      } catch (error) {
+        if (error instanceof z.ZodError) {
+          const errorMessages = error.issues.map((e) => `${e.path.join(".")}: ${e.message}`).join("; ");
+          errors.push({
+            stageId: stage.id,
+            error: `Config validation failed: ${errorMessages}`
+          });
+        } else {
+          errors.push({
+            stageId: stage.id,
+            error: String(error)
+          });
+        }
+      }
+    }
+    return {
+      valid: errors.length === 0,
+      errors
+    };
+  }
+  /**
+   * Estimate total cost for the workflow
+   */
+  estimateCost(input, config) {
+    let totalCost = 0;
+    const currentInput = input;
+    for (const node of this.stages) {
+      if (node.stage.estimateCost) {
+        const stageConfig = config[node.stage.id] || {};
+        totalCost += node.stage.estimateCost(currentInput, stageConfig);
+      }
+    }
+    return totalCost;
+  }
+  /**
+   * Get configuration schemas for all stages in this workflow
+   * Returns a map of stageId → { schema, defaults, name, description }
+   */
+  getStageConfigs() {
+    const configs = {};
+    for (const node of this.stages) {
+      const stage = node.stage;
+      const defaults = {};
+      if (stage.configSchema instanceof z.ZodObject) {
+        const shape = stage.configSchema.shape;
+        for (const [key, fieldSchema] of Object.entries(shape)) {
+          let unwrapped = fieldSchema;
+          if (unwrapped instanceof z.ZodOptional) {
+            unwrapped = unwrapped._def.innerType;
+          }
+          if (unwrapped instanceof z.ZodDefault) {
+            const defaultValueFn = unwrapped._def.defaultValue;
+            defaults[key] = typeof defaultValueFn === "function" ? defaultValueFn() : defaultValueFn;
+          }
+        }
+      }
+      configs[stage.id] = {
+        schema: stage.configSchema,
+        defaults,
+        name: stage.name,
+        description: stage.description
+      };
+    }
+    return configs;
+  }
+  /**
+   * Generate default configuration object for all stages
+   * Automatically discovers all stage configs - add/remove stages and this updates automatically
+   */
+  getDefaultConfig() {
+    const stageConfigs = this.getStageConfigs();
+    const config = {};
+    for (const [stageId, meta] of Object.entries(stageConfigs)) {
+      config[stageId] = meta.defaults;
+    }
+    return config;
+  }
+  /**
+   * Get all stages in a specific execution group
+   */
+  getStagesInExecutionGroup(groupIndex) {
+    return this.stages.filter((node) => node.executionGroup === groupIndex).map((node) => node.stage);
+  }
+  /**
+   * Get the sequential index of a stage (0-based)
+   */
+  getStageIndex(stageId) {
+    return this.stages.findIndex((node) => node.stage.id === stageId);
+  }
+  /**
+   * Get the execution group index for a stage
+   */
+  getExecutionGroupIndex(stageId) {
+    const node = this.stages.find((node2) => node2.stage.id === stageId);
+    if (!node) throw new Error(`Stage ${stageId} not found in workflow`);
+    return node.executionGroup;
+  }
+  /**
+   * Get the ID of the stage immediately preceding the given stage
+   */
+  getPreviousStageId(stageId) {
+    const index = this.getStageIndex(stageId);
+    if (index <= 0) return void 0;
+    return this.stages[index - 1].stage.id;
+  }
+};
+var WorkflowBuilder = class {
+  constructor(id, name, description, inputSchema, currentOutputSchema) {
+    this.id = id;
+    this.name = name;
+    this.description = description;
+    this.inputSchema = inputSchema;
+    this.currentOutputSchema = currentOutputSchema;
+  }
+  stages = [];
+  currentExecutionGroup = 0;
+  /**
+   * Add a stage to the workflow (sequential execution)
+   *
+   * Automatically accumulates the stage's output in the context under its stage ID.
+   * This provides type-safe access to all previous stage outputs.
+   *
+   * Note: This accepts any stage regardless of strict input type matching.
+   * This is necessary because stages using passthrough() can accept objects
+   * with additional fields beyond what's declared in their input schema.
+   * Runtime validation via Zod ensures type safety at execution time.
+   *
+   * Validates that all declared dependencies exist in the workflow.
+   */
+  pipe(stage) {
+    if (stage.dependencies) {
+      const existingStageIds = this.stages.map((s) => s.stage.id);
+      const missingDeps = stage.dependencies.filter(
+        (dep) => !existingStageIds.includes(dep)
+      );
+      if (missingDeps.length > 0) {
+        throw new Error(
+          `Stage "${stage.id}" has missing dependencies: ${missingDeps.join(", ")}. These stages must be added to the workflow before "${stage.id}". Current stages: ${existingStageIds.length === 0 ? "(none)" : existingStageIds.join(", ")}`
+        );
+      }
+    }
+    this.currentExecutionGroup++;
+    this.stages.push({
+      stage,
+      executionGroup: this.currentExecutionGroup
+    });
+    const builder = this;
+    builder.currentOutputSchema = stage.outputSchema;
+    return builder;
+  }
+  /**
+   * Add a stage with strict input type checking
+   *
+   * Note: pipeStrict() and pipeLoose() have been removed as they were
+   * just aliases for pipe(). Use pipe() for all stage chaining.
+   */
+  /**
+   * Add multiple stages that execute in parallel
+   *
+   * All stages receive the same input (current output)
+   * Their outputs are merged into an object by index AND accumulated in context by stage ID.
+   *
+   * Note: This accepts stages regardless of strict input type matching.
+   * This is necessary because stages using passthrough() can accept objects
+   * with additional fields. Runtime validation via Zod ensures type safety.
+   *
+   * Validates that all declared dependencies exist in the workflow.
+   */
+  parallel(stages) {
+    const existingStageIds = this.stages.map((s) => s.stage.id);
+    for (const stage of stages) {
+      if (stage.dependencies) {
+        const missingDeps = stage.dependencies.filter(
+          (dep) => !existingStageIds.includes(dep)
+        );
+        if (missingDeps.length > 0) {
+          throw new Error(
+            `Stage "${stage.id}" (in parallel group) has missing dependencies: ${missingDeps.join(
+              ", "
+            )}. These stages must be added to the workflow before this parallel group. Current stages: ${existingStageIds.length === 0 ? "(none)" : existingStageIds.join(", ")}`
+          );
+        }
+      }
+    }
+    this.currentExecutionGroup++;
+    for (const stage of stages) {
+      this.stages.push({
+        stage,
+        executionGroup: this.currentExecutionGroup
+      });
+    }
+    const mergedSchema = z.object(
+      stages.reduce(
+        (acc, stage, index) => {
+          acc[index] = stage.outputSchema;
+          return acc;
+        },
+        {}
+      )
+    );
+    const builder = this;
+    builder.currentOutputSchema = mergedSchema;
+    return builder;
+  }
+  /**
+   * Build the final workflow
+   */
+  build() {
+    return new Workflow(
+      this.id,
+      this.name,
+      this.description,
+      this.inputSchema,
+      this.currentOutputSchema,
+      this.stages
+    );
+  }
+  /**
+   * Get current stage count
+   */
+  getStageCount() {
+    return this.stages.length;
+  }
+  /**
+   * Get execution group count
+   */
+  getExecutionGroupCount() {
+    return this.currentExecutionGroup;
+  }
+};
+var AIConfigSchema = z.object({
+  /** The model to use for AI operations */
+  model: ModelKey.default("gemini-2.5-flash"),
+  /** Temperature for AI generations (0-2) */
+  temperature: z.number().min(0).max(2).default(0.7),
+  /** Maximum tokens to generate (undefined = model default) */
+  maxTokens: z.number().positive().optional()
+});
+var ConcurrencyConfigSchema = z.object({
+  /** Maximum concurrent operations (for parallel processing) */
+  concurrency: z.number().positive().default(5),
+  /** Delay between operations in milliseconds (rate limiting) */
+  delayMs: z.number().nonnegative().default(0),
+  /** Maximum retries on failure */
+  maxRetries: z.number().nonnegative().default(3)
+});
+var FeatureFlagsConfigSchema = z.object({
+  /** Feature flags for conditional stage behavior */
+  featureFlags: z.record(z.string(), z.boolean()).default({})
+});
+var DebugConfigSchema = z.object({
+  /** Enable verbose logging */
+  verbose: z.boolean().default(false),
+  /** Dry run mode (no side effects) */
+  dryRun: z.boolean().default(false)
+});
+function withAIConfig(schema) {
+  return schema.merge(AIConfigSchema);
+}
+function withConcurrency(schema) {
+  return schema.merge(ConcurrencyConfigSchema);
+}
+function withFeatureFlags(schema) {
+  return schema.merge(FeatureFlagsConfigSchema);
+}
+function withStandardConfig(schema) {
+  return schema.merge(AIConfigSchema).merge(ConcurrencyConfigSchema).merge(FeatureFlagsConfigSchema);
+}
+z.object({});
+z.object({
+  model: ModelKey.default("gemini-2.5-flash"),
+  temperature: z.number().min(0).max(2).default(0.7)
+});
+
+export { AIConfigSchema, ConcurrencyConfigSchema, DebugConfigSchema, FeatureFlagsConfigSchema, Workflow, WorkflowBuilder, withAIConfig, withConcurrency, withFeatureFlags, withStandardConfig };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/core/workflow.ts","../src/core/config-presets.ts"],"names":["node","z"],"mappings":";;;;;;;;;AA2CO,IAAM,WAAN,MAIL;AAAA,EACA,YACkB,EAAA,EACA,IAAA,EACA,aACA,WAAA,EACA,YAAA,EACC,QACD,WAAA,EAChB;AAPgB,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA;AACA,IAAA,IAAA,CAAA,IAAA,GAAA,IAAA;AACA,IAAA,IAAA,CAAA,WAAA,GAAA,WAAA;AACA,IAAA,IAAA,CAAA,WAAA,GAAA,WAAA;AACA,IAAA,IAAA,CAAA,YAAA,GAAA,YAAA;AACC,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACD,IAAA,IAAA,CAAA,WAAA,GAAA,WAAA;AAAA,EACf;AAAA;AAAA;AAAA;AAAA;AAAA,EAMH,gBAAA,GAAkC;AAChC,IAAA,MAAM,MAAA,uBAAa,GAAA,EAAyB;AAE5C,IAAA,KAAA,MAAW,IAAA,IAAQ,KAAK,MAAA,EAAQ;AAC9B,MAAA,MAAM,QAAQ,MAAA,CAAO,GAAA,CAAI,IAAA,CAAK,cAAc,KAAK,EAAC;AAClD,MAAA,KAAA,CAAM,KAAK,IAAI,CAAA;AACf,MAAA,MAAA,CAAO,GAAA,CAAI,IAAA,CAAK,cAAA,EAAgB,KAAK,CAAA;AAAA,IACvC;AAGA,IAAA,MAAM,YAAA,GAAe,KAAA,CAAM,IAAA,CAAK,MAAA,CAAO,IAAA,EAAM,CAAA,CAAE,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAM,CAAA,GAAI,CAAC,CAAA;AACnE,IAAA,OAAO,YAAA,CAAa,GAAA,CAAI,CAAC,QAAA,KAAa;AACpC,MAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,GAAA,CAAI,QAAQ,CAAA;AACjC,MAAA,IAAI,CAAC,KAAA,EAAO,MAAM,IAAI,KAAA,CAAM,CAAA,MAAA,EAAS,QAAQ,CAAA,UAAA,CAAY,CAAA;AACzD,MAAA,OAAO,KAAA;AAAA,IACT,CAAC,CAAA;AAAA,EACH;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS,OAAA,EAAmD;AAC1D,IAAA,MAAM,IAAA,GAAO,KAAK,MAAA,CAAO,IAAA,CAAK,CAAC,CAAA,KAAM,CAAA,CAAE,KAAA,CAAM,EAAA,KAAO,OAAO,CAAA;AAC3D,IAAA,OAAO,IAAA,EAAM,KAAA;AAAA,EACf;AAAA;AAAA;AAAA;AAAA,EAKA,YAAA,GAA4B;AAC1B,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,MAAM,CAAA;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,iBAAA,GAA4B;AAC1B,IAAA,MAAM,aAAA,GAAgB,KAAK,gBAAA,EAAiB;AAC5C,IAAA,MAAM,QAAkB,EAAC;AAEzB,IAAA,KAAA,CAAM,KAAK,CAAA,UAAA,EAAa,IAAA,CAAK,IAAI,CAAA,EAAA,EAAK,IAAA,CAAK,EAAE,CAAA,CAAA,CAAG,CAAA;AAChD,IAAA,KAAA,CAAM,IAAA,CAAK,CAAA,cAAA,EAAiB,IAAA,CAAK,MAAA,CAAO,MAAM,CAAA,CAAE,CAAA;AAChD,IAAA,KAAA,CAAM,IAAA,CAAK,CAAA,kBAAA,EAAqB,aAAA,CAAc,MAAM,CAAA,CAAE,CAAA;AACtD,IAAA,KAAA,CAAM,KAAK,EAAE,CAAA;AACb,IAAA,KAAA,CAAM,KAAK,kBAAkB,CAAA;AAC7B,IAAA,KAAA,CAAM,KAAK,kBAAkB,CAAA;AAE7B,IAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,aAAA,CAAc,QAAQ,CAAA,EAAA,EAAK;AAC7C,MAAA,MAAM,KAAA,GAAQ,cAAc,CAAC,CAAA;AAC7B,MAAA,MAAM,cAAc,CAAA,GAAI,CAAA;AAExB,MAAA,IAAI,KAAA,CAAM,WAAW,CAAA,EAAG;AAEtB,QAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,CAAC,CAAA,CAAE,KAAA;AACvB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAG,WAAW,CAAA,EAAA,EAAK,MAAM,IAAI,CAAA,EAAA,EAAK,KAAA,CAAM,EAAE,CAAA,CAAA,CAAG,CAAA;AACxD,QAAA,IAAI,MAAM,WAAA,EAAa;AACrB,UAAA,KAAA,CAAM,IAAA,CAAK,CAAA,GAAA,EAAM,KAAA,CAAM,WAAW,CAAA,CAAE,CAAA;AAAA,QACtC;AAAA,MACF,CAAA,MAAO;AAEL,QAAA,KAAA,CAAM,IAAA,CAAK,CAAA,EAAG,WAAW,CAAA,YAAA,CAAc,CAAA;AACvC,QAAA,KAAA,MAAW,QAAQ,KAAA,EAAO;AACxB,UAAA,MAAM,QAAQ,IAAA,CAAK,KAAA;AACnB,UAAA,KAAA,CAAM,KAAK,CAAA,KAAA,EAAQ,KAAA,CAAM,IAAI,CAAA,EAAA,EAAK,KAAA,CAAM,EAAE,CAAA,CAAA,CAAG,CAAA;AAC7C,UAAA,IAAI,MAAM,WAAA,EAAa;AACrB,YAAA,KAAA,CAAM,IAAA,CAAK,CAAA,KAAA,EAAQ,KAAA,CAAM,WAAW,CAAA,CAAE,CAAA;AAAA,UACxC;AAAA,QACF;AAAA,MACF;AACA,MAAA,KAAA,CAAM,KAAK,EAAE,CAAA;AAAA,IACf;AAEA,IAAA,OAAO,KAAA,CAAM,KAAK,IAAI,CAAA;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,WAAA,GAAwB;AACtB,IAAA,OAAO,KAAK,MAAA,CAAO,GAAA,CAAI,CAAC,IAAA,KAAS,IAAA,CAAK,MAAM,EAAE,CAAA;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,SAAS,OAAA,EAA0B;AACjC,IAAA,OAAO,IAAA,CAAK,OAAO,IAAA,CAAK,CAAC,SAAS,IAAA,CAAK,KAAA,CAAM,OAAO,OAAO,CAAA;AAAA,EAC7D;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,eAAe,MAAA,EAGb;AACA,IAAA,MAAM,SAAoD,EAAC;AAE3D,IAAA,KAAA,MAAW,IAAA,IAAQ,KAAK,MAAA,EAAQ;AAC9B,MAAA,MAAM,QAAQ,IAAA,CAAK,KAAA;AACnB,MAAA,MAAM,WAAA,GAAc,MAAA,CAAO,KAAA,CAAM,EAAE,KAAK,EAAC;AAEzC,MAAA,IAAI;AACF,QAAA,KAAA,CAAM,YAAA,CAAa,MAAM,WAAW,CAAA;AAAA,MACtC,SAAS,KAAA,EAAO;AACd,QAAA,IAAI,KAAA,YAAiB,EAAE,QAAA,EAAU;AAC/B,UAAA,MAAM,gBAAgB,KAAA,CAAM,MAAA,CACzB,IAAI,CAAC,CAAA,KAAkB,GAAG,CAAA,CAAE,IAAA,CAAK,IAAA,CAAK,GAAG,CAAC,CAAA,EAAA,EAAK,CAAA,CAAE,OAAO,CAAA,CAAE,CAAA,CAC1D,KAAK,IAAI,CAAA;AACZ,UAAA,MAAA,CAAO,IAAA,CAAK;AAAA,YACV,SAAS,KAAA,CAAM,EAAA;AAAA,YACf,KAAA,EAAO,6BAA6B,aAAa,CAAA;AAAA,WAClD,CAAA;AAAA,QACH,CAAA,MAAO;AACL,UAAA,MAAA,CAAO,IAAA,CAAK;AAAA,YACV,SAAS,KAAA,CAAM,EAAA;AAAA,YACf,KAAA,EAAO,OAAO,KAAK;AAAA,WACpB,CAAA;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAEA,IAAA,OAAO;AAAA,MACL,KAAA,EAAO,OAAO,MAAA,KAAW,CAAA;AAAA,MACzB;AAAA,KACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,YAAA,CACE,OACA,MAAA,EACQ;AACR,IAAA,IAAI,SAAA,GAAY,CAAA;AAChB,IAAA,MAAM,YAAA,GAAe,KAAA;AAErB,IAAA,KAAA,MAAW,IAAA,IAAQ,KAAK,MAAA,EAAQ;AAC9B,MAAA,IAAI,IAAA,CAAK,MAAM,YAAA,EAAc;AAC3B,QAAA,MAAM,cAAc,MAAA,CAAO,IAAA,CAAK,KAAA,CAAM,EAAE,KAAK,EAAC;AAC9C,QAAA,SAAA,IAAa,IAAA,CAAK,KAAA,CAAM,YAAA,CAAa,YAAA,EAAc,WAAW,CAAA;AAAA,MAChE;AAAA,IAGF;AAEA,IAAA,OAAO,SAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,eAAA,GAQE;AACA,IAAA,MAAM,UAQF,EAAC;AAEL,IAAA,KAAA,MAAW,IAAA,IAAQ,KAAK,MAAA,EAAQ;AAC9B,MAAA,MAAM,QAAQ,IAAA,CAAK,KAAA;AAGnB,MAAA,MAAM,WAAoC,EAAC;AAE3C,MAAA,IAAI,KAAA,CAAM,YAAA,YAAwB,CAAA,CAAE,SAAA,EAAW;AAC7C,QAAA,MAAM,KAAA,GAAQ,MAAM,YAAA,CAAa,KAAA;AACjC,QAAA,KAAA,MAAW,CAAC,GAAA,EAAK,WAAW,KAAK,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,EAAG;AACtD,UAAA,IAAI,SAAA,GAAY,WAAA;AAGhB,UAAA,IAAI,SAAA,YAAqB,EAAE,WAAA,EAAa;AACtC,YAAA,SAAA,GAAa,UAAU,IAAA,CAAa,SAAA;AAAA,UACtC;AAGA,UAAA,IAAI,SAAA,YAAqB,EAAE,UAAA,EAAY;AACrC,YAAA,MAAM,cAAA,GAAkB,UAAU,IAAA,CAAa,YAAA;AAC/C,YAAA,QAAA,CAAS,GAAG,CAAA,GACV,OAAO,cAAA,KAAmB,UAAA,GACtB,gBAAe,GACf,cAAA;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAEA,MAAA,OAAA,CAAQ,KAAA,CAAM,EAAE,CAAA,GAAI;AAAA,QAClB,QAAQ,KAAA,CAAM,YAAA;AAAA,QACd,QAAA;AAAA,QACA,MAAM,KAAA,CAAM,IAAA;AAAA,QACZ,aAAa,KAAA,CAAM;AAAA,OACrB;AAAA,IACF;AAEA,IAAA,OAAO,OAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAA,GAA4D;AAC1D,IAAA,MAAM,YAAA,GAAe,KAAK,eAAA,EAAgB;AAC1C,IAAA,MAAM,SAAkD,EAAC;AAEzD,IAAA,KAAA,MAAW,CAAC,OAAA,EAAS,IAAI,KAAK,MAAA,CAAO,OAAA,CAAQ,YAAY,CAAA,EAAG;AAC1D,MAAA,MAAA,CAAO,OAAO,IAAI,IAAA,CAAK,QAAA;AAAA,IACzB;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAIA,0BAA0B,UAAA,EAA4C;AACpE,IAAA,OAAO,IAAA,CAAK,MAAA,CACT,MAAA,CAAO,CAAC,IAAA,KAAS,IAAA,CAAK,cAAA,KAAmB,UAAU,CAAA,CACnD,GAAA,CAAI,CAAC,IAAA,KAAS,KAAK,KAAK,CAAA;AAAA,EAC7B;AAAA;AAAA;AAAA;AAAA,EAKA,cAAc,OAAA,EAAyB;AACrC,IAAA,OAAO,IAAA,CAAK,OAAO,SAAA,CAAU,CAAC,SAAS,IAAA,CAAK,KAAA,CAAM,OAAO,OAAO,CAAA;AAAA,EAClE;AAAA;AAAA;AAAA;AAAA,EAKA,uBAAuB,OAAA,EAAyB;AAC9C,IAAA,MAAM,IAAA,GAAO,KAAK,MAAA,CAAO,IAAA,CAAK,CAACA,KAAAA,KAASA,KAAAA,CAAK,KAAA,CAAM,EAAA,KAAO,OAAO,CAAA;AACjE,IAAA,IAAI,CAAC,IAAA,EAAM,MAAM,IAAI,KAAA,CAAM,CAAA,MAAA,EAAS,OAAO,CAAA,sBAAA,CAAwB,CAAA;AACnE,IAAA,OAAO,IAAA,CAAK,cAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,mBAAmB,OAAA,EAAqC;AACtD,IAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,aAAA,CAAc,OAAO,CAAA;AACxC,IAAA,IAAI,KAAA,IAAS,GAAG,OAAO,MAAA;AACvB,IAAA,OAAO,IAAA,CAAK,MAAA,CAAO,KAAA,GAAQ,CAAC,EAAE,KAAA,CAAM,EAAA;AAAA,EACtC;AACF;AAMO,IAAM,kBAAN,MAIL;AAAA,EAIA,WAAA,CACU,EAAA,EACA,IAAA,EACA,WAAA,EACA,aACA,mBAAA,EACR;AALQ,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA;AACA,IAAA,IAAA,CAAA,IAAA,GAAA,IAAA;AACA,IAAA,IAAA,CAAA,WAAA,GAAA,WAAA;AACA,IAAA,IAAA,CAAA,WAAA,GAAA,WAAA;AACA,IAAA,IAAA,CAAA,mBAAA,GAAA,mBAAA;AAAA,EACP;AAAA,EATK,SAAsB,EAAC;AAAA,EACvB,qBAAA,GAAwB,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAuBhC,KAME,KAAA,EAKA;AAEA,IAAA,IAAI,MAAM,YAAA,EAAc;AACtB,MAAA,MAAM,gBAAA,GAAmB,KAAK,MAAA,CAAO,GAAA,CAAI,CAAC,CAAA,KAAM,CAAA,CAAE,MAAM,EAAE,CAAA;AAC1D,MAAA,MAAM,WAAA,GAAc,MAAM,YAAA,CAAa,MAAA;AAAA,QACrC,CAAC,GAAA,KAAQ,CAAC,gBAAA,CAAiB,SAAS,GAAG;AAAA,OACzC;AAEA,MAAA,IAAI,WAAA,CAAY,SAAS,CAAA,EAAG;AAC1B,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,UAAU,KAAA,CAAM,EAAE,+BAA+B,WAAA,CAAY,IAAA,CAAK,IAAI,CAAC,CAAA,qDAAA,EACf,MAAM,EAAE,CAAA,mBAAA,EAE5D,iBAAiB,MAAA,KAAW,CAAA,GACxB,WACA,gBAAA,CAAiB,IAAA,CAAK,IAAI,CAChC,CAAA;AAAA,SACJ;AAAA,MACF;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,qBAAA,EAAA;AAEL,IAAA,IAAA,CAAK,OAAO,IAAA,CAAK;AAAA,MACf,KAAA;AAAA,MACA,gBAAgB,IAAA,CAAK;AAAA,KACtB,CAAA;AAGD,IAAA,MAAM,OAAA,GAAU,IAAA;AAKhB,IAAC,OAAA,CAAgB,sBAAsB,KAAA,CAAM,YAAA;AAE7C,IAAA,OAAO,OAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAqBA,SAOE,MAAA,EAaA;AAEA,IAAA,MAAM,gBAAA,GAAmB,KAAK,MAAA,CAAO,GAAA,CAAI,CAAC,CAAA,KAAM,CAAA,CAAE,MAAM,EAAE,CAAA;AAE1D,IAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,MAAA,IAAI,MAAM,YAAA,EAAc;AACtB,QAAA,MAAM,WAAA,GAAc,MAAM,YAAA,CAAa,MAAA;AAAA,UACrC,CAAC,GAAA,KAAQ,CAAC,gBAAA,CAAiB,SAAS,GAAG;AAAA,SACzC;AAEA,QAAA,IAAI,WAAA,CAAY,SAAS,CAAA,EAAG;AAC1B,UAAA,MAAM,IAAI,KAAA;AAAA,YACR,CAAA,OAAA,EAAU,KAAA,CAAM,EAAE,CAAA,gDAAA,EAAmD,WAAA,CAAY,IAAA;AAAA,cAC/E;AAAA,aACD,4FAGG,gBAAA,CAAiB,MAAA,KAAW,IACxB,QAAA,GACA,gBAAA,CAAiB,IAAA,CAAK,IAAI,CAChC,CAAA;AAAA,WACJ;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,IAAA,IAAA,CAAK,qBAAA,EAAA;AAGL,IAAA,KAAA,MAAW,SAAS,MAAA,EAAQ;AAC1B,MAAA,IAAA,CAAK,OAAO,IAAA,CAAK;AAAA,QACf,KAAA;AAAA,QACA,gBAAgB,IAAA,CAAK;AAAA,OACtB,CAAA;AAAA,IACH;AAGA,IAAA,MAAM,eAAe,CAAA,CAAE,MAAA;AAAA,MACrB,MAAA,CAAO,MAAA;AAAA,QACL,CAAC,GAAA,EAAK,KAAA,EAAO,KAAA,KAAU;AACrB,UAAA,GAAA,CAAI,KAAK,IAAI,KAAA,CAAM,YAAA;AACnB,UAAA,OAAO,GAAA;AAAA,QACT,CAAA;AAAA,QACA;AAAC;AACH,KACF;AAEA,IAAA,MAAM,OAAA,GAAU,IAAA;AAahB,IAAA,OAAA,CAAQ,mBAAA,GAAsB,YAAA;AAE9B,IAAA,OAAO,OAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,KAAA,GAAoD;AAClD,IAAA,OAAO,IAAI,QAAA;AAAA,MACT,IAAA,CAAK,EAAA;AAAA,MACL,IAAA,CAAK,IAAA;AAAA,MACL,IAAA,CAAK,WAAA;AAAA,MACL,IAAA,CAAK,WAAA;AAAA,MACL,IAAA,CAAK,mBAAA;AAAA,MACL,IAAA,CAAK;AAAA,KACP;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,aAAA,GAAwB;AACtB,IAAA,OAAO,KAAK,MAAA,CAAO,MAAA;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA,EAKA,sBAAA,GAAiC;AAC/B,IAAA,OAAO,IAAA,CAAK,qBAAA;AAAA,EACd;AACF;AC/eO,IAAM,cAAA,GAAiBC,EAAE,MAAA,CAAO;AAAA;AAAA,EAErC,KAAA,EAAO,QAAA,CAAS,OAAA,CAAQ,kBAAkB,CAAA;AAAA;AAAA,EAE1C,WAAA,EAAaA,CAAAA,CAAE,MAAA,EAAO,CAAE,GAAA,CAAI,CAAC,CAAA,CAAE,GAAA,CAAI,CAAC,CAAA,CAAE,OAAA,CAAQ,GAAG,CAAA;AAAA;AAAA,EAEjD,WAAWA,CAAAA,CAAE,MAAA,EAAO,CAAE,QAAA,GAAW,QAAA;AACnC,CAAC;AAOM,IAAM,uBAAA,GAA0BA,EAAE,MAAA,CAAO;AAAA;AAAA,EAE9C,aAAaA,CAAAA,CAAE,MAAA,GAAS,QAAA,EAAS,CAAE,QAAQ,CAAC,CAAA;AAAA;AAAA,EAE5C,SAASA,CAAAA,CAAE,MAAA,GAAS,WAAA,EAAY,CAAE,QAAQ,CAAC,CAAA;AAAA;AAAA,EAE3C,YAAYA,CAAAA,CAAE,MAAA,GAAS,WAAA,EAAY,CAAE,QAAQ,CAAC;AAChD,CAAC;AAOM,IAAM,wBAAA,GAA2BA,EAAE,MAAA,CAAO;AAAA;AAAA,EAE/C,YAAA,EAAcA,CAAAA,CAAE,MAAA,CAAOA,CAAAA,CAAE,MAAA,EAAO,EAAGA,CAAAA,CAAE,OAAA,EAAS,CAAA,CAAE,OAAA,CAAQ,EAAE;AAC5D,CAAC;AAOM,IAAM,iBAAA,GAAoBA,EAAE,MAAA,CAAO;AAAA;AAAA,EAExC,OAAA,EAASA,CAAAA,CAAE,OAAA,EAAQ,CAAE,QAAQ,KAAK,CAAA;AAAA;AAAA,EAElC,MAAA,EAAQA,CAAAA,CAAE,OAAA,EAAQ,CAAE,QAAQ,KAAK;AACnC,CAAC;AAmBM,SAAS,aAAsC,MAAA,EAAwB;AAC5E,EAAA,OAAO,MAAA,CAAO,MAAM,cAAc,CAAA;AACpC;AAaO,SAAS,gBACd,MAAA,EACA;AACA,EAAA,OAAO,MAAA,CAAO,MAAM,uBAAuB,CAAA;AAC7C;AAaO,SAAS,iBACd,MAAA,EACA;AACA,EAAA,OAAO,MAAA,CAAO,MAAM,wBAAwB,CAAA;AAC9C;AAiCO,SAAS,mBACd,MAAA,EACA;AACA,EAAA,OAAO,MAAA,CACJ,MAAM,cAAc,CAAA,CACpB,MAAM,uBAAuB,CAAA,CAC7B,MAAM,wBAAwB,CAAA;AACnC;AAiDiCA,CAAAA,CAAE,MAAA,CAAO,EAAE;AAKPA,EAAE,MAAA,CAAO;AAAA,EAC5C,KAAA,EAAO,QAAA,CAAS,OAAA,CAAQ,kBAAkB,CAAA;AAAA,EAC1C,WAAA,EAAaA,CAAAA,CAAE,MAAA,EAAO,CAAE,GAAA,CAAI,CAAC,CAAA,CAAE,GAAA,CAAI,CAAC,CAAA,CAAE,OAAA,CAAQ,GAAG;AACnD,CAAC","file":"index.js","sourcesContent":["/**\n * Workflow Builder - Fluent API for composing type-safe workflows\n *\n * Workflows are composed of stages that are executed sequentially or in parallel.\n * The builder ensures type safety: output of one stage matches input of next stage.\n *\n * ## Type System Features\n *\n * ### Automatic Context Inference\n * The workflow context type is automatically accumulated as you pipe stages.\n * Use `InferWorkflowContext<typeof workflow>` to extract the context type.\n *\n * ```typescript\n * const workflow = new WorkflowBuilder(...)\n *   .pipe(stage1)\n *   .pipe(stage2)\n *   .build();\n *\n * // Auto-generated type\n * type MyContext = InferWorkflowContext<typeof workflow>;\n * // = { \"stage-1\": Stage1Output, \"stage-2\": Stage2Output }\n * ```\n *\n * ### Stage ID Constants\n * Use `workflow.stageIds` for type-safe stage ID references.\n */\n\nimport { z } from \"zod\";\nimport type { Stage } from \"./stage\";\n\n// ============================================================================\n// Stage Node - Represents a stage in the execution plan\n// ============================================================================\n\nexport interface StageNode {\n  stage: Stage<any, any, any>;\n  executionGroup: number; // For parallel execution grouping\n}\n\n// ============================================================================\n// Workflow - Complete workflow definition\n// ============================================================================\n\nexport class Workflow<\n  TInput extends z.ZodTypeAny,\n  TOutput extends z.ZodTypeAny,\n  TContext extends Record<string, unknown> = {},\n> {\n  constructor(\n    public readonly id: string,\n    public readonly name: string,\n    public readonly description: string,\n    public readonly inputSchema: TInput,\n    public readonly outputSchema: TOutput,\n    private readonly stages: StageNode[],\n    public readonly contextType?: TContext, // Type-only, for inference\n  ) {}\n\n  /**\n   * Get execution plan as groups of stages\n   * Stages in the same group can be executed in parallel\n   */\n  getExecutionPlan(): StageNode[][] {\n    const groups = new Map<number, StageNode[]>();\n\n    for (const node of this.stages) {\n      const group = groups.get(node.executionGroup) || [];\n      group.push(node);\n      groups.set(node.executionGroup, group);\n    }\n\n    // Return groups in order\n    const sortedGroups = Array.from(groups.keys()).sort((a, b) => a - b);\n    return sortedGroups.map((groupNum) => {\n      const group = groups.get(groupNum);\n      if (!group) throw new Error(`Group ${groupNum} not found`);\n      return group;\n    });\n  }\n\n  /**\n   * Get a specific stage by ID\n   */\n  getStage(stageId: string): Stage<any, any, any> | undefined {\n    const node = this.stages.find((n) => n.stage.id === stageId);\n    return node?.stage;\n  }\n\n  /**\n   * Get all stages in order\n   */\n  getAllStages(): StageNode[] {\n    return [...this.stages];\n  }\n\n  /**\n   * Get a visual representation of the workflow execution order\n   */\n  getExecutionOrder(): string {\n    const executionPlan = this.getExecutionPlan();\n    const lines: string[] = [];\n\n    lines.push(`Workflow: ${this.name} (${this.id})`);\n    lines.push(`Total stages: ${this.stages.length}`);\n    lines.push(`Execution groups: ${executionPlan.length}`);\n    lines.push(\"\");\n    lines.push(\"Execution Order:\");\n    lines.push(\"================\");\n\n    for (let i = 0; i < executionPlan.length; i++) {\n      const group = executionPlan[i];\n      const groupNumber = i + 1;\n\n      if (group.length === 1) {\n        // Sequential stage\n        const stage = group[0].stage;\n        lines.push(`${groupNumber}. ${stage.name} (${stage.id})`);\n        if (stage.description) {\n          lines.push(`   ${stage.description}`);\n        }\n      } else {\n        // Parallel stages\n        lines.push(`${groupNumber}. [PARALLEL]`);\n        for (const node of group) {\n          const stage = node.stage;\n          lines.push(`   - ${stage.name} (${stage.id})`);\n          if (stage.description) {\n            lines.push(`     ${stage.description}`);\n          }\n        }\n      }\n      lines.push(\"\");\n    }\n\n    return lines.join(\"\\n\");\n  }\n\n  /**\n   * Get all stage IDs in execution order\n   *\n   * @returns Array of stage IDs\n   *\n   * @example\n   * ```typescript\n   * const ids = workflow.getStageIds();\n   * // [\"data-extraction\", \"guidelines\", \"generator\"]\n   * ```\n   */\n  getStageIds(): string[] {\n    return this.stages.map((node) => node.stage.id);\n  }\n\n  /**\n   * Check if a stage ID exists in this workflow\n   *\n   * @param stageId - The stage ID to check\n   * @returns true if the stage exists\n   */\n  hasStage(stageId: string): boolean {\n    return this.stages.some((node) => node.stage.id === stageId);\n  }\n\n  /**\n   * Validate workflow configuration before execution\n   * Checks that all stage configs match their schemas\n   *\n   * @param config - Configuration object with keys matching stage IDs\n   * @returns Validation result with any errors\n   */\n  validateConfig(config: Record<string, unknown>): {\n    valid: boolean;\n    errors: Array<{ stageId: string; error: string }>;\n  } {\n    const errors: Array<{ stageId: string; error: string }> = [];\n\n    for (const node of this.stages) {\n      const stage = node.stage;\n      const stageConfig = config[stage.id] || {};\n\n      try {\n        stage.configSchema.parse(stageConfig);\n      } catch (error) {\n        if (error instanceof z.ZodError) {\n          const errorMessages = error.issues\n            .map((e: z.ZodIssue) => `${e.path.join(\".\")}: ${e.message}`)\n            .join(\"; \");\n          errors.push({\n            stageId: stage.id,\n            error: `Config validation failed: ${errorMessages}`,\n          });\n        } else {\n          errors.push({\n            stageId: stage.id,\n            error: String(error),\n          });\n        }\n      }\n    }\n\n    return {\n      valid: errors.length === 0,\n      errors,\n    };\n  }\n\n  /**\n   * Estimate total cost for the workflow\n   */\n  estimateCost(\n    input: z.infer<TInput>,\n    config: Record<string, unknown>,\n  ): number {\n    let totalCost = 0;\n    const currentInput = input;\n\n    for (const node of this.stages) {\n      if (node.stage.estimateCost) {\n        const stageConfig = config[node.stage.id] || {};\n        totalCost += node.stage.estimateCost(currentInput, stageConfig);\n      }\n      // Note: We can't accurately propagate input for estimation without execution\n      // This is a rough estimate only\n    }\n\n    return totalCost;\n  }\n\n  /**\n   * Get configuration schemas for all stages in this workflow\n   * Returns a map of stageId → { schema, defaults, name, description }\n   */\n  getStageConfigs(): Record<\n    string,\n    {\n      schema: z.ZodTypeAny;\n      defaults: Record<string, unknown>;\n      name: string;\n      description?: string;\n    }\n  > {\n    const configs: Record<\n      string,\n      {\n        schema: z.ZodTypeAny;\n        defaults: Record<string, unknown>;\n        name: string;\n        description?: string;\n      }\n    > = {};\n\n    for (const node of this.stages) {\n      const stage = node.stage;\n\n      // Extract defaults from schema\n      const defaults: Record<string, unknown> = {};\n\n      if (stage.configSchema instanceof z.ZodObject) {\n        const shape = stage.configSchema.shape as Record<string, z.ZodTypeAny>;\n        for (const [key, fieldSchema] of Object.entries(shape)) {\n          let unwrapped = fieldSchema;\n\n          // Unwrap ZodOptional if present\n          if (unwrapped instanceof z.ZodOptional) {\n            unwrapped = (unwrapped._def as any).innerType;\n          }\n\n          // Check for ZodDefault\n          if (unwrapped instanceof z.ZodDefault) {\n            const defaultValueFn = (unwrapped._def as any).defaultValue;\n            defaults[key] =\n              typeof defaultValueFn === \"function\"\n                ? defaultValueFn()\n                : defaultValueFn;\n          }\n        }\n      }\n\n      configs[stage.id] = {\n        schema: stage.configSchema,\n        defaults,\n        name: stage.name,\n        description: stage.description,\n      };\n    }\n\n    return configs;\n  }\n\n  /**\n   * Generate default configuration object for all stages\n   * Automatically discovers all stage configs - add/remove stages and this updates automatically\n   */\n  getDefaultConfig(): Record<string, Record<string, unknown>> {\n    const stageConfigs = this.getStageConfigs();\n    const config: Record<string, Record<string, unknown>> = {};\n\n    for (const [stageId, meta] of Object.entries(stageConfigs)) {\n      config[stageId] = meta.defaults;\n    }\n\n    return config;\n  }\n  /**\n   * Get all stages in a specific execution group\n   */\n  getStagesInExecutionGroup(groupIndex: number): Stage<any, any, any>[] {\n    return this.stages\n      .filter((node) => node.executionGroup === groupIndex)\n      .map((node) => node.stage);\n  }\n\n  /**\n   * Get the sequential index of a stage (0-based)\n   */\n  getStageIndex(stageId: string): number {\n    return this.stages.findIndex((node) => node.stage.id === stageId);\n  }\n\n  /**\n   * Get the execution group index for a stage\n   */\n  getExecutionGroupIndex(stageId: string): number {\n    const node = this.stages.find((node) => node.stage.id === stageId);\n    if (!node) throw new Error(`Stage ${stageId} not found in workflow`);\n    return node.executionGroup;\n  }\n\n  /**\n   * Get the ID of the stage immediately preceding the given stage\n   */\n  getPreviousStageId(stageId: string): string | undefined {\n    const index = this.getStageIndex(stageId);\n    if (index <= 0) return undefined;\n    return this.stages[index - 1].stage.id;\n  }\n}\n\n// ============================================================================\n// Workflow Builder - Fluent API with Context Accumulation\n// ============================================================================\n\nexport class WorkflowBuilder<\n  TInput extends z.ZodTypeAny,\n  TCurrentOutput extends z.ZodTypeAny,\n  TContext extends Record<string, unknown> = {},\n> {\n  private stages: StageNode[] = [];\n  private currentExecutionGroup = 0;\n\n  constructor(\n    private id: string,\n    private name: string,\n    private description: string,\n    private inputSchema: TInput,\n    private currentOutputSchema: TCurrentOutput,\n  ) {}\n\n  /**\n   * Add a stage to the workflow (sequential execution)\n   *\n   * Automatically accumulates the stage's output in the context under its stage ID.\n   * This provides type-safe access to all previous stage outputs.\n   *\n   * Note: This accepts any stage regardless of strict input type matching.\n   * This is necessary because stages using passthrough() can accept objects\n   * with additional fields beyond what's declared in their input schema.\n   * Runtime validation via Zod ensures type safety at execution time.\n   *\n   * Validates that all declared dependencies exist in the workflow.\n   */\n  pipe<\n    TStageInput extends z.ZodTypeAny,\n    TStageOutput extends z.ZodTypeAny,\n    TStageConfig extends z.ZodTypeAny,\n    TStageContext extends Record<string, unknown>,\n  >(\n    stage: Stage<TStageInput, TStageOutput, TStageConfig, TStageContext>,\n  ): WorkflowBuilder<\n    TInput,\n    TStageOutput,\n    TContext & { [x: string]: z.infer<TStageOutput> }\n  > {\n    // Validate stage dependencies\n    if (stage.dependencies) {\n      const existingStageIds = this.stages.map((s) => s.stage.id);\n      const missingDeps = stage.dependencies.filter(\n        (dep) => !existingStageIds.includes(dep),\n      );\n\n      if (missingDeps.length > 0) {\n        throw new Error(\n          `Stage \"${stage.id}\" has missing dependencies: ${missingDeps.join(\", \")}. ` +\n            `These stages must be added to the workflow before \"${stage.id}\". ` +\n            `Current stages: ${\n              existingStageIds.length === 0\n                ? \"(none)\"\n                : existingStageIds.join(\", \")\n            }`,\n        );\n      }\n    }\n\n    this.currentExecutionGroup++;\n\n    this.stages.push({\n      stage: stage as Stage<any, any, any, any>,\n      executionGroup: this.currentExecutionGroup,\n    });\n\n    // Return builder with new output type and accumulated context\n    const builder = this as unknown as WorkflowBuilder<\n      TInput,\n      TStageOutput,\n      TContext & { [x: string]: z.infer<TStageOutput> }\n    >;\n    (builder as any).currentOutputSchema = stage.outputSchema;\n\n    return builder;\n  }\n\n  /**\n   * Add a stage with strict input type checking\n   *\n   * Note: pipeStrict() and pipeLoose() have been removed as they were\n   * just aliases for pipe(). Use pipe() for all stage chaining.\n   */\n\n  /**\n   * Add multiple stages that execute in parallel\n   *\n   * All stages receive the same input (current output)\n   * Their outputs are merged into an object by index AND accumulated in context by stage ID.\n   *\n   * Note: This accepts stages regardless of strict input type matching.\n   * This is necessary because stages using passthrough() can accept objects\n   * with additional fields. Runtime validation via Zod ensures type safety.\n   *\n   * Validates that all declared dependencies exist in the workflow.\n   */\n  parallel<\n    TStages extends {\n      id: string;\n      outputSchema: z.ZodTypeAny;\n      dependencies?: string[];\n    }[],\n  >(\n    stages: [...TStages],\n  ): WorkflowBuilder<\n    TInput,\n    z.ZodTypeAny,\n    TContext & {\n      [K in TStages[number][\"id\"]]: TStages[number] extends {\n        outputSchema: infer O;\n      }\n        ? O extends z.ZodTypeAny\n          ? z.infer<O>\n          : never\n        : never;\n    }\n  > {\n    // Validate dependencies for all parallel stages\n    const existingStageIds = this.stages.map((s) => s.stage.id);\n\n    for (const stage of stages) {\n      if (stage.dependencies) {\n        const missingDeps = stage.dependencies.filter(\n          (dep) => !existingStageIds.includes(dep),\n        );\n\n        if (missingDeps.length > 0) {\n          throw new Error(\n            `Stage \"${stage.id}\" (in parallel group) has missing dependencies: ${missingDeps.join(\n              \", \",\n            )}. ` +\n              `These stages must be added to the workflow before this parallel group. ` +\n              `Current stages: ${\n                existingStageIds.length === 0\n                  ? \"(none)\"\n                  : existingStageIds.join(\", \")\n              }`,\n          );\n        }\n      }\n    }\n\n    this.currentExecutionGroup++;\n\n    // Add all stages to same execution group\n    for (const stage of stages) {\n      this.stages.push({\n        stage: stage as Stage<any, any, any, any>,\n        executionGroup: this.currentExecutionGroup,\n      });\n    }\n\n    // Create merged output schema\n    const mergedSchema = z.object(\n      stages.reduce(\n        (acc, stage, index) => {\n          acc[index] = stage.outputSchema;\n          return acc;\n        },\n        {} as Record<number, z.ZodTypeAny>,\n      ),\n    ) as any;\n\n    const builder = this as unknown as WorkflowBuilder<\n      TInput,\n      any,\n      TContext & {\n        [K in TStages[number][\"id\"]]: TStages[number] extends Stage<\n          any,\n          infer O,\n          any\n        >\n          ? z.infer<O>\n          : never;\n      }\n    >;\n    builder.currentOutputSchema = mergedSchema;\n\n    return builder as any;\n  }\n\n  /**\n   * Build the final workflow\n   */\n  build(): Workflow<TInput, TCurrentOutput, TContext> {\n    return new Workflow(\n      this.id,\n      this.name,\n      this.description,\n      this.inputSchema,\n      this.currentOutputSchema,\n      this.stages,\n    );\n  }\n\n  /**\n   * Get current stage count\n   */\n  getStageCount(): number {\n    return this.stages.length;\n  }\n\n  /**\n   * Get execution group count\n   */\n  getExecutionGroupCount(): number {\n    return this.currentExecutionGroup;\n  }\n}\n\n// ============================================================================\n// Type Inference Utilities (Supplementary to Code Generator)\n// ============================================================================\n\n/**\n * NOTE: For most use cases, prefer using the generated types from `__generated__.ts`\n * which are created by running `pnpm generate:workflow-types`.\n *\n * These inference utilities are useful for:\n * - Quick prototyping before running the generator\n * - Dynamic workflows not covered by the generator\n * - Type assertions in tests\n */\n\n/**\n * Extract the workflow context type from a Workflow instance\n *\n * The workflow context type is automatically accumulated as stages are piped.\n * Each stage's output is added to the context under its stage ID.\n *\n * @example\n * ```typescript\n * const workflow = new WorkflowBuilder(...)\n *   .pipe(dataExtractionStage)  // id: \"data-extraction\"\n *   .pipe(guidelinesStage)       // id: \"guidelines\"\n *   .build();\n *\n * // Extract context type automatically\n * type MyWorkflowContext = InferWorkflowContext<typeof workflow>;\n * // = {\n * //   \"data-extraction\": DataExtractionOutput;\n * //   \"guidelines\": GuidelinesOutput;\n * // }\n *\n * // Use in stage definitions\n * export const myStage = defineStage<\n *   \"none\",\n *   typeof OutputSchema,\n *   typeof ConfigSchema,\n *   MyWorkflowContext\n * >({ ... });\n * ```\n */\nexport type InferWorkflowContext<W> = W extends Workflow<any, any, infer C>\n  ? C\n  : never;\n\n/**\n * Extract the input type from a Workflow instance\n *\n * @example\n * ```typescript\n * type Input = InferWorkflowInput<typeof myWorkflow>;\n * ```\n */\nexport type InferWorkflowInput<W> = W extends Workflow<infer I, any, any>\n  ? z.infer<I>\n  : never;\n\n/**\n * Extract the output type from a Workflow instance\n *\n * @example\n * ```typescript\n * type Output = InferWorkflowOutput<typeof myWorkflow>;\n * ```\n */\nexport type InferWorkflowOutput<W> = W extends Workflow<any, infer O, any>\n  ? z.infer<O>\n  : never;\n\n/**\n * Extract stage IDs as a union type from a Workflow instance\n *\n * Useful for creating type-safe stage ID references.\n *\n * @example\n * ```typescript\n * type StageId = InferWorkflowStageIds<typeof myWorkflow>;\n * // = \"data-extraction\" | \"guidelines\" | \"generator\"\n *\n * function getStageOutput(stageId: StageId) { ... }\n * ```\n */\nexport type InferWorkflowStageIds<W> = W extends Workflow<any, any, infer C>\n  ? keyof C & string\n  : never;\n\n/**\n * Get the output type for a specific stage ID from a Workflow\n *\n * @example\n * ```typescript\n * type DataOutput = InferStageOutputById<typeof workflow, \"data-extraction\">;\n * ```\n */\nexport type InferStageOutputById<W, K extends string> = W extends Workflow<\n  any,\n  any,\n  infer C\n>\n  ? K extends keyof C\n    ? C[K]\n    : never\n  : never;\n","/**\n * Config Presets - Common configuration patterns for workflow stages\n *\n * These presets provide standardized schemas for common stage configurations,\n * reducing boilerplate and ensuring consistency across stages.\n *\n * @example\n * ```typescript\n * import { defineStage } from \"./stage-factory\";\n * import { withAIConfig, withStandardConfig } from \"./config-presets\";\n *\n * // Use AI-focused config preset\n * const myStage = defineStage({\n *   id: \"my-stage\",\n *   name: \"My Stage\",\n *   schemas: {\n *     input: z.object({ data: z.string() }),\n *     output: z.object({ result: z.string() }),\n *     config: withAIConfig(z.object({\n *       customField: z.string(),\n *     })),\n *   },\n *   async execute(ctx) {\n *     // ctx.config.model, ctx.config.temperature available\n *     // plus ctx.config.customField\n *   },\n * });\n *\n * // Use standard config preset (includes AI + concurrency + feature flags)\n * const fullStage = defineStage({\n *   id: \"full-stage\",\n *   name: \"Full Stage\",\n *   schemas: {\n *     input: \"none\",\n *     output: z.object({ result: z.string() }),\n *     config: withStandardConfig(z.object({\n *       specificOption: z.boolean().default(true),\n *     })),\n *   },\n *   async execute(ctx) {\n *     // All standard config + custom fields available\n *   },\n * });\n * ```\n */\n\nimport { z } from \"zod\";\nimport { ModelKey } from \"../ai/model-helper\";\n\n// ============================================================================\n// Base Config Schemas\n// ============================================================================\n\n/**\n * AI/LLM configuration options\n */\nexport const AIConfigSchema = z.object({\n  /** The model to use for AI operations */\n  model: ModelKey.default(\"gemini-2.5-flash\"),\n  /** Temperature for AI generations (0-2) */\n  temperature: z.number().min(0).max(2).default(0.7),\n  /** Maximum tokens to generate (undefined = model default) */\n  maxTokens: z.number().positive().optional(),\n});\n\nexport type AIConfig = z.infer<typeof AIConfigSchema>;\n\n/**\n * Concurrency and rate limiting configuration\n */\nexport const ConcurrencyConfigSchema = z.object({\n  /** Maximum concurrent operations (for parallel processing) */\n  concurrency: z.number().positive().default(5),\n  /** Delay between operations in milliseconds (rate limiting) */\n  delayMs: z.number().nonnegative().default(0),\n  /** Maximum retries on failure */\n  maxRetries: z.number().nonnegative().default(3),\n});\n\nexport type ConcurrencyConfig = z.infer<typeof ConcurrencyConfigSchema>;\n\n/**\n * Feature flag configuration for conditional behavior\n */\nexport const FeatureFlagsConfigSchema = z.object({\n  /** Feature flags for conditional stage behavior */\n  featureFlags: z.record(z.string(), z.boolean()).default({}),\n});\n\nexport type FeatureFlagsConfig = z.infer<typeof FeatureFlagsConfigSchema>;\n\n/**\n * Logging and debugging configuration\n */\nexport const DebugConfigSchema = z.object({\n  /** Enable verbose logging */\n  verbose: z.boolean().default(false),\n  /** Dry run mode (no side effects) */\n  dryRun: z.boolean().default(false),\n});\n\nexport type DebugConfig = z.infer<typeof DebugConfigSchema>;\n\n// ============================================================================\n// Preset Combinators\n// ============================================================================\n\n/**\n * Add AI configuration to any schema\n *\n * @example\n * ```typescript\n * const myConfig = withAIConfig(z.object({\n *   customField: z.string(),\n * }));\n * // Result has: model, temperature, maxTokens, customField\n * ```\n */\nexport function withAIConfig<T extends z.ZodRawShape>(schema: z.ZodObject<T>) {\n  return schema.merge(AIConfigSchema);\n}\n\n/**\n * Add concurrency configuration to any schema\n *\n * @example\n * ```typescript\n * const myConfig = withConcurrency(z.object({\n *   items: z.array(z.string()),\n * }));\n * // Result has: concurrency, delayMs, maxRetries, items\n * ```\n */\nexport function withConcurrency<T extends z.ZodRawShape>(\n  schema: z.ZodObject<T>,\n) {\n  return schema.merge(ConcurrencyConfigSchema);\n}\n\n/**\n * Add feature flags configuration to any schema\n *\n * @example\n * ```typescript\n * const myConfig = withFeatureFlags(z.object({\n *   setting: z.boolean(),\n * }));\n * // Result has: featureFlags, setting\n * ```\n */\nexport function withFeatureFlags<T extends z.ZodRawShape>(\n  schema: z.ZodObject<T>,\n) {\n  return schema.merge(FeatureFlagsConfigSchema);\n}\n\n/**\n * Add debug configuration to any schema\n *\n * @example\n * ```typescript\n * const myConfig = withDebug(z.object({\n *   processType: z.string(),\n * }));\n * // Result has: verbose, dryRun, processType\n * ```\n */\nexport function withDebug<T extends z.ZodRawShape>(schema: z.ZodObject<T>) {\n  return schema.merge(DebugConfigSchema);\n}\n\n/**\n * Standard config preset combining AI + Concurrency + Feature Flags\n *\n * This is the recommended preset for most stages that need:\n * - AI/LLM operations\n * - Parallel processing\n * - Feature flagging\n *\n * @example\n * ```typescript\n * const myConfig = withStandardConfig(z.object({\n *   customOption: z.boolean().default(true),\n * }));\n * // Result has all standard fields plus customOption\n * ```\n */\nexport function withStandardConfig<T extends z.ZodRawShape>(\n  schema: z.ZodObject<T>,\n) {\n  return schema\n    .merge(AIConfigSchema)\n    .merge(ConcurrencyConfigSchema)\n    .merge(FeatureFlagsConfigSchema);\n}\n\n/**\n * Full config preset with all available options including debug\n *\n * @example\n * ```typescript\n * const myConfig = withFullConfig(z.object({\n *   specificField: z.number(),\n * }));\n * // Result has AI + Concurrency + FeatureFlags + Debug + specificField\n * ```\n */\nexport function withFullConfig<T extends z.ZodRawShape>(\n  schema: z.ZodObject<T>,\n) {\n  return schema\n    .merge(AIConfigSchema)\n    .merge(ConcurrencyConfigSchema)\n    .merge(FeatureFlagsConfigSchema)\n    .merge(DebugConfigSchema);\n}\n\n// ============================================================================\n// Preset Type Helpers\n// ============================================================================\n\n/**\n * Extract the inferred type from a config created with presets\n */\nexport type InferConfig<T extends z.ZodTypeAny> = z.infer<T>;\n\n/**\n * Standard config type (AI + Concurrency + FeatureFlags)\n */\nexport type StandardConfig = AIConfig & ConcurrencyConfig & FeatureFlagsConfig;\n\n/**\n * Full config type (Standard + Debug)\n */\nexport type FullConfig = StandardConfig & DebugConfig;\n\n// ============================================================================\n// Utility Schemas\n// ============================================================================\n\n/**\n * Empty config schema - for stages that don't need configuration\n */\nexport const EmptyConfigSchema = z.object({});\n\n/**\n * Minimal AI config - just model and temperature\n */\nexport const MinimalAIConfigSchema = z.object({\n  model: ModelKey.default(\"gemini-2.5-flash\"),\n  temperature: z.number().min(0).max(2).default(0.7),\n});\n\nexport type MinimalAIConfig = z.infer<typeof MinimalAIConfigSchema>;\n"]}