@mastra/inngest 0.0.0-course-20250527170450 → 0.0.0-custom-instrumentation-20250626084921

package/dist/_tsup-dts-rollup.d.ts

@@ -1,8 +1,8 @@
+import type { Agent } from '@mastra/core';
 import type { BaseContext } from 'inngest';
 import { ClientOptions } from 'inngest';
-import { cloneStep } from '@mastra/core/workflows';
-import { createStep } from '@mastra/core/workflows';
 import { DefaultExecutionEngine } from '@mastra/core/workflows';
+import type { Emitter } from '@mastra/core/workflows';
 import type { ExecuteFunction } from '@mastra/core/workflows';
 import type { ExecutionContext } from '@mastra/core/workflows';
 import type { ExecutionEngine } from '@mastra/core/workflows';
@@ -12,38 +12,87 @@ import type { Inngest } from 'inngest';
 import { InngestFunction } from 'inngest';
 import { InngestMiddleware } from 'inngest';
 import type { Mastra } from '@mastra/core';
+import type { ReadableStream as ReadableStream_2 } from 'node:stream/web';
 import { Run } from '@mastra/core/workflows';
 import { RuntimeContext } from '@mastra/core/di';
+import type { SerializedStepFlowEntry } from '@mastra/core/workflows';
 import { serve as serve_2 } from 'inngest/hono';
 import type { Span } from '@opentelemetry/api';
 import type { Step } from '@mastra/core/workflows';
 import type { StepFlowEntry } from '@mastra/core/workflows';
 import type { StepResult } from '@mastra/core/workflows';
+import type { StreamEvent } from '@mastra/core/workflows';
+import { Tool } from '@mastra/core/tools';
+import type { ToolExecutionContext } from '@mastra/core';
+import type { WatchEvent } from '@mastra/core/workflows';
 import { Workflow } from '@mastra/core/workflows';
 import type { WorkflowConfig } from '@mastra/core/workflows';
 import type { WorkflowResult } from '@mastra/core/workflows';
 import type { WorkflowRun } from '@mastra/core';
 import type { WorkflowRuns } from '@mastra/core';
-import type { z } from 'zod';
+import { z } from 'zod';
 
