@mastra/core 1.0.0-beta.21 → 1.0.0-beta.22

package/dist/workflows/workflow.d.ts

@@ -10,17 +10,18 @@ import type { PubSub } from '../events/pubsub.js';
 import type { Mastra } from '../mastra/index.js';
 import type { TracingContext, TracingOptions, TracingPolicy } from '../observability/index.js';
 import type { Processor } from '../processors/index.js';
-import { ProcessorStepSchema, ProcessorStepOutputSchema } from '../processors/step-schema.js';
+import { ProcessorStepOutputSchema, ProcessorStepInputSchema } from '../processors/step-schema.js';
 import type { StorageListWorkflowRunsInput } from '../storage/index.js';
-import type { OutputSchema } from '../stream/base/schema.js';
+import type { InferZodLikeSchema, OutputSchema, SchemaWithValidation } from '../stream/base/schema.js';
 import { WorkflowRunOutput } from '../stream/RunOutput.js';
-import type { ToolExecutionContext } from '../tools/types.js';
+import { Tool } from '../tools/index.js';
+import type { ToolExecutionContext } from '../tools/index.js';
 import type { DynamicArgument } from '../types/index.js';
 import { PUBSUB_SYMBOL } from './constants.js';
 import type { ExecutionEngine, ExecutionGraph } from './execution-engine.js';
