@mastra/ai-sdk 1.0.0-beta.3 → 1.0.0-beta.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,14 +1,42 @@
1
1
  import type { AgentExecutionOptions } from '@mastra/core/agent';
2
+ import type { MessageListInput } from '@mastra/core/agent/message-list';
3
+ import type { Mastra } from '@mastra/core/mastra';
2
4
  import { registerApiRoute } from '@mastra/core/server';
3
5
  import type { OutputSchema } from '@mastra/core/stream';
6
+ import type { InferUIMessageChunk, UIMessage } from 'ai';
7
+ export type NetworkStreamHandlerParams<OUTPUT extends OutputSchema = undefined> = AgentExecutionOptions<OUTPUT, 'mastra'> & {
8
+ messages: MessageListInput;
9
+ };
10
+ export type NetworkStreamHandlerOptions<OUTPUT extends OutputSchema = undefined> = {
11
+ mastra: Mastra;
12
+ agentId: string;
13
+ params: NetworkStreamHandlerParams<OUTPUT>;
14
+ defaultOptions?: AgentExecutionOptions<OUTPUT, 'mastra'>;
15
+ };
16
+ /**
17
+ * Framework-agnostic handler for streaming agent network execution in AI SDK format.
18
+ * Use this function directly when you need to handle network streaming outside of Hono/registerApiRoute.
19
+ *
20
+ * @example
21
+ * // Next.js App Router
22
+ * export async function POST(req: Request) {
23
+ * const params = await req.json();
24
+ * return handleNetworkStream({
25
+ * mastra,
26
+ * agentId: 'my-agent',
27
+ * params,
28
+ * });
29
+ * }
30
+ */
31
+ export declare function handleNetworkStream<UI_MESSAGE extends UIMessage, OUTPUT extends OutputSchema = undefined>({ mastra, agentId, params, defaultOptions, }: NetworkStreamHandlerOptions<OUTPUT>): Promise<ReadableStream<InferUIMessageChunk<UI_MESSAGE>>>;
4
32
  export type NetworkRouteOptions<OUTPUT extends OutputSchema = undefined> = {
5
33
  path: `${string}:agentId${string}`;
6
34
  agent?: never;
7
- defaultOptions?: AgentExecutionOptions<OUTPUT, 'aisdk'>;
35
+ defaultOptions?: AgentExecutionOptions<OUTPUT, 'mastra'>;
8
36
  } | {
9
37
  path: string;
10
38
  agent: string;
11
- defaultOptions?: AgentExecutionOptions<OUTPUT, 'aisdk'>;
39
+ defaultOptions?: AgentExecutionOptions<OUTPUT, 'mastra'>;
12
40
  };
13
41
  export declare function networkRoute<OUTPUT extends OutputSchema = undefined>({ path, agent, defaultOptions, }: NetworkRouteOptions<OUTPUT>): ReturnType<typeof registerApiRoute>;
14
42
  //# sourceMappingURL=network-route.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"network-route.d.ts","sourceRoot":"","sources":["../src/network-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAIxD,MAAM,MAAM,mBAAmB,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IACnE;IAAE,IAAI,EAAE,GAAG,MAAM,WAAW,MAAM,EAAE,CAAC;IAAC,KAAK,CAAC,EAAE,KAAK,CAAC;IAAC,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GAC9G;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,CAAC;AAE7F,wBAAgB,YAAY,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACpE,IAA0B,EAC1B,KAAK,EACL,cAAc,GACf,EAAE,mBAAmB,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAmGnE"}
