pipeai 0.8.1 → 0.8.2

package/dist/index.d.cts

@@ -775,6 +775,7 @@ declare class Workflow<TContext, TInput = void, TOutput = void, TGates extends R
     step<TNextOutput>(id: string, fn: (params: {
         ctx: Readonly<TContext>;
         input: TOutput;
+        writer?: UIMessageStreamWriter;
     }) => MaybePromise<TNextOutput>): Workflow<TContext, TInput, TNextOutput, TGates>;
     gate<TResponse = TOutput, Id extends string = string>(id: Id & (Id extends keyof TGates ? never : Id), options?: {
         payload?: (params: {

package/dist/index.d.ts

@@ -775,6 +775,7 @@ declare class Workflow<TContext, TInput = void, TOutput = void, TGates extends R
     step<TNextOutput>(id: string, fn: (params: {
         ctx: Readonly<TContext>;
         input: TOutput;
+        writer?: UIMessageStreamWriter;
     }) => MaybePromise<TNextOutput>): Workflow<TContext, TInput, TNextOutput, TGates>;
     gate<TResponse = TOutput, Id extends string = string>(id: Id & (Id extends keyof TGates ? never : Id), options?: {
         payload?: (params: {

package/dist/index.js

@@ -1309,7 +1309,10 @@ var Workflow = class _Workflow extends SealedWorkflow {
         execute: async (state) => {
           state.output = await fn({
             ctx: state.ctx,
-            input: state.output
+            input: state.output,
+            // Present in stream mode (undefined in generate mode), letting the
+            // inline step emit UIMessageChunk parts onto the workflow's stream.
+            writer: state.writer
           });
         }
       };