-declare function cloneWorkflow<TWorkflowId extends string = string, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>, TSteps extends Step<string, any, any, any, any>[] = Step<string, any, any, any, any>[]>(workflow: InngestWorkflow<TSteps, string, TInput, TOutput>, opts: {
-    id: TWorkflowId;
-}): InngestWorkflow<TSteps, TWorkflowId, TInput, TOutput>;
+export declare function createStep<TStepId extends string, TStepInput extends z.ZodType<any>, TStepOutput extends z.ZodType<any>, TResumeSchema extends z.ZodType<any>, TSuspendSchema extends z.ZodType<any>>(params: {
+    id: TStepId;
+    description?: string;
+    inputSchema: TStepInput;
+    outputSchema: TStepOutput;
+    resumeSchema?: TResumeSchema;
+    suspendSchema?: TSuspendSchema;
+    execute: ExecuteFunction<z.infer<TStepInput>, z.infer<TStepOutput>, z.infer<TResumeSchema>, z.infer<TSuspendSchema>, InngestEngineType>;
+}): Step<TStepId, TStepInput, TStepOutput, TResumeSchema, TSuspendSchema, InngestEngineType>;
+
+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, any>): Step<TStepId, TStepInput, TStepOutput, TResumeSchema, TSuspendSchema, InngestEngineType>;
+
+export declare function createStep<TSchemaIn extends z.ZodType<any>, TSchemaOut extends z.ZodType<any>, TContext extends ToolExecutionContext<TSchemaIn>>(tool: Tool<TSchemaIn, TSchemaOut, TContext> & {
+    inputSchema: TSchemaIn;
+    outputSchema: TSchemaOut;
+    execute: (context: TContext) => Promise<any>;
+}): Step<string, TSchemaIn, TSchemaOut, z.ZodType<any>, z.ZodType<any>, InngestEngineType>;
 
 export declare function init(inngest: Inngest): {
-    createWorkflow<TWorkflowId extends string = string, TInput extends z.ZodType<any> = z.ZodType<any, z.ZodTypeDef, any>, TOutput extends z.ZodType<any> = z.ZodType<any, z.ZodTypeDef, any>, TSteps extends Step<string, any, any>[] = Step<string, any, any, z.ZodType<any, z.ZodTypeDef, any>, z.ZodType<any, z.ZodTypeDef, any>>[]>(params: WorkflowConfig<TWorkflowId, TInput, TOutput, TSteps>): InngestWorkflow<TSteps, TWorkflowId, TInput, TOutput, TInput>;
+    createWorkflow<TWorkflowId extends string = string, TInput extends z.ZodType<any> = z.ZodType<any, z.ZodTypeDef, any>, TOutput extends z.ZodType<any> = z.ZodType<any, z.ZodTypeDef, any>, TSteps extends Step<string, any, any, any, any, InngestEngineType>[] = Step<string, any, any, any, any, InngestEngineType>[]>(params: WorkflowConfig<TWorkflowId, TInput, TOutput, TSteps>): InngestWorkflow<InngestEngineType, TSteps, TWorkflowId, TInput, TOutput, TInput>;
     createStep: typeof createStep;
-    cloneStep: typeof cloneStep;
-    cloneWorkflow: typeof cloneWorkflow;
+    cloneStep<TStepId extends string>(step: Step<string, any, any, any, any, InngestEngineType>, opts: {
+        id: TStepId;
+    }): Step<TStepId, any, any, any, any, InngestEngineType>;
+    cloneWorkflow<TWorkflowId extends string = string, TInput extends z.ZodType<any> = z.ZodType<any, z.ZodTypeDef, any>, TOutput extends z.ZodType<any> = z.ZodType<any, z.ZodTypeDef, any>, TSteps extends Step<string, any, any, any, any, InngestEngineType>[] = Step<string, any, any, any, any, InngestEngineType>[], TPrevSchema extends z.ZodType<any> = TInput>(workflow: Workflow<InngestEngineType, TSteps, string, TInput, TOutput, TPrevSchema>, opts: {
+        id: TWorkflowId;
+    }): Workflow<InngestEngineType, TSteps, TWorkflowId, TInput, TOutput, TPrevSchema>;
+};
+
+export declare type InngestEngineType = {
+    step: any;
 };
 
 export declare class InngestExecutionEngine extends DefaultExecutionEngine {
     private inngestStep;
     private inngestAttempts;
     constructor(mastra: Mastra, inngestStep: BaseContext<Inngest>['step'], inngestAttempts?: number);
-    protected fmtReturnValue<TOutput>(executionSpan: Span | undefined, emitter: {
-        emit: (event: string, data: any) => Promise<void>;
-    }, stepResults: Record<string, StepResult<any, any, any, any>>, lastOutput: StepResult<any, any, any, any>, error?: Error | string): Promise<TOutput>;
+    execute<TInput, TOutput>(params: {
+        workflowId: string;
+        runId: string;
+        graph: ExecutionGraph;
+        serializedStepGraph: SerializedStepFlowEntry[];
+        input?: TInput;
+        resume?: {
+            steps: string[];
+            stepResults: Record<string, StepResult<any, any, any, any>>;
+            resumePayload: any;
+            resumePath: number[];
+        };
+        emitter: Emitter;
+        retryConfig?: {
+            attempts?: number;
+            delay?: number;
+        };
+        runtimeContext: RuntimeContext;
+    }): Promise<TOutput>;
+    protected fmtReturnValue<TOutput>(executionSpan: Span | undefined, emitter: Emitter, stepResults: Record<string, StepResult<any, any, any, any>>, lastOutput: StepResult<any, any, any, any>, error?: Error | string): Promise<TOutput>;
     superExecuteStep({ workflowId, runId, step, stepResults, executionContext, resume, prevOutput, emitter, runtimeContext, }: {
         workflowId: string;
         runId: string;
@@ -55,11 +104,17 @@ export declare class InngestExecutionEngine extends DefaultExecutionEngine {
             resumePayload: any;
         };
         prevOutput: any;
-        emitter: {
-            emit: (event: string, data: any) => Promise<void>;
-        };
+        emitter: Emitter;
         runtimeContext: RuntimeContext;
     }): Promise<StepResult<any, any, any, any>>;
+    executeSleep({ id, duration }: {
+        id: string;
+        duration: number;
+    }): Promise<void>;
+    executeWaitForEvent({ event, timeout }: {
+        event: string;
+        timeout?: number;
+    }): Promise<any>;
     executeStep({ step, stepResults, executionContext, resume, prevOutput, emitter, runtimeContext, }: {
         step: Step<string, any, any>;
         stepResults: Record<string, StepResult<any, any, any, any>>;
@@ -79,26 +134,29 @@ export declare class InngestExecutionEngine extends DefaultExecutionEngine {
             runId?: string;
         };
         prevOutput: any;
-        emitter: {
-            emit: (event: string, data: any) => Promise<void>;
-        };
+        emitter: Emitter;
         runtimeContext: RuntimeContext;
     }): Promise<StepResult<any, any, any, any>>;
-    persistStepUpdate({ workflowId, runId, stepResults, executionContext, }: {
+    persistStepUpdate({ workflowId, runId, stepResults, executionContext, serializedStepGraph, workflowStatus, result, error, }: {
         workflowId: string;
         runId: string;
         stepResults: Record<string, StepResult<any, any, any, any>>;
+        serializedStepGraph: SerializedStepFlowEntry[];
         executionContext: ExecutionContext;
+        workflowStatus: 'success' | 'failed' | 'suspended' | 'running';
+        result?: Record<string, any>;
+        error?: string | Error;
     }): Promise<void>;
-    executeConditional({ workflowId, runId, entry, prevOutput, prevStep, stepResults, resume, executionContext, emitter, runtimeContext, }: {
+    executeConditional({ workflowId, runId, entry, prevOutput, prevStep, stepResults, serializedStepGraph, resume, executionContext, emitter, runtimeContext, }: {
         workflowId: string;
         runId: string;
         entry: {
             type: 'conditional';
             steps: StepFlowEntry[];
-            conditions: ExecuteFunction<any, any, any, any>[];
+            conditions: ExecuteFunction<any, any, any, any, InngestEngineType>[];
         };
         prevStep: StepFlowEntry;
+        serializedStepGraph: SerializedStepFlowEntry[];
         prevOutput: any;
         stepResults: Record<string, StepResult<any, any, any, any>>;
         resume?: {
@@ -108,21 +166,21 @@ export declare class InngestExecutionEngine extends DefaultExecutionEngine {
             resumePath: number[];
         };
         executionContext: ExecutionContext;
-        emitter: {
-            emit: (event: string, data: any) => Promise<void>;
-        };
+        emitter: Emitter;
         runtimeContext: RuntimeContext;
     }): Promise<StepResult<any, any, any, any>>;
 }
 
-export declare class InngestRun<TSteps extends Step<string, any, any>[] = Step<string, any, any>[], TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>> extends Run<TSteps, TInput, TOutput> {
+export declare class InngestRun<TEngineType = InngestEngineType, TSteps extends Step<string, any, any>[] = Step<string, any, any>[], TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>> extends Run<TEngineType, TSteps, TInput, TOutput> {
     #private;
     private inngest;
+    serializedStepGraph: SerializedStepFlowEntry[];
     constructor(params: {
         workflowId: string;
         runId: string;
         executionEngine: ExecutionEngine;
         executionGraph: ExecutionGraph;
+        serializedStepGraph: SerializedStepFlowEntry[];
         mastra?: Mastra;
         retryConfig?: {
             attempts?: number;
@@ -132,6 +190,7 @@ export declare class InngestRun<TSteps extends Step<string, any, any>[] = Step<s
     }, inngest: Inngest);
     getRuns(eventId: string): Promise<any>;
     getRunOutput(eventId: string): Promise<any>;
+    sendEvent(event: string, data: any): Promise<void>;
     start({ inputData, }: {
         inputData?: z.infer<TInput>;
         runtimeContext?: RuntimeContext;
@@ -141,10 +200,22 @@ export declare class InngestRun<TSteps extends Step<string, any, any>[] = Step<s
         step: Step<string, any, any, TResumeSchema, any> | [...Step<string, any, any, any, any>[], Step<string, any, any, TResumeSchema, any>] | string | string[];
         runtimeContext?: RuntimeContext;
     }): Promise<WorkflowResult<TOutput, TSteps>>;
-    watch(cb: (event: any) => void): () => void;
+    _resume<TResumeSchema extends z.ZodType<any>>(params: {
+        resumeData?: z.infer<TResumeSchema>;
+        step: Step<string, any, any, TResumeSchema, any> | [...Step<string, any, any, any, any>[], Step<string, any, any, TResumeSchema, any>] | string | string[];
+        runtimeContext?: RuntimeContext;
+    }): Promise<WorkflowResult<TOutput, TSteps>>;
+    watch(cb: (event: WatchEvent) => void, type?: 'watch' | 'watch-v2'): () => void;
+    stream({ inputData, runtimeContext }?: {
+        inputData?: z.infer<TInput>;
+        runtimeContext?: RuntimeContext;
+    }): {
+        stream: ReadableStream_2<StreamEvent>;
+        getWorkflowState: () => Promise<WorkflowResult<TOutput, TSteps>>;
+    };
 }
 
-export declare class InngestWorkflow<TSteps extends Step<string, any, any>[] = Step<string, any, any>[], TWorkflowId extends string = string, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>, TPrevSchema extends z.ZodType<any> = TInput> extends Workflow<TSteps, TWorkflowId, TInput, TOutput, TPrevSchema> {
+export declare class InngestWorkflow<TEngineType = InngestEngineType, TSteps extends Step<string, any, any>[] = Step<string, any, any>[], TWorkflowId extends string = string, TInput extends z.ZodType<any> = z.ZodType<any>, TOutput extends z.ZodType<any> = z.ZodType<any>, TPrevSchema extends z.ZodType<any> = TInput> extends Workflow<TEngineType, TSteps, TWorkflowId, TInput, TOutput, TPrevSchema> {
     #private;
     inngest: Inngest;
     private function;
@@ -157,10 +228,14 @@ export declare class InngestWorkflow<TSteps extends Step<string, any, any>[] = S
         resourceId?: string;
     }): Promise<WorkflowRuns>;
     getWorkflowRunById(runId: string): Promise<WorkflowRun | null>;
+    getWorkflowRunExecutionResult(runId: string): Promise<WatchEvent['payload']['workflowState'] | null>;
     __registerMastra(mastra: Mastra): void;
     createRun(options?: {
         runId?: string;
-    }): Run<TSteps, TInput, TOutput>;
+    }): Run<TEngineType, TSteps, TInput, TOutput>;
+    createRunAsync(options?: {
+        runId?: string;
+    }): Promise<Run<TEngineType, TSteps, TInput, TOutput>>;
     getFunction(): InngestFunction<Omit<InngestFunction.Options<Inngest<ClientOptions>, InngestMiddleware.Stack, InngestFunction.Trigger<string>[] | [{
     cron: string;
     } & Partial<Record<"event" | "if", never>>] | [{