1
+ {"version":3,"file":"network-route.d.ts","sourceRoot":"","sources":["../src/network-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACxE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD,OAAO,KAAK,EAAE,mBAAmB,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAGzD,MAAM,MAAM,0BAA0B,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAAI,qBAAqB,CACrG,MAAM,EACN,QAAQ,CACT,GAAG;IACF,QAAQ,EAAE,gBAAgB,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,2BAA2B,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IAAI;IACjF,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,0BAA0B,CAAC,MAAM,CAAC,CAAC;IAC3C,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;CAC1D,CAAC;AAEF;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,mBAAmB,CAAC,UAAU,SAAS,SAAS,EAAE,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EAC/G,MAAM,EACN,OAAO,EACP,MAAM,EACN,cAAc,GACf,EAAE,2BAA2B,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,CAqBhG;AAED,MAAM,MAAM,mBAAmB,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,IACnE;IAAE,IAAI,EAAE,GAAG,MAAM,WAAW,MAAM,EAAE,CAAC;IAAC,KAAK,CAAC,EAAE,KAAK,CAAC;IAAC,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;CAAE,GAC/G;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,cAAc,CAAC,EAAE,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;CAAE,CAAC;AAE9F,wBAAgB,YAAY,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAAE,EACpE,IAA0B,EAC1B,KAAK,EACL,cAAc,GACf,EAAE,mBAAmB,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CAwFnE"}
@@ -1,8 +1,9 @@
1
1
  import type { LLMStepResult } from '@mastra/core/agent';
2
2
  import type { ChunkType, DataChunkType, NetworkChunkType } from '@mastra/core/stream';
3
3
  import type { WorkflowRunStatus, WorkflowStepStatus } from '@mastra/core/workflows';
4
- import type { InferUIMessageChunk, UIMessage } from 'ai';
4
+ import type { InferUIMessageChunk, UIMessage, UIMessageStreamOptions } from 'ai';
5
5
  import type { ZodType } from 'zod';
6
+ import type { ToolAgentChunkType, ToolWorkflowChunkType, ToolNetworkChunkType } from './helpers.js';
6
7
  type LanguageModelV2Usage = {
7
8
  /**
8
9
  The number of input (prompt) tokens used.
@@ -67,20 +68,25 @@ export type AgentDataPart = {
67
68
  id: string;
68
69
  data: LLMStepResult;
69
70
  };
70
- export declare function WorkflowStreamToAISDKTransformer(): import("stream/web").TransformStream<ChunkType, WorkflowDataPart | ChunkType | {
71
+ declare const PRIMITIVE_CACHE_SYMBOL: unique symbol;
72
+ export declare function WorkflowStreamToAISDKTransformer({ includeTextStreamParts, }?: {
73
+ includeTextStreamParts?: boolean;
74
+ }): import("stream/web").TransformStream<ChunkType, ToolAgentChunkType | ToolWorkflowChunkType | ToolNetworkChunkType | WorkflowDataPart | ChunkType | {
71
75
  data?: string;
72
76
  type?: "start" | "finish";
73
- }>;
74
- export declare function AgentNetworkToAISDKTransformer(): import("stream/web").TransformStream<NetworkChunkType, DataChunkType | NetworkDataPart | {
77
+ } | InferUIMessageChunk<UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>>>;
78
+ export declare function AgentNetworkToAISDKTransformer(): import("stream/web").TransformStream<NetworkChunkType, DataChunkType | NetworkDataPart | InferUIMessageChunk<UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>> | {
75
79
  data?: string;
76
80
  type?: "start" | "finish";
77
- } | InferUIMessageChunk<UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>>>;
78
- export declare function AgentStreamToAISDKTransformer<TOutput extends ZodType<any>>({ lastMessageId, sendStart, sendFinish, sendReasoning, sendSources, }: {
81
+ }>;
82
+ export declare function AgentStreamToAISDKTransformer<TOutput extends ZodType<any>>({ lastMessageId, sendStart, sendFinish, sendReasoning, sendSources, messageMetadata, onError, }: {
79
83
  lastMessageId?: string;
80
84
  sendStart?: boolean;
81
85
  sendFinish?: boolean;
82
86
  sendReasoning?: boolean;
83
87
  sendSources?: boolean;
88
+ messageMetadata?: UIMessageStreamOptions<UIMessage>['messageMetadata'];
89
+ onError?: UIMessageStreamOptions<UIMessage>['onError'];
84
90
  }): import("stream/web").TransformStream<ChunkType<TOutput>, object>;
85
91
  export declare function transformAgent<TOutput extends ZodType<any>>(payload: ChunkType<TOutput>, bufferedSteps: Map<string, any>): {
86
92
  type: "data-tool-agent";
@@ -90,12 +96,120 @@ export declare function transformAgent<TOutput extends ZodType<any>>(payload: Ch
90
96
  export declare function transformWorkflow<TOutput extends ZodType<any>>(payload: ChunkType<TOutput>, bufferedWorkflows: Map<string, {
91
97
  name: string;
92
98
  steps: Record<string, StepResult>;
93
- }>, isNested?: boolean): (DataChunkType & {
99
+ }>, isNested?: boolean, includeTextStreamParts?: boolean): (DataChunkType & {
94
100
  from: never;
95
101
  runId: never;
96
102
  metadata?: Record<string, any> | undefined;
97
103
  payload: never;
98
- }) | {
104
+ }) | ToolAgentChunkType | ToolWorkflowChunkType | ToolNetworkChunkType | {
105
+ type: "text-start";
106
+ id: string;
107
+ providerMetadata?: import("ai").ProviderMetadata;
108
+ } | {
109
+ type: "text-delta";
110
+ delta: string;
111
+ id: string;
112
+ providerMetadata?: import("ai").ProviderMetadata;
113
+ } | {
114
+ type: "text-end";
115
+ id: string;
116
+ providerMetadata?: import("ai").ProviderMetadata;
117
+ } | {
118
+ type: "reasoning-start";
119
+ id: string;
120
+ providerMetadata?: import("ai").ProviderMetadata;
121
+ } | {
122
+ type: "reasoning-delta";
123
+ id: string;
124
+ delta: string;
125
+ providerMetadata?: import("ai").ProviderMetadata;
126
+ } | {
127
+ type: "reasoning-end";
128
+ id: string;
129
+ providerMetadata?: import("ai").ProviderMetadata;
130
+ } | {
131
+ type: "error";
132
+ errorText: string;
133
+ } | {
134
+ type: "tool-input-available";
135
+ toolCallId: string;
136
+ toolName: string;
137
+ input: unknown;
138
+ providerExecuted?: boolean;
139
+ providerMetadata?: import("ai").ProviderMetadata;
140
+ dynamic?: boolean;
141
+ } | {
142
+ type: "tool-input-error";
143
+ toolCallId: string;
144
+ toolName: string;
145
+ input: unknown;
146
+ providerExecuted?: boolean;
147
+ providerMetadata?: import("ai").ProviderMetadata;
148
+ dynamic?: boolean;
149
+ errorText: string;
150
+ } | {
151
+ type: "tool-output-available";
152
+ toolCallId: string;
153
+ output: unknown;
154
+ providerExecuted?: boolean;
155
+ dynamic?: boolean;
156
+ preliminary?: boolean;
157
+ } | {
158
+ type: "tool-output-error";
159
+ toolCallId: string;
160
+ errorText: string;
161
+ providerExecuted?: boolean;
162
+ dynamic?: boolean;
163
+ } | {
164
+ type: "tool-input-start";
165
+ toolCallId: string;
166
+ toolName: string;
167
+ providerExecuted?: boolean;
168
+ dynamic?: boolean;
169
+ } | {
170
+ type: "tool-input-delta";
171
+ toolCallId: string;
172
+ inputTextDelta: string;
173
+ } | {
174
+ type: "source-url";
175
+ sourceId: string;
176
+ url: string;
177
+ title?: string;
178
+ providerMetadata?: import("ai").ProviderMetadata;
179
+ } | {
180
+ type: "source-document";
181
+ sourceId: string;
182
+ mediaType: string;
183
+ title: string;
184
+ filename?: string;
185
+ providerMetadata?: import("ai").ProviderMetadata;
186
+ } | {
187
+ type: "file";
188
+ url: string;
189
+ mediaType: string;
190
+ providerMetadata?: import("ai").ProviderMetadata;
191
+ } | {
192
+ type: "start-step";
193
+ } | {
194
+ type: "finish-step";
195
+ } | {
196
+ type: "abort";
197
+ } | {
198
+ type: `data-${string}`;
199
+ id?: string;
200
+ data: unknown;
201
+ transient?: boolean;
202
+ } | {
203
+ type: "start";
204
+ messageId?: string;
205
+ messageMetadata?: unknown;
206
+ } | {
207
+ type: "finish";
208
+ messageMetadata?: unknown;
209
+ } | {
210
+ type: "message-metadata";
211
+ messageMetadata: unknown;
212
+ } | {
99
213
  readonly type: "data-workflow" | "data-tool-workflow";
100
214
  readonly id: string;
101
215
  readonly data: {
@@ -128,10 +242,16 @@ export declare function transformWorkflow<TOutput extends ZodType<any>>(payload:
128
242
  };
129
243
  readonly status: WorkflowRunStatus;
130
244
  };
131
- } | null;
245
+ } | null | undefined;
132
246
  export declare function transformNetwork(payload: NetworkChunkType, bufferedNetworks: Map<string, {
133
247
  name: string;
134
- steps: StepResult[];
248
+ steps: (StepResult & {
249
+ id: string;
250
+ iteration: number;
251
+ task: null | Record<string, unknown>;
252
+ input: StepResult['input'];
253
+ [PRIMITIVE_CACHE_SYMBOL]: Map<string, any>;
254
+ })[];
135
255
  usage: LanguageModelV2Usage | null;
136
256
  output: unknown | null;
137
257
  }>, isNested?: boolean): InferUIMessageChunk<UIMessage> | NetworkDataPart | DataChunkType | null;
@@ -1 +1 @@
1
- {"version":3,"file":"transformers.d.ts","sourceRoot":"","sources":["../src/transformers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACtF,OAAO,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AACpF,OAAO,KAAK,EAAE,mBAAmB,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AACzD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,KAAK,CAAC;AASnC,KAAK,oBAAoB,GAAG;IAC1B;;OAEG;IACH,WAAW,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;OAEG;IACH,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC;;;;OAIG;IACH,WAAW,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACrC;;OAEG;IACH,iBAAiB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACxC,CAAC;AAEF,KAAK,UAAU,GAAG;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,kBAAkB,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACtC,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC/C,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC/C,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,EAAE,eAAe,GAAG,oBAAoB,CAAC;IAC7C,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,iBAAiB,CAAC;QAC1B,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAClC,MAAM,EAAE;YACN,KAAK,EAAE;gBACL,WAAW,EAAE,MAAM,CAAC;gBACpB,YAAY,EAAE,MAAM,CAAC;gBACrB,WAAW,EAAE,MAAM,CAAC;aACrB,CAAC;SACH,GAAG,IAAI,CAAC;KACV,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,cAAc,GAAG,mBAAmB,CAAC;IAC3C,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,SAAS,GAAG,UAAU,CAAC;QAC/B,KAAK,EAAE,UAAU,EAAE,CAAC;QACpB,KAAK,EAAE,oBAAoB,GAAG,IAAI,CAAC;QACnC,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;KACxB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,IAAI,EAAE,iBAAiB,CAAC;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,aAAa,CAAC;CACrB,CAAC;AAEF,wBAAgB,gCAAgC;WAWjC,MAAM;WACN,OAAO,GAAG,QAAQ;GAqBhC;AAED,wBAAgB,8BAA8B;WAS/B,MAAM;WACN,OAAO,GAAG,QAAQ;6FAqBhC;AAED,wBAAgB,6BAA6B,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,CAAC,EAAE,EAC1E,aAAa,EACb,SAAS,EACT,UAAU,EACV,aAAa,EACb,WAAW,GACZ,EAAE;IACD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,oEA6DA;AAED,wBAAgB,cAAc,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,CAAC,EACzD,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,EAC3B,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC;;;;SAoJhC;AAED,wBAAgB,iBAAiB,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,CAAC,EAC5D,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,EAC3B,iBAAiB,EAAE,GAAG,CACpB,MAAM,EACN;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;CACnC,CACF,EACD,QAAQ,CAAC,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;SA2GnB;AAED,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,gBAAgB,EACzB,gBAAgB,EAAE,GAAG,CACnB,MAAM,EACN;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAAC,KAAK,EAAE,oBAAoB,GAAG,IAAI,CAAC;IAAC,MAAM,EAAE,OAAO,GAAG,IAAI,CAAA;CAAE,CAClG,EACD,QAAQ,CAAC,EAAE,OAAO,GACjB,mBAAmB,CAAC,SAAS,CAAC,GAAG,eAAe,GAAG,aAAa,GAAG,IAAI,CA2OzE"}
1
+ {"version":3,"file":"transformers.d.ts","sourceRoot":"","sources":["../src/transformers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACtF,OAAO,KAAK,EAAE,iBAAiB,EAAE,kBAAkB,EAAuB,MAAM,wBAAwB,CAAC;AACzG,OAAO,KAAK,EAAE,mBAAmB,EAA2B,SAAS,EAAE,sBAAsB,EAAE,MAAM,IAAI,CAAC;AAC1G,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,KAAK,CAAC;AAEnC,OAAO,KAAK,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AASjG,KAAK,oBAAoB,GAAG;IAC1B;;OAEG;IACH,WAAW,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;OAEG;IACH,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC;;;;OAIG;IACH,WAAW,EAAE,MAAM,GAAG,SAAS,CAAC;IAChC;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACrC;;OAEG;IACH,iBAAiB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACxC,CAAC;AAEF,KAAK,UAAU,GAAG;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,kBAAkB,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IACtC,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC/C,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC/C,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,EAAE,eAAe,GAAG,oBAAoB,CAAC;IAC7C,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,iBAAiB,CAAC;QAC1B,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAClC,MAAM,EAAE;YACN,KAAK,EAAE;gBACL,WAAW,EAAE,MAAM,CAAC;gBACpB,YAAY,EAAE,MAAM,CAAC;gBACrB,WAAW,EAAE,MAAM,CAAC;aACrB,CAAC;SACH,GAAG,IAAI,CAAC;KACV,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,cAAc,GAAG,mBAAmB,CAAC;IAC3C,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACJ,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,SAAS,GAAG,UAAU,CAAC;QAC/B,KAAK,EAAE,UAAU,EAAE,CAAC;QACpB,KAAK,EAAE,oBAAoB,GAAG,IAAI,CAAC;QACnC,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;KACxB,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,IAAI,EAAE,iBAAiB,CAAC;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,aAAa,CAAC;CACrB,CAAC;AAGF,QAAA,MAAM,sBAAsB,eAA4B,CAAC;AAEzD,wBAAgB,gCAAgC,CAAC,EAC/C,sBAAsB,GACvB,GAAE;IAAE,sBAAsB,CAAC,EAAE,OAAO,CAAA;CAAO;WAW7B,MAAM;WACN,OAAO,GAAG,QAAQ;6FAwBhC;AAED,wBAAgB,8BAA8B;WAoB/B,MAAM;WACN,OAAO,GAAG,QAAQ;GAqBhC;AAED,wBAAgB,6BAA6B,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,CAAC,EAAE,EAC1E,aAAa,EACb,SAAS,EACT,UAAU,EACV,aAAa,EACb,WAAW,EACX,eAAe,EACf,OAAO,GACR,EAAE;IACD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,eAAe,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,CAAC;IACvE,OAAO,CAAC,EAAE,sBAAsB,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,CAAC;CACxD,oEA8DA;AAED,wBAAgB,cAAc,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,CAAC,EACzD,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,EAC3B,aAAa,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC;;;;SAoJhC;AAED,wBAAgB,iBAAiB,CAAC,OAAO,SAAS,OAAO,CAAC,GAAG,CAAC,EAC5D,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,EAC3B,iBAAiB,EAAE,GAAG,CACpB,MAAM,EACN;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;CACnC,CACF,EACD,QAAQ,CAAC,EAAE,OAAO,EAClB,sBAAsB,CAAC,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qBAqIjC;AAED,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,gBAAgB,EACzB,gBAAgB,EAAE,GAAG,CACnB,MAAM,EACN;IACE,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,CAAC,UAAU,GAAG;QACnB,EAAE,EAAE,MAAM,CAAC;QACX,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACrC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC,sBAAsB,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC5C,CAAC,EAAE,CAAC;IACL,KAAK,EAAE,oBAAoB,GAAG,IAAI,CAAC;IACnC,MAAM,EAAE,OAAO,GAAG,IAAI,CAAC;CACxB,CACF,EACD,QAAQ,CAAC,EAAE,OAAO,GACjB,mBAAmB,CAAC,SAAS,CAAC,GAAG,eAAe,GAAG,aAAa,GAAG,IAAI,CA0VzE"}
package/dist/utils.d.ts CHANGED
@@ -1,5 +1,6 @@
1
- import type { DataChunkType, NetworkChunkType } from '@mastra/core/stream';
1
+ import type { ChunkType, DataChunkType, NetworkChunkType, OutputSchema } from '@mastra/core/stream';
2
2
  export declare const isDataChunkType: (chunk: any) => chunk is DataChunkType;
3
+ export declare const isMastraTextStreamChunk: (chunk: any) => chunk is ChunkType<OutputSchema>;
3
4
  export declare function safeParseErrorObject(obj: unknown): string;
4
5
  export declare const isAgentExecutionDataChunkType: (chunk: any) => chunk is Omit<NetworkChunkType, "payload"> & {
5
6
  payload: DataChunkType;
@@ -1 +1 @@
1
- {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE3E,eAAO,MAAM,eAAe,GAAI,OAAO,GAAG,KAAG,KAAK,IAAI,aAErD,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM,CAgBzD;AAED,eAAO,MAAM,6BAA6B,GACxC,OAAO,GAAG,KACT,KAAK,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,CAAC,GAAG;IAAE,OAAO,EAAE,aAAa,CAAA;CAWvE,CAAC;AAEF,eAAO,MAAM,gCAAgC,GAC3C,OAAO,GAAG,KACT,KAAK,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,CAAC,GAAG;IAAE,OAAO,EAAE,aAAa,CAAA;CAWvE,CAAC"}
1
+ {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEpG,eAAO,MAAM,eAAe,GAAI,OAAO,GAAG,KAAG,KAAK,IAAI,aAErD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAAI,OAAO,GAAG,KAAG,KAAK,IAAI,SAAS,CAAC,YAAY,CA8BnF,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM,CAgBzD;AAED,eAAO,MAAM,6BAA6B,GACxC,OAAO,GAAG,KACT,KAAK,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,CAAC,GAAG;IAAE,OAAO,EAAE,aAAa,CAAA;CAWvE,CAAC;AAEF,eAAO,MAAM,gCAAgC,GAC3C,OAAO,GAAG,KACT,KAAK,IAAI,IAAI,CAAC,gBAAgB,EAAE,SAAS,CAAC,GAAG;IAAE,OAAO,EAAE,aAAa,CAAA;CAWvE,CAAC"}
@@ -1,10 +1,47 @@
1
+ import type { Mastra } from '@mastra/core/mastra';
2
+ import type { TracingOptions } from '@mastra/core/observability';
3
+ import type { RequestContext } from '@mastra/core/request-context';
1
4
  import { registerApiRoute } from '@mastra/core/server';
5
+ import type { InferUIMessageChunk, UIMessage } from 'ai';
6
+ export type WorkflowStreamHandlerParams = {
7
+ runId?: string;
8
+ resourceId?: string;
9
+ inputData?: Record<string, any>;
10
+ resumeData?: Record<string, any>;
11
+ requestContext?: RequestContext;
12
+ tracingOptions?: TracingOptions;
13
+ step?: string;
14
+ };
15
+ export type WorkflowStreamHandlerOptions = {
16
+ mastra: Mastra;
17
+ workflowId: string;
18
+ params: WorkflowStreamHandlerParams;
19
+ includeTextStreamParts?: boolean;
20
+ };
21
+ /**
22
+ * Framework-agnostic handler for streaming workflow execution in AI SDK format.
23
+ * Use this function directly when you need to handle workflow streaming outside of Hono/registerApiRoute.
24
+ *
25
+ * @example
26
+ * // Next.js App Router
27
+ * export async function POST(req: Request) {
28
+ * const params = await req.json();
29
+ * return handleWorkflowStream({
30
+ * mastra,
31
+ * workflowId: 'my-workflow',
32
+ * params,
33
+ * });
34
+ * }
35
+ */
36
+ export declare function handleWorkflowStream<UI_MESSAGE extends UIMessage>({ mastra, workflowId, params, includeTextStreamParts, }: WorkflowStreamHandlerOptions): Promise<ReadableStream<InferUIMessageChunk<UI_MESSAGE>>>;
2
37
  export type WorkflowRouteOptions = {
3
38
  path: `${string}:workflowId${string}`;
4
39
  workflow?: never;
40
+ includeTextStreamParts?: boolean;
5
41
  } | {
6
42
  path: string;
7
43
  workflow: string;
44
+ includeTextStreamParts?: boolean;
8
45
  };
9
- export declare function workflowRoute({ path, workflow, }: WorkflowRouteOptions): ReturnType<typeof registerApiRoute>;
46
+ export declare function workflowRoute({ path, workflow, includeTextStreamParts, }: WorkflowRouteOptions): ReturnType<typeof registerApiRoute>;
10
47
  //# sourceMappingURL=workflow-route.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"workflow-route.d.ts","sourceRoot":"","sources":["../src/workflow-route.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAcvD,MAAM,MAAM,oBAAoB,GAC5B;IAAE,IAAI,EAAE,GAAG,MAAM,cAAc,MAAM,EAAE,CAAC;IAAC,QAAQ,CAAC,EAAE,KAAK,CAAA;CAAE,GAC3D;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAAC;AAEvC,wBAAgB,aAAa,CAAC,EAC5B,IAA0C,EAC1C,QAAQ,GACT,EAAE,oBAAoB,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CA2F5D"}
1
+ {"version":3,"file":"workflow-route.d.ts","sourceRoot":"","sources":["../src/workflow-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,KAAK,EAAE,mBAAmB,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAGzD,MAAM,MAAM,2BAA2B,GAAG;IACxC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAChC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACjC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,2BAA2B,CAAC;IACpC,sBAAsB,CAAC,EAAE,OAAO,CAAC;CAClC,CAAC;AAEF;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,oBAAoB,CAAC,UAAU,SAAS,SAAS,EAAE,EACvE,MAAM,EACN,UAAU,EACV,MAAM,EACN,sBAA6B,GAC9B,EAAE,4BAA4B,GAAG,OAAO,CAAC,cAAc,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,CAqBzF;AAED,MAAM,MAAM,oBAAoB,GAC5B;IAAE,IAAI,EAAE,GAAG,MAAM,cAAc,MAAM,EAAE,CAAC;IAAC,QAAQ,CAAC,EAAE,KAAK,CAAC;IAAC,sBAAsB,CAAC,EAAE,OAAO,CAAA;CAAE,GAC7F;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,sBAAsB,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAEzE,wBAAgB,aAAa,CAAC,EAC5B,IAA0C,EAC1C,QAAQ,EACR,sBAA6B,GAC9B,EAAE,oBAAoB,GAAG,UAAU,CAAC,OAAO,gBAAgB,CAAC,CA6F5D"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@mastra/ai-sdk",
3
- "version": "1.0.0-beta.3",
3
+ "version": "1.0.0-beta.5",
4
4
  "description": "Adds custom API routes to be compatible with the AI SDK UI parts",
5
5
  "type": "module",
6
6
  "main": "dist/index.js",
@@ -37,16 +37,16 @@
37
37
  },
38
38
  "devDependencies": {
39
39
  "@types/json-schema": "^7.0.15",
40
- "@vitest/coverage-v8": "4.0.8",
41
- "@vitest/ui": "4.0.8",
40
+ "@vitest/coverage-v8": "4.0.12",
41
+ "@vitest/ui": "4.0.12",
42
42
  "eslint": "^9.37.0",
43
43
  "tsup": "^8.5.0",
44
44
  "typescript": "^5.8.3",
45
- "vitest": "^4.0.8",
45
+ "vitest": "4.0.12",
46
46
  "zod": "^3.25.76",
47
- "@internal/types-builder": "0.0.28",
48
- "@mastra/core": "1.0.0-beta.4",
49
- "@internal/lint": "0.0.53"
47
+ "@internal/lint": "0.0.53",
48
+ "@mastra/core": "1.0.0-beta.6",
49
+ "@internal/types-builder": "0.0.28"
50
50
  },
51
51
  "keywords": [
52
52
  "mastra",