-import type { ConditionFunction, ExecuteFunction, LoopConditionFunction, Step, SuspendOptions } from './step.js';
-import type { DefaultEngineType, DynamicMapping, ExtractSchemaFromStep, ExtractSchemaType, PathsToStringProps, SerializedStepFlowEntry, StepFlowEntry, StepResult, StepsRecord, StepWithComponent, StreamEvent, SubsetOf, TimeTravelContext, WorkflowConfig, WorkflowEngineType, WorkflowOptions, WorkflowResult, WorkflowType, WorkflowRunStatus, WorkflowState, WorkflowStateField, WorkflowStreamEvent, ToolStep, StepParams, OutputWriter } from './types.js';
-export type AgentStepOptions<TOutput extends OutputSchema = undefined> = Omit<AgentExecutionOptions<TOutput> & AgentStreamOptions, 'format' | 'tracingContext' | 'requestContext' | 'abortSignal' | 'context' | 'onStepFinish' | 'output' | 'experimental_output' | 'resourceId' | 'threadId' | 'scorers'>;
+import type { ConditionFunction, ExecuteFunction, InnerOutput, LoopConditionFunction, Step, SuspendOptions } from './step.js';
+import type { DefaultEngineType, DynamicMapping, ExtractSchemaFromStep, ExtractSchemaType, PathsToStringProps, SerializedStepFlowEntry, StepFlowEntry, StepResult, StepsRecord, StepWithComponent, StreamEvent, SubsetOf, TimeTravelContext, WorkflowConfig, WorkflowEngineType, WorkflowOptions, WorkflowResult, WorkflowType, WorkflowRunStatus, WorkflowState, WorkflowStateField, WorkflowStreamEvent, StepParams, OutputWriter } from './types.js';
+export type AgentStepOptions<TOUTPUT> = Omit<AgentExecutionOptions<TOUTPUT> & AgentStreamOptions, 'format' | 'tracingContext' | 'requestContext' | 'abortSignal' | 'context' | 'onStepFinish' | 'output' | 'experimental_output' | 'resourceId' | 'threadId' | 'scorers'>;
 export declare function mapVariable<TStep extends Step<string, any, any, any, any, any>>({ step, path, }: {
     step: TStep;
     path: PathsToStringProps<ExtractSchemaType<ExtractSchemaFromStep<TStep, 'outputSchema'>>> | '.';
@@ -28,7 +29,7 @@ export declare function mapVariable<TStep extends Step<string, any, any, any, an
     step: TStep;
     path: PathsToStringProps<ExtractSchemaType<ExtractSchemaFromStep<TStep, 'outputSchema'>>> | '.';
 };
-export declare function mapVariable<TWorkflow extends Workflow<any, any, any, any, any, any>>({ initData: TWorkflow, path, }: {
+export declare function mapVariable<TWorkflow extends Workflow<any, any, any, any, any, any, any>>({ initData: TWorkflow, path, }: {
     initData: TWorkflow;
     path: PathsToStringProps<ExtractSchemaType<ExtractSchemaFromStep<TWorkflow, 'inputSchema'>>> | '.';
 }): {
@@ -36,7 +37,7 @@ export declare function mapVariable<TWorkflow extends Workflow<any, any, any, an
     path: PathsToStringProps<ExtractSchemaType<ExtractSchemaFromStep<TWorkflow, 'inputSchema'>>> | '.';
 };
 /**
- * Creates a new workflow step
+ * Creates a step from explicit params (IMPORTANT: FIRST overload for best error messages when using .then in workflows)
  * @param params Configuration parameters for the step
  * @param params.id Unique identifier for the step
  * @param params.description Optional description of what the step does
@@ -45,29 +46,60 @@ export declare function mapVariable<TWorkflow extends Workflow<any, any, any, an
  * @param params.execute Function that performs the step's operations
  * @returns A Step object that can be added to the workflow
  */
-export declare function createStep<TStepId extends string, TState extends z.ZodObject<any>, TStepInput extends z.ZodType<any>, TStepOutput extends z.ZodType<any>, TResumeSchema extends z.ZodType<any>, TSuspendSchema extends z.ZodType<any>>(params: StepParams<TStepId, TState, TStepInput, TStepOutput, TResumeSchema, TSuspendSchema>): Step<TStepId, TState, TStepInput, TStepOutput, TResumeSchema, TSuspendSchema, DefaultEngineType>;
-export declare function createStep<TStepId extends string, TStepOutput extends z.ZodType<any>>(agent: Agent<TStepId, any>, agentOptions: AgentStepOptions<TStepOutput> & {
+export declare function createStep<TStepId extends string, TStateSchema extends z.ZodTypeAny | undefined, TInputSchema extends z.ZodTypeAny, TOutputSchema extends z.ZodTypeAny, TResumeSchema extends z.ZodTypeAny | undefined = undefined, TSuspendSchema extends z.ZodTypeAny | undefined = undefined>(params: StepParams<TStepId, TStateSchema, TInputSchema, TOutputSchema, TResumeSchema, TSuspendSchema>): Step<TStepId, TStateSchema extends z.ZodTypeAny ? z.infer<TStateSchema> : unknown, z.infer<TInputSchema>, z.infer<TOutputSchema>, TResumeSchema extends z.ZodTypeAny ? z.infer<TResumeSchema> : unknown, TSuspendSchema extends z.ZodTypeAny ? z.infer<TSuspendSchema> : unknown, DefaultEngineType>;
+/**
+ * Creates a step from an agent with structured output
+ */
+export declare function createStep<TStepId extends string, TStepOutput>(agent: Agent<TStepId, any>, agentOptions: AgentStepOptions<TStepOutput> & {
     structuredOutput: {
-        schema: TStepOutput;
+        schema: OutputSchema<TStepOutput>;
     };
     retries?: number;
     scorers?: DynamicArgument<MastraScorers>;
-}): Step<TStepId, any, z.ZodObject<{
-    prompt: z.ZodString;
-}>, TStepOutput, z.ZodType<any>, z.ZodType<any>, DefaultEngineType>;
-export declare function createStep<TStepId extends string, TStepInput extends z.ZodObject<{
-    prompt: z.ZodString;
-}>, TStepOutput extends z.ZodObject<{
-    text: z.ZodString;
-}>, TResumeSchema extends z.ZodType<any>, TSuspendSchema extends z.ZodType<any>>(agent: Agent<TStepId, any>, agentOptions?: AgentStepOptions & {
+}): Step<TStepId, unknown, {
+    prompt: string;
+}, TStepOutput, unknown, unknown, DefaultEngineType>;
+/**
+ * Creates a step from an agent (defaults to { text: string } output)
+ */
+export declare function createStep<TStepId extends string, TStepInput extends {
+    prompt: string;
+}, TStepOutput extends {
+    text: string;
+}, TResume, TSuspend>(agent: Agent<TStepId, any>, agentOptions?: AgentStepOptions<TStepOutput> & {
     retries?: number;
     scorers?: DynamicArgument<MastraScorers>;
-}): Step<TStepId, any, TStepInput, TStepOutput, TResumeSchema, TSuspendSchema, DefaultEngineType>;
-export declare function createStep<TSchemaIn extends z.ZodType<any>, TSuspendSchema extends z.ZodType<any>, TResumeSchema extends z.ZodType<any>, TSchemaOut extends z.ZodType<any>, TContext extends ToolExecutionContext<TSuspendSchema, TResumeSchema>>(tool: ToolStep<TSchemaIn, TSuspendSchema, TResumeSchema, TSchemaOut, TContext>, toolOptions?: {
+}): Step<TStepId, unknown, TStepInput, TStepOutput, TResume, TSuspend, DefaultEngineType>;
+/**
+ * Creates a step from a tool
+ */
+export declare function createStep<TSchemaIn, TSuspend, TResume, TSchemaOut, TContext extends ToolExecutionContext<TSuspend, TResume>, TId extends string>(tool: Tool<TSchemaIn, TSchemaOut, TSuspend, TResume, TContext, TId>, toolOptions?: {
     retries?: number;
     scorers?: DynamicArgument<MastraScorers>;
-}): Step<string, any, TSchemaIn, TSchemaOut, z.ZodType<any>, z.ZodType<any>, DefaultEngineType>;
-export declare function createStep<TProcessorId extends string>(processor: Processor<TProcessorId>): Step<`processor:${TProcessorId}`, any, typeof ProcessorStepSchema, typeof ProcessorStepOutputSchema, any, any, DefaultEngineType>;
+}): Step<TId, unknown, TSchemaIn, TSchemaOut, TSuspend, TResume, DefaultEngineType>;
+/**
+ * Creates a step from a Processor - wraps a Processor as a workflow step
+ * Note: We require at least one processor method to distinguish from StepParams
+ */
+export declare function createStep<TProcessorId extends string>(processor: (Processor<TProcessorId> & {
+    processInput: Function;
+}) | (Processor<TProcessorId> & {
+    processInputStream: Function;
+}) | (Processor<TProcessorId> & {
+    processInputStep: Function;
+}) | (Processor<TProcessorId> & {
+    processOutputStream: Function;
+}) | (Processor<TProcessorId> & {
+    processOutputResult: Function;
+}) | (Processor<TProcessorId> & {
+    processOutputStep: Function;
+})): Step<`processor:${TProcessorId}`, unknown, z.infer<typeof ProcessorStepInputSchema>, z.infer<typeof ProcessorStepOutputSchema>, unknown, unknown, DefaultEngineType>;
+/**
+ * IMPORTANT: Fallback overload - provides better error messages when StepParams doesn't match
+ * This should be LAST and will show clearer errors about what's wrong
+ * This is a copy of first one, KEEP THIS IN SYNC!
+ */
+export declare function createStep<TStepId extends string, TStateSchema extends z.ZodTypeAny | undefined, TInputSchema extends z.ZodTypeAny, TOutputSchema extends z.ZodTypeAny, TResumeSchema extends z.ZodTypeAny | undefined = undefined, TSuspendSchema extends z.ZodTypeAny | undefined = undefined>(params: StepParams<TStepId, TStateSchema, TInputSchema, TOutputSchema, TResumeSchema, TSuspendSchema>): Step<TStepId, TStateSchema extends z.ZodTypeAny ? z.infer<TStateSchema> : unknown, z.infer<TInputSchema>, z.infer<TOutputSchema>, TResumeSchema extends z.ZodTypeAny ? z.infer<TResumeSchema> : unknown, TSuspendSchema extends z.ZodTypeAny ? z.infer<TSuspendSchema> : unknown, DefaultEngineType>;
 export declare function cloneStep<TStepId extends string>(step: Step<string, any, any, any, any, any, DefaultEngineType>, opts: {
     id: TStepId;
 }): Step<TStepId, any, any, any, any, any, DefaultEngineType>;
@@ -76,17 +108,17 @@ export declare function cloneStep<TStepId extends string>(step: Step<string, any
  * A Processor must have an 'id' property and at least one processor method.
  */
 export declare function isProcessor(obj: unknown): obj is Processor;
-export declare function createWorkflow<TWorkflowId extends string = string, TState extends z.ZodObject<any> = z.ZodObject<any>, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>, TSteps extends Step<string, any, any, any, any, any, DefaultEngineType>[] = Step<string, any, any, any, any, any, DefaultEngineType>[]>(params: WorkflowConfig<TWorkflowId, TState, TInput, TOutput, TSteps>): Workflow<DefaultEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TInput>;
-export declare function cloneWorkflow<TWorkflowId extends string = string, TState extends z.ZodObject<any> = z.ZodObject<any>, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>, TSteps extends Step<string, any, any, any, any, any, DefaultEngineType>[] = Step<string, any, any, any, any, any, DefaultEngineType>[], TPrevSchema extends z.ZodType<any> = TInput>(workflow: Workflow<DefaultEngineType, TSteps, string, TState, TInput, TOutput, TPrevSchema>, opts: {
+export declare function createWorkflow<TWorkflowId extends string = string, TState = unknown, TInput = unknown, TOutput = unknown, TSteps extends Step<string, any, any, any, any, any, DefaultEngineType>[] = Step[]>(params: WorkflowConfig<TWorkflowId, TState, TInput, TOutput, TSteps>): Workflow<DefaultEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TInput>;
+export declare function cloneWorkflow<TWorkflowId extends string = string, TState = unknown, TInput = unknown, TOutput = unknown, TSteps extends Step<string, any, any, any, any, any, DefaultEngineType>[] = Step<string, any, any, any, any, any, DefaultEngineType>[], TPrevSchema = TInput>(workflow: Workflow<DefaultEngineType, TSteps, string, TState, TInput, TOutput, TPrevSchema>, opts: {
     id: TWorkflowId;
 }): Workflow<DefaultEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TPrevSchema>;
-export declare class Workflow<TEngineType = any, TSteps extends Step<string, any, any, any, any, any, TEngineType>[] = Step<string, any, any, any, any, any, TEngineType>[], TWorkflowId extends string = string, TState extends z.ZodObject<any> = z.ZodObject<any>, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>, TPrevSchema extends z.ZodType<any> = TInput> extends MastraBase implements Step<TWorkflowId, TState, TInput, TOutput, any, any, DefaultEngineType> {
+export declare class Workflow<TEngineType = DefaultEngineType, TSteps extends Step<string, any, any, any, any, any, TEngineType>[] = Step<string, unknown, unknown, unknown, unknown, unknown, TEngineType>[], TWorkflowId extends string = string, TState = unknown, TInput = unknown, TOutput = unknown, TPrevSchema = TInput> extends MastraBase implements Step<TWorkflowId, TState, TInput, TOutput | undefined, any, any, DefaultEngineType> {
     #private;
     id: TWorkflowId;
     description?: string | undefined;
-    inputSchema: TInput;
-    outputSchema: TOutput;
-    stateSchema?: TState;
+    inputSchema: SchemaWithValidation<TInput>;
+    outputSchema: SchemaWithValidation<TOutput>;
+    stateSchema?: SchemaWithValidation<TState>;
     steps: Record<string, StepWithComponent>;
     stepDefs?: TSteps;
     engineType: WorkflowEngineType;
@@ -103,7 +135,7 @@ export declare class Workflow<TEngineType = any, TSteps extends Step<string, any
     };
     constructor({ mastra, id, inputSchema, outputSchema, stateSchema, description, executionEngine, retryConfig, steps, options, type, }: WorkflowConfig<TWorkflowId, TState, TInput, TOutput, TSteps>);
     get runs(): Map<string, Run<TEngineType, TSteps, TState, TInput, TOutput>>;
-    get mastra(): Mastra<Record<string, Agent<any, import("../agent").ToolsInput>>, Record<string, Workflow<any, any, any, any, any, any, any>>, Record<string, import("../vector").MastraVector<any>>, Record<string, import("../tts").MastraTTS>, import("../logger").IMastraLogger, Record<string, import("../mcp").MCPServerBase<any>>, Record<string, import("../evals").MastraScorer<any, any, any, any>>, Record<string, import("../tools").ToolAction<any, any, any, any, any, any>>, Record<string, Processor<any, unknown>>, Record<string, import("../memory").MastraMemory>> | undefined;
+    get mastra(): Mastra<Record<string, Agent<any, import("../agent").ToolsInput, undefined>>, Record<string, Workflow<any, any, any, any, any, any, any>>, Record<string, import("../vector").MastraVector<any>>, Record<string, import("../tts").MastraTTS>, import("../logger").IMastraLogger, Record<string, import("../mcp").MCPServerBase<any>>, Record<string, import("../evals").MastraScorer<any, any, any, any>>, Record<string, import("../tools").ToolAction<any, any, any, any, any, any>>, Record<string, Processor<any, unknown>>, Record<string, import("../memory").MastraMemory>> | undefined;
     get options(): Omit<WorkflowOptions, "validateInputs" | "shouldPersistSnapshot"> & Required<Pick<WorkflowOptions, "validateInputs" | "shouldPersistSnapshot">>;
     __registerMastra(mastra: Mastra): void;
     __registerPrimitives(p: MastraPrimitives): void;
@@ -112,24 +144,27 @@ export declare class Workflow<TEngineType = any, TSteps extends Step<string, any
      * Adds a step to the workflow
      * @param step The step to add to the workflow
      * @returns The workflow instance for chaining
+     *
+     * The step's inputSchema must be satisfied by the previous step's output (or workflow input for first step).
+     * This means: TPrevSchema must be assignable to TStepInput
      */
-    then<TStepId extends string, TStepState extends z.ZodObject<any>, TStepInputSchema extends z.ZodType<any>, TSchemaOut extends z.ZodType<any>>(step: Step<TStepId, SubsetOf<TStepState, TState>, z.TypeOf<TPrevSchema> extends z.TypeOf<TStepInputSchema> ? TStepInputSchema : never, TSchemaOut, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut>;
+    then<TStepId extends string, TStepState, TStepInput, TSchemaOut>(step: Step<TStepId, unknown extends TStepState ? TStepState : SubsetOf<TStepState, TState>, TPrevSchema extends TStepInput ? TStepInput : TPrevSchema, TSchemaOut, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut>;
     /**
      * Adds a sleep step to the workflow
      * @param duration The duration to sleep for
      * @returns The workflow instance for chaining
      */
-    sleep(duration: number | ExecuteFunction<z.infer<TState>, z.infer<TPrevSchema>, number, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TPrevSchema>;
+    sleep(duration: number | ExecuteFunction<TState, TPrevSchema, number, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TPrevSchema>;
     /**
      * Adds a sleep until step to the workflow
      * @param date The date to sleep until
      * @returns The workflow instance for chaining
      */
-    sleepUntil(date: Date | ExecuteFunction<z.infer<TState>, z.infer<TPrevSchema>, Date, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TPrevSchema>;
+    sleepUntil(date: Date | ExecuteFunction<TState, TPrevSchema, Date, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TPrevSchema>;
     /**
      * @deprecated waitForEvent has been removed. Please use suspend/resume instead.
      */
-    waitForEvent<TStepState extends z.ZodObject<any>, TStepInputSchema extends TPrevSchema, TStepId extends string, TSchemaOut extends z.ZodType<any>>(_event: string, _step: Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType>, _opts?: {
+    waitForEvent<TStepState, TStepInputSchema extends TPrevSchema, TStepId extends string, TSchemaOut>(_event: string, _step: Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType>, _opts?: {
         timeout?: number;
     }): void;
     map(mappingConfig: {
@@ -138,31 +173,31 @@ export declare class Workflow<TEngineType = any, TSteps extends Step<string, any
             path: string;
         } | {
             value: any;
-            schema: z.ZodType<any>;
+            schema: SchemaWithValidation<any>;
         } | {
             initData: Workflow<TEngineType, any, any, any, any, any, any>;
             path: string;
         } | {
             requestContextPath: string;
-            schema: z.ZodType<any>;
-        } | DynamicMapping<TPrevSchema, z.ZodType<any>>;
-    } | ExecuteFunction<z.infer<TState>, z.infer<TPrevSchema>, any, any, any, TEngineType>, stepOptions?: {
+            schema: SchemaWithValidation<any>;
+        } | DynamicMapping<TPrevSchema, any>;
+    } | ExecuteFunction<TState, TPrevSchema, any, any, any, TEngineType>, stepOptions?: {
         id?: string | null;
     }): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, any>;
     parallel<TParallelSteps extends readonly Step<string, any, TPrevSchema, any, any, any, TEngineType>[]>(steps: TParallelSteps & {
-        [K in keyof TParallelSteps]: TParallelSteps[K] extends Step<string, infer S extends z.ZodObject<any>, TPrevSchema, infer O, any, // Don't infer TResumeSchema - causes issues with heterogeneous tuples
-        any, // Don't infer TSuspendSchema - causes issues with heterogeneous tuples
+        [K in keyof TParallelSteps]: TParallelSteps[K] extends Step<string, infer S, TPrevSchema, infer O, any, // Don't infer TResume - causes issues with heterogeneous tuples
+        any, // Don't infer TSuspend - causes issues with heterogeneous tuples
         TEngineType> ? Step<string, SubsetOf<S, TState>, TPrevSchema, O, any, any, TEngineType> : `Error: Expected Step with state schema that is a subset of workflow state`;
-    }): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, z.ZodObject<{ [K in keyof StepsRecord<TParallelSteps>]: StepsRecord<TParallelSteps>[K]["outputSchema"]; }, any, z.ZodTypeAny>>;
+    }): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, { [K in keyof StepsRecord<TParallelSteps>]: InferZodLikeSchema<StepsRecord<TParallelSteps>[K]["outputSchema"]>; }>;
     branch<TBranchSteps extends Array<[
-        ConditionFunction<z.infer<TState>, z.infer<TPrevSchema>, any, any, TEngineType>,
+        ConditionFunction<TState, TPrevSchema, any, any, any, TEngineType>,
         Step<string, any, TPrevSchema, any, any, any, TEngineType>
-    ]>>(steps: TBranchSteps): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, z.ZodObject<{ [K in keyof StepsRecord<{ [K_1 in keyof TBranchSteps]: TBranchSteps[K_1][1]; }[number][]>]: z.ZodOptional<StepsRecord<{ [K_1 in keyof TBranchSteps]: TBranchSteps[K_1][1]; }[number][]>[K]["outputSchema"]>; }, any, z.ZodTypeAny>>;
-    dowhile<TStepState extends z.ZodObject<any>, TStepInputSchema extends TPrevSchema, TStepId extends string, TSchemaOut extends z.ZodType<any>>(step: Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType>, condition: LoopConditionFunction<z.infer<TState>, any, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut>;
-    dountil<TStepState extends z.ZodObject<any>, TStepInputSchema extends TPrevSchema, TStepId extends string, TSchemaOut extends z.ZodType<any>>(step: Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType>, condition: LoopConditionFunction<z.infer<TState>, any, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut>;
-    foreach<TPrevIsArray extends TPrevSchema extends z.ZodArray<any> ? true : false, TStepState extends z.ZodObject<any>, TStepInputSchema extends TPrevSchema extends z.ZodArray<infer TElement> ? TElement : never, TStepId extends string, TSchemaOut extends z.ZodType<any>>(step: TPrevIsArray extends true ? Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType> : 'Previous step must return an array type', opts?: {
+    ]>>(steps: TBranchSteps): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, { [K in keyof StepsRecord<{ [K_1 in keyof TBranchSteps]: TBranchSteps[K_1][1]; }[number][]>]?: InferZodLikeSchema<StepsRecord<{ [K_1 in keyof TBranchSteps]: TBranchSteps[K_1][1]; }[number][]>[K]["outputSchema"]>; }>;
+    dowhile<TStepState, TStepInputSchema extends TPrevSchema, TStepId extends string, TSchemaOut>(step: Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType>, condition: LoopConditionFunction<TState, any, any, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut>;
+    dountil<TStepState, TStepInputSchema extends TPrevSchema, TStepId extends string, TSchemaOut>(step: Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType>, condition: LoopConditionFunction<TState, any, any, any, any, TEngineType>): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut>;
+    foreach<TPrevIsArray extends TPrevSchema extends any[] ? true : false, TStepState, TStepInputSchema extends TPrevSchema extends (infer TElement)[] ? TElement : never, TStepId extends string, TSchemaOut>(step: TPrevIsArray extends true ? Step<TStepId, SubsetOf<TStepState, TState>, TStepInputSchema, TSchemaOut, any, any, TEngineType> : 'Previous step must return an array type', opts?: {
         concurrency: number;
-    }): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, z.ZodArray<TSchemaOut>>;
+    }): Workflow<TEngineType, TSteps, TWorkflowId, TState, TInput, TOutput, TSchemaOut[]>;
     /**
      * Builds the execution graph for this workflow
      * @returns The execution graph that can be used to execute the workflow
@@ -194,15 +229,14 @@ export declare class Workflow<TEngineType = any, TSteps extends Step<string, any
     }): Promise<MastraScorers>;
     execute({ runId, inputData, resumeData, state, setState, suspend, restart, resume, timeTravel, [PUBSUB_SYMBOL]: pubsub, mastra, requestContext, abort, abortSignal, retryCount, tracingContext, outputWriter, validateInputs, perStep, }: {
         runId?: string;
-        inputData: z.infer<TInput>;
-        resumeData?: any;
-        state: z.infer<TState>;
-        setState: (state: z.infer<TState>) => Promise<void>;
-        getStepResult<T extends Step<any, any, any, any, any, any, TEngineType>>(stepId: T): T['outputSchema'] extends undefined ? unknown : z.infer<NonNullable<T['outputSchema']>>;
-        suspend: (suspendPayload: any, suspendOptions?: SuspendOptions) => Promise<any>;
+        inputData: TInput;
+        resumeData?: unknown;
+        state: TState;
+        setState: (state: TState) => Promise<void>;
+        suspend: (suspendPayload: any, suspendOptions?: SuspendOptions) => InnerOutput | Promise<InnerOutput>;
         restart?: boolean;
         timeTravel?: {
-            inputData?: z.infer<TInput>;
+            inputData?: TInput;
             steps: string[];
             nestedStepResults?: Record<string, Record<string, StepResult<any, any, any, any>>>;
             resumeData?: any;
@@ -226,7 +260,7 @@ export declare class Workflow<TEngineType = any, TSteps extends Step<string, any
         outputWriter?: OutputWriter;
         validateInputs?: boolean;
         perStep?: boolean;
-    }): Promise<z.infer<TOutput>>;
+    }): Promise<TOutput | undefined>;
     listWorkflowRuns(args?: StorageListWorkflowRunsInput): Promise<import("../storage").WorkflowRuns>;
     listActiveWorkflowRuns(): Promise<{
         runs: import("../storage").WorkflowRun[];
@@ -255,7 +289,7 @@ export declare class Workflow<TEngineType = any, TSteps extends Step<string, any
 /**
  * Represents a workflow run that can be executed
  */
-export declare class Run<TEngineType = any, TSteps extends Step<string, any, any, any, any, any, TEngineType>[] = Step<string, any, any, any, any, any, TEngineType>[], TState extends z.ZodObject<any> = z.ZodObject<any>, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>> {
+export declare class Run<TEngineType = DefaultEngineType, TSteps extends Step<string, any, any, any, any, any, TEngineType>[] = Step<string, unknown, unknown, unknown, unknown, unknown, TEngineType>[], TState = unknown, TInput = unknown, TOutput = unknown> {
     #private;
     protected pubsub: PubSub;
     /**
@@ -304,12 +338,12 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
     readonly workflowSteps: Record<string, StepWithComponent>;
     workflowRunStatus: WorkflowRunStatus;
     readonly workflowEngineType: WorkflowEngineType;
-    get mastra(): Mastra<Record<string, Agent<any, import("../agent").ToolsInput>>, Record<string, Workflow<any, any, any, any, any, any, any>>, Record<string, import("../vector").MastraVector<any>>, Record<string, import("../tts").MastraTTS>, import("../logger").IMastraLogger, Record<string, import("../mcp").MCPServerBase<any>>, Record<string, import("../evals").MastraScorer<any, any, any, any>>, Record<string, import("../tools").ToolAction<any, any, any, any, any, any>>, Record<string, Processor<any, unknown>>, Record<string, import("../memory").MastraMemory>> | undefined;
-    protected streamOutput?: WorkflowRunOutput<WorkflowResult<TState, TInput, TOutput, TSteps>>;
+    get mastra(): Mastra<Record<string, Agent<any, import("../agent").ToolsInput, undefined>>, Record<string, Workflow<any, any, any, any, any, any, any>>, Record<string, import("../vector").MastraVector<any>>, Record<string, import("../tts").MastraTTS>, import("../logger").IMastraLogger, Record<string, import("../mcp").MCPServerBase<any>>, Record<string, import("../evals").MastraScorer<any, any, any, any>>, Record<string, import("../tools").ToolAction<any, any, any, any, any, any>>, Record<string, Processor<any, unknown>>, Record<string, import("../memory").MastraMemory>> | undefined;
+    streamOutput?: WorkflowRunOutput<WorkflowResult<TState, TInput, TOutput, TSteps>>;
     protected closeStreamAction?: () => Promise<void>;
     protected executionResults?: Promise<WorkflowResult<TState, TInput, TOutput, TSteps>>;
-    protected stateSchema?: z.ZodObject<any>;
-    protected inputSchema?: z.ZodType<any>;
+    protected stateSchema?: SchemaWithValidation<TState>;
+    protected inputSchema?: SchemaWithValidation<TInput>;
     protected cleanup?: () => void;
     protected retryConfig?: {
         attempts?: number;
@@ -319,8 +353,8 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
         workflowId: string;
         runId: string;
         resourceId?: string;
-        stateSchema?: z.ZodObject<any>;
-        inputSchema?: z.ZodType<any>;
+        stateSchema?: SchemaWithValidation<TState>;
+        inputSchema?: SchemaWithValidation<TInput>;
         executionEngine: ExecutionEngine;
         executionGraph: ExecutionGraph;
         mastra?: Mastra;
@@ -342,13 +376,19 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * This aborts any running execution and updates the workflow status to 'canceled' in storage.
      */
     cancel(): Promise<void>;
-    protected _validateInput(inputData: z.input<TInput>): Promise<z.input<TInput>>;
-    protected _validateInitialState(initialState: z.input<TState>): Promise<z.input<TState>>;
-    protected _validateResumeData<TResumeSchema extends z.ZodType<any>>(resumeData: z.input<TResumeSchema>, suspendedStep?: StepWithComponent): Promise<z.input<TResumeSchema>>;
-    protected _validateTimetravelInputData<TInputSchema extends z.ZodType<any>>(inputData: z.input<TInputSchema>, step: Step<string, any, TInputSchema, any, any, any, TEngineType>): Promise<z.input<TInputSchema>>;
-    protected _start({ inputData, initialState, requestContext, outputWriter, tracingContext, tracingOptions, format, outputOptions, perStep, }: {
-        inputData?: z.input<TInput>;
-        initialState?: z.input<TState>;
+    protected _validateInput(inputData?: TInput): Promise<TInput | undefined>;
+    protected _validateInitialState(initialState?: TState): Promise<TState | undefined>;
+    protected _validateResumeData<TResume>(resumeData: TResume, suspendedStep?: StepWithComponent): Promise<TResume>;
+    protected _validateTimetravelInputData<TInput>(inputData: TInput, step: Step<string, any, TInput, any, any, any, TEngineType>): Promise<TInput>;
+    protected _start({ inputData, initialState, requestContext, outputWriter, tracingContext, tracingOptions, format, outputOptions, perStep, }: (TInput extends unknown ? {
+        inputData?: TInput;
+    } : {
+        inputData: TInput;
+    }) & (TState extends unknown ? {
+        initialState?: TState;
+    } : {
+        initialState: TState;
+    }) & {
         requestContext?: RequestContext;
         outputWriter?: OutputWriter;
         tracingContext?: TracingContext;
@@ -365,9 +405,15 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * @param input The input data for the workflow
      * @returns A promise that resolves to the workflow output
      */
-    start(args: {
-        inputData?: z.input<TInput>;
-        initialState?: z.input<TState>;
+    start(args: (TInput extends unknown ? {
+        inputData?: TInput;
+    } : {
+        inputData: TInput;
+    }) & (TState extends unknown ? {
+        initialState?: TState;
+    } : {
+        initialState: TState;
+    }) & {
         requestContext?: RequestContext;
         outputWriter?: OutputWriter;
         tracingContext?: TracingContext;
@@ -385,9 +431,15 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * @param args The input data and configuration for the workflow
      * @returns A promise that resolves immediately with the runId
      */
-    startAsync(args: {
-        inputData?: z.input<TInput>;
-        initialState?: z.input<TState>;
+    startAsync(args: (TInput extends unknown ? {
+        inputData?: TInput;
+    } : {
+        inputData: TInput;
+    }) & (TState extends unknown ? {
+        initialState?: TState;
+    } : {
+        initialState: TState;
+    }) & {
         requestContext?: RequestContext;
         tracingOptions?: TracingOptions;
         outputOptions?: {
@@ -403,8 +455,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * @param input The input data for the workflow
      * @returns A promise that resolves to the workflow output
      */
-    streamLegacy({ inputData, requestContext, onChunk, tracingContext, tracingOptions, }?: {
-        inputData?: z.input<TInput>;
+    streamLegacy({ inputData, requestContext, onChunk, tracingContext, tracingOptions, }?: (TInput extends unknown ? {
+        inputData?: TInput;
+    } : {
+        inputData: TInput;
+    }) & {
         requestContext?: RequestContext;
         tracingContext?: TracingContext;
         onChunk?: (chunk: StreamEvent) => Promise<unknown>;
@@ -425,23 +480,24 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * @returns A readable stream of the workflow events
      */
     observeStream(): ReadableStream<WorkflowStreamEvent>;
-    streamAsync({ inputData, requestContext, perStep, }?: {
-        inputData?: z.input<TInput>;
-        requestContext?: RequestContext;
-        perStep?: boolean;
-    }): Promise<ReturnType<typeof this.stream>>;
     /**
      * Starts the workflow execution with the provided input as a stream
      * @param input The input data for the workflow
      * @returns A promise that resolves to the workflow output
      */
-    stream({ inputData, requestContext, tracingContext, tracingOptions, closeOnSuspend, initialState, outputOptions, perStep, }?: {
-        inputData?: z.input<TInput>;
+    stream({ inputData, requestContext, tracingContext, tracingOptions, closeOnSuspend, initialState, outputOptions, perStep, }: (TInput extends unknown ? {
+        inputData?: TInput;
+    } : {
+        inputData: TInput;
+    }) & (TState extends unknown ? {
+        initialState?: TState;
+    } : {
+        initialState: TState;
+    }) & {
         requestContext?: RequestContext;
         tracingContext?: TracingContext;
         tracingOptions?: TracingOptions;
         closeOnSuspend?: boolean;
-        initialState?: z.input<TState>;
         outputOptions?: {
             includeState?: boolean;
             includeResumeLabels?: boolean;
@@ -453,11 +509,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * @param input The input data for the workflow
      * @returns A promise that resolves to the workflow output
      */
-    resumeStream<TResumeSchema extends z.ZodType<any>>({ step, resumeData, requestContext, tracingContext, tracingOptions, forEachIndex, outputOptions, perStep, }?: {
-        resumeData?: z.input<TResumeSchema>;
-        step?: Step<string, any, any, any, TResumeSchema, any, TEngineType> | [
+    resumeStream<TResume>({ step, resumeData, requestContext, tracingContext, tracingOptions, forEachIndex, outputOptions, perStep, }?: {
+        resumeData?: TResume;
+        step?: Step<string, any, any, any, TResume, any, TEngineType> | [
             ...Step<string, any, any, any, any, any, TEngineType>[],
-            Step<string, any, any, any, TResumeSchema, any, TEngineType>
+            Step<string, any, any, any, TResume, any, TEngineType>
         ] | string | string[];
         requestContext?: RequestContext;
         tracingContext?: TracingContext;
@@ -477,11 +533,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
      * @internal
      */
     watchAsync(cb: (event: WorkflowStreamEvent) => void): Promise<() => void>;
-    resume<TResumeSchema extends z.ZodType<any>>(params: {
-        resumeData?: z.input<TResumeSchema>;
-        step?: Step<string, any, any, any, TResumeSchema, any, TEngineType> | [
+    resume<TResume>(params: {
+        resumeData?: TResume;
+        step?: Step<string, any, any, any, TResume, any, TEngineType> | [
             ...Step<string, any, any, any, any, any, TEngineType>[],
-            Step<string, any, any, any, TResumeSchema, any, TEngineType>
+            Step<string, any, any, any, TResume, any, TEngineType>
         ] | string | string[];
         label?: string;
         requestContext?: RequestContext;
@@ -506,11 +562,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
         tracingContext?: TracingContext;
         tracingOptions?: TracingOptions;
     }): Promise<WorkflowResult<TState, TInput, TOutput, TSteps>>;
-    protected _resume<TResumeSchema extends z.ZodType<any>>(params: {
-        resumeData?: z.input<TResumeSchema>;
-        step?: Step<string, any, any, TResumeSchema, any, any, TEngineType> | [
+    protected _resume<TResume>(params: {
+        resumeData?: TResume;
+        step?: Step<string, any, any, TResume, any, any, TEngineType> | [
             ...Step<string, any, any, any, any, any, TEngineType>[],
-            Step<string, any, any, TResumeSchema, any, any, TEngineType>
+            Step<string, any, any, TResume, any, any, TEngineType>
         ] | string | string[];
         label?: string;
         requestContext?: RequestContext;
@@ -533,14 +589,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
         tracingContext?: TracingContext;
         tracingOptions?: TracingOptions;
     }): Promise<WorkflowResult<TState, TInput, TOutput, TSteps>>;
-    protected _timeTravel<TInputSchema extends z.ZodType<any>>({ inputData, resumeData, initialState, step: stepParam, context, nestedStepsContext, requestContext, outputWriter, tracingContext, tracingOptions, outputOptions, perStep, }: {
-        inputData?: z.input<TInputSchema>;
+    protected _timeTravel<TInput>({ inputData, resumeData, initialState, step: stepParam, context, nestedStepsContext, requestContext, outputWriter, tracingContext, tracingOptions, outputOptions, perStep, }: {
+        inputData?: TInput;
         resumeData?: any;
-        initialState?: z.input<TState>;
-        step: Step<string, any, TInputSchema, any, any, any, TEngineType> | [
-            ...Step<string, any, any, any, any, any, TEngineType>[],
-            Step<string, any, TInputSchema, any, any, any, TEngineType>
-        ] | string | string[];
+        initialState?: TState;
+        step: Step<string, any, TInput, any, any, any, TEngineType> | [...Step<string, any, any, any, any, any, TEngineType>[], Step<string, any, TInput, any, any, any, TEngineType>] | string | string[];
         context?: TimeTravelContext<any, any, any, any>;
         nestedStepsContext?: Record<string, TimeTravelContext<any, any, any, any>>;
         requestContext?: RequestContext;
@@ -553,14 +606,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
         };
         perStep?: boolean;
     }): Promise<WorkflowResult<TState, TInput, TOutput, TSteps>>;
-    timeTravel<TInputSchema extends z.ZodType<any>>(args: {
-        inputData?: z.input<TInputSchema>;
+    timeTravel<TInput>(args: {
+        inputData?: TInput;
         resumeData?: any;
-        initialState?: z.input<TState>;
-        step: Step<string, any, TInputSchema, any, any, any, TEngineType> | [
-            ...Step<string, any, any, any, any, any, TEngineType>[],
-            Step<string, any, TInputSchema, any, any, any, TEngineType>
-        ] | string | string[];
+        initialState?: TState;
+        step: Step<string, any, TInput, any, any, any, TEngineType> | [...Step<string, any, any, any, any, any, TEngineType>[], Step<string, any, TInput, any, any, any, TEngineType>] | string | string[];
         context?: TimeTravelContext<any, any, any, any>;
         nestedStepsContext?: Record<string, TimeTravelContext<any, any, any, any>>;
         requestContext?: RequestContext;
@@ -573,14 +623,11 @@ export declare class Run<TEngineType = any, TSteps extends Step<string, any, any
         };
         perStep?: boolean;
     }): Promise<WorkflowResult<TState, TInput, TOutput, TSteps>>;
-    timeTravelStream<TInputSchema extends z.ZodType<any>>({ inputData, resumeData, initialState, step, context, nestedStepsContext, requestContext, tracingContext, tracingOptions, outputOptions, perStep, }: {
-        inputData?: z.input<TInputSchema>;
-        initialState?: z.input<TState>;
+    timeTravelStream<TTravelInput>({ inputData, resumeData, initialState, step, context, nestedStepsContext, requestContext, tracingContext, tracingOptions, outputOptions, perStep, }: {
+        inputData?: TTravelInput;
+        initialState?: TState;
         resumeData?: any;
-        step: Step<string, any, TInputSchema, any, any, any, TEngineType> | [
-            ...Step<string, any, any, any, any, any, TEngineType>[],
-            Step<string, any, TInputSchema, any, any, any, TEngineType>
-        ] | string | string[];
+        step: Step<string, any, any, any, any, any, TEngineType> | [...Step<string, any, any, any, any, any, TEngineType>[], Step<string, any, any, any, any, any, TEngineType>] | string | string[];
         context?: TimeTravelContext<any, any, any, any>;
         nestedStepsContext?: Record<string, TimeTravelContext<any, any, any, any>>;
         requestContext?: RequestContext;