@mastra/client-js 0.0.0-fix-memory-xxhash-20250409202110 → 0.0.0-generate-message-id-20250512171942
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.
- package/CHANGELOG.md +445 -2
- package/{LICENSE → LICENSE.md} +3 -1
- package/dist/index.cjs +569 -15
- package/dist/index.d.cts +213 -10
- package/dist/index.d.ts +213 -10
- package/dist/index.js +565 -15
- package/package.json +12 -7
- package/src/adapters/agui.test.ts +180 -0
- package/src/adapters/agui.ts +239 -0
- package/src/client.ts +56 -11
- package/src/index.test.ts +4 -4
- package/src/resources/a2a.ts +88 -0
- package/src/resources/agent.ts +26 -35
- package/src/resources/base.ts +1 -1
- package/src/resources/index.ts +2 -0
- package/src/resources/memory-thread.ts +1 -8
- package/src/resources/network.ts +6 -12
- package/src/resources/tool.ts +15 -3
- package/src/resources/vnext-workflow.ts +261 -0
- package/src/resources/workflow.ts +38 -2
- package/src/types.ts +43 -5
- package/src/utils/zod-to-json-schema.ts +10 -0
package/src/resources/agent.ts
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
1
|
+
import { processDataStream } from '@ai-sdk/ui-utils';
|
|
1
2
|
import type { GenerateReturn } from '@mastra/core';
|
|
2
3
|
import type { JSONSchema7 } from 'json-schema';
|
|
3
4
|
import { ZodSchema } from 'zod';
|
|
4
|
-
import { zodToJsonSchema } from 'zod-to-json-schema';
|
|
5
|
-
import { processDataStream } from '@ai-sdk/ui-utils';
|
|
5
|
+
import { zodToJsonSchema } from '../utils/zod-to-json-schema';
|
|
6
6
|
|
|
7
7
|
import type {
|
|
8
8
|
GenerateParams,
|
|
@@ -14,29 +14,7 @@ import type {
|
|
|
14
14
|
} from '../types';
|
|
15
15
|
|
|
16
16
|
import { BaseResource } from './base';
|
|
17
|
-
|
|
18
|
-
export class AgentTool extends BaseResource {
|
|
19
|
-
constructor(
|
|
20
|
-
options: ClientOptions,
|
|
21
|
-
private agentId: string,
|
|
22
|
-
private toolId: string,
|
|
23
|
-
) {
|
|
24
|
-
super(options);
|
|
25
|
-
}
|
|
26
|
-
|
|
27
|
-
/**
|
|
28
|
-
* Executes a specific tool for an agent
|
|
29
|
-
* @param params - Parameters required for tool execution
|
|
30
|
-
* @returns Promise containing tool execution results
|
|
31
|
-
*/
|
|
32
|
-
/** @deprecated use CreateRun/startRun */
|
|
33
|
-
execute(params: { data: any }): Promise<any> {
|
|
34
|
-
return this.request(`/api/agents/${this.agentId}/tools/${this.toolId}/execute`, {
|
|
35
|
-
method: 'POST',
|
|
36
|
-
body: params,
|
|
37
|
-
});
|
|
38
|
-
}
|
|
39
|
-
}
|
|
17
|
+
import type { RuntimeContext } from '@mastra/core/di';
|
|
40
18
|
|
|
41
19
|
export class AgentVoice extends BaseResource {
|
|
42
20
|
constructor(
|
|
@@ -122,11 +100,9 @@ export class Agent extends BaseResource {
|
|
|
122
100
|
): Promise<GenerateReturn<T>> {
|
|
123
101
|
const processedParams = {
|
|
124
102
|
...params,
|
|
125
|
-
output:
|
|
126
|
-
experimental_output:
|
|
127
|
-
|
|
128
|
-
? zodToJsonSchema(params.experimental_output)
|
|
129
|
-
: params.experimental_output,
|
|
103
|
+
output: zodToJsonSchema(params.output),
|
|
104
|
+
experimental_output: zodToJsonSchema(params.experimental_output),
|
|
105
|
+
runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
|
|
130
106
|
};
|
|
131
107
|
|
|
132
108
|
return this.request(`/api/agents/${this.agentId}/generate`, {
|
|
@@ -149,11 +125,9 @@ export class Agent extends BaseResource {
|
|
|
149
125
|
> {
|
|
150
126
|
const processedParams = {
|
|
151
127
|
...params,
|
|
152
|
-
output:
|
|
153
|
-
experimental_output:
|
|
154
|
-
|
|
155
|
-
? zodToJsonSchema(params.experimental_output)
|
|
156
|
-
: params.experimental_output,
|
|
128
|
+
output: zodToJsonSchema(params.output),
|
|
129
|
+
experimental_output: zodToJsonSchema(params.experimental_output),
|
|
130
|
+
runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
|
|
157
131
|
};
|
|
158
132
|
|
|
159
133
|
const response: Response & {
|
|
@@ -187,6 +161,23 @@ export class Agent extends BaseResource {
|
|
|
187
161
|
return this.request(`/api/agents/${this.agentId}/tools/${toolId}`);
|
|
188
162
|
}
|
|
189
163
|
|
|
164
|
+
/**
|
|
165
|
+
* Executes a tool for the agent
|
|
166
|
+
* @param toolId - ID of the tool to execute
|
|
167
|
+
* @param params - Parameters required for tool execution
|
|
168
|
+
* @returns Promise containing the tool execution results
|
|
169
|
+
*/
|
|
170
|
+
executeTool(toolId: string, params: { data: any; runtimeContext?: RuntimeContext }): Promise<any> {
|
|
171
|
+
const body = {
|
|
172
|
+
data: params.data,
|
|
173
|
+
runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
|
|
174
|
+
};
|
|
175
|
+
return this.request(`/api/agents/${this.agentId}/tools/${toolId}/execute`, {
|
|
176
|
+
method: 'POST',
|
|
177
|
+
body,
|
|
178
|
+
});
|
|
179
|
+
}
|
|
180
|
+
|
|
190
181
|
/**
|
|
191
182
|
* Retrieves evaluation results for the agent
|
|
192
183
|
* @returns Promise containing agent evaluations
|
package/src/resources/base.ts
CHANGED
package/src/resources/index.ts
CHANGED
|
@@ -1,13 +1,6 @@
|
|
|
1
1
|
import type { StorageThreadType } from '@mastra/core';
|
|
2
2
|
|
|
3
|
-
import type {
|
|
4
|
-
CreateMemoryThreadParams,
|
|
5
|
-
GetMemoryThreadMessagesResponse,
|
|
6
|
-
GetMemoryThreadResponse,
|
|
7
|
-
ClientOptions,
|
|
8
|
-
SaveMessageToMemoryParams,
|
|
9
|
-
UpdateMemoryThreadParams,
|
|
10
|
-
} from '../types';
|
|
3
|
+
import type { GetMemoryThreadMessagesResponse, ClientOptions, UpdateMemoryThreadParams } from '../types';
|
|
11
4
|
|
|
12
5
|
import { BaseResource } from './base';
|
|
13
6
|
|
package/src/resources/network.ts
CHANGED
|
@@ -1,12 +1,12 @@
|
|
|
1
|
+
import { processDataStream } from '@ai-sdk/ui-utils';
|
|
1
2
|
import type { GenerateReturn } from '@mastra/core';
|
|
2
3
|
import type { JSONSchema7 } from 'json-schema';
|
|
3
4
|
import { ZodSchema } from 'zod';
|
|
4
|
-
import { zodToJsonSchema } from 'zod-to-json-schema';
|
|
5
|
+
import { zodToJsonSchema } from '../utils/zod-to-json-schema';
|
|
5
6
|
|
|
6
7
|
import type { GenerateParams, ClientOptions, StreamParams, GetNetworkResponse } from '../types';
|
|
7
8
|
|
|
8
9
|
import { BaseResource } from './base';
|
|
9
|
-
import { processDataStream } from '@ai-sdk/ui-utils';
|
|
10
10
|
|
|
11
11
|
export class Network extends BaseResource {
|
|
12
12
|
constructor(
|
|
@@ -34,11 +34,8 @@ export class Network extends BaseResource {
|
|
|
34
34
|
): Promise<GenerateReturn<T>> {
|
|
35
35
|
const processedParams = {
|
|
36
36
|
...params,
|
|
37
|
-
output:
|
|
38
|
-
experimental_output:
|
|
39
|
-
params.experimental_output instanceof ZodSchema
|
|
40
|
-
? zodToJsonSchema(params.experimental_output)
|
|
41
|
-
: params.experimental_output,
|
|
37
|
+
output: zodToJsonSchema(params.output),
|
|
38
|
+
experimental_output: zodToJsonSchema(params.experimental_output),
|
|
42
39
|
};
|
|
43
40
|
|
|
44
41
|
return this.request(`/api/networks/${this.networkId}/generate`, {
|
|
@@ -61,11 +58,8 @@ export class Network extends BaseResource {
|
|
|
61
58
|
> {
|
|
62
59
|
const processedParams = {
|
|
63
60
|
...params,
|
|
64
|
-
output:
|
|
65
|
-
experimental_output:
|
|
66
|
-
params.experimental_output instanceof ZodSchema
|
|
67
|
-
? zodToJsonSchema(params.experimental_output)
|
|
68
|
-
: params.experimental_output,
|
|
61
|
+
output: zodToJsonSchema(params.output),
|
|
62
|
+
experimental_output: zodToJsonSchema(params.experimental_output),
|
|
69
63
|
};
|
|
70
64
|
|
|
71
65
|
const response: Response & {
|
package/src/resources/tool.ts
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import type { RuntimeContext } from '@mastra/core/di';
|
|
1
2
|
import type { GetToolResponse, ClientOptions } from '../types';
|
|
2
3
|
|
|
3
4
|
import { BaseResource } from './base';
|
|
@@ -23,10 +24,21 @@ export class Tool extends BaseResource {
|
|
|
23
24
|
* @param params - Parameters required for tool execution
|
|
24
25
|
* @returns Promise containing the tool execution results
|
|
25
26
|
*/
|
|
26
|
-
execute(params: { data: any }): Promise<any> {
|
|
27
|
-
|
|
27
|
+
execute(params: { data: any; runId?: string; runtimeContext?: RuntimeContext }): Promise<any> {
|
|
28
|
+
const url = new URLSearchParams();
|
|
29
|
+
|
|
30
|
+
if (params.runId) {
|
|
31
|
+
url.set('runId', params.runId);
|
|
32
|
+
}
|
|
33
|
+
|
|
34
|
+
const body = {
|
|
35
|
+
data: params.data,
|
|
36
|
+
runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
|
|
37
|
+
};
|
|
38
|
+
|
|
39
|
+
return this.request(`/api/tools/${this.toolId}/execute?${url.toString()}`, {
|
|
28
40
|
method: 'POST',
|
|
29
|
-
body
|
|
41
|
+
body,
|
|
30
42
|
});
|
|
31
43
|
}
|
|
32
44
|
}
|
|
@@ -0,0 +1,261 @@
|
|
|
1
|
+
import { RuntimeContext } from '@mastra/core/runtime-context';
|
|
2
|
+
import type {
|
|
3
|
+
ClientOptions,
|
|
4
|
+
GetVNextWorkflowResponse,
|
|
5
|
+
GetWorkflowRunsParams,
|
|
6
|
+
GetWorkflowRunsResponse,
|
|
7
|
+
VNextWorkflowRunResult,
|
|
8
|
+
VNextWorkflowWatchResult,
|
|
9
|
+
} from '../types';
|
|
10
|
+
|
|
11
|
+
import { BaseResource } from './base';
|
|
12
|
+
|
|
13
|
+
const RECORD_SEPARATOR = '\x1E';
|
|
14
|
+
|
|
15
|
+
export class VNextWorkflow extends BaseResource {
|
|
16
|
+
constructor(
|
|
17
|
+
options: ClientOptions,
|
|
18
|
+
private workflowId: string,
|
|
19
|
+
) {
|
|
20
|
+
super(options);
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
/**
|
|
24
|
+
* Creates an async generator that processes a readable stream and yields vNext workflow records
|
|
25
|
+
* separated by the Record Separator character (\x1E)
|
|
26
|
+
*
|
|
27
|
+
* @param stream - The readable stream to process
|
|
28
|
+
* @returns An async generator that yields parsed records
|
|
29
|
+
*/
|
|
30
|
+
private async *streamProcessor(stream: ReadableStream): AsyncGenerator<VNextWorkflowWatchResult, void, unknown> {
|
|
31
|
+
const reader = stream.getReader();
|
|
32
|
+
|
|
33
|
+
// Track if we've finished reading from the stream
|
|
34
|
+
let doneReading = false;
|
|
35
|
+
// Buffer to accumulate partial chunks
|
|
36
|
+
let buffer = '';
|
|
37
|
+
|
|
38
|
+
try {
|
|
39
|
+
while (!doneReading) {
|
|
40
|
+
// Read the next chunk from the stream
|
|
41
|
+
const { done, value } = await reader.read();
|
|
42
|
+
doneReading = done;
|
|
43
|
+
|
|
44
|
+
// Skip processing if we're done and there's no value
|
|
45
|
+
if (done && !value) continue;
|
|
46
|
+
|
|
47
|
+
try {
|
|
48
|
+
// Decode binary data to text
|
|
49
|
+
const decoded = value ? new TextDecoder().decode(value) : '';
|
|
50
|
+
|
|
51
|
+
// Split the combined buffer and new data by record separator
|
|
52
|
+
const chunks = (buffer + decoded).split(RECORD_SEPARATOR);
|
|
53
|
+
|
|
54
|
+
// The last chunk might be incomplete, so save it for the next iteration
|
|
55
|
+
buffer = chunks.pop() || '';
|
|
56
|
+
|
|
57
|
+
// Process complete chunks
|
|
58
|
+
for (const chunk of chunks) {
|
|
59
|
+
if (chunk) {
|
|
60
|
+
// Only process non-empty chunks
|
|
61
|
+
if (typeof chunk === 'string') {
|
|
62
|
+
try {
|
|
63
|
+
const parsedChunk = JSON.parse(chunk);
|
|
64
|
+
yield parsedChunk;
|
|
65
|
+
} catch {
|
|
66
|
+
// Silently ignore parsing errors to maintain stream processing
|
|
67
|
+
// This allows the stream to continue even if one record is malformed
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
} catch {
|
|
73
|
+
// Silently ignore parsing errors to maintain stream processing
|
|
74
|
+
// This allows the stream to continue even if one record is malformed
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
// Process any remaining data in the buffer after stream is done
|
|
79
|
+
if (buffer) {
|
|
80
|
+
try {
|
|
81
|
+
yield JSON.parse(buffer);
|
|
82
|
+
} catch {
|
|
83
|
+
// Ignore parsing error for final chunk
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
} finally {
|
|
87
|
+
// Always ensure we clean up the reader
|
|
88
|
+
reader.cancel().catch(() => {
|
|
89
|
+
// Ignore cancel errors
|
|
90
|
+
});
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
|
|
94
|
+
/**
|
|
95
|
+
* Retrieves details about the vNext workflow
|
|
96
|
+
* @returns Promise containing vNext workflow details including steps and graphs
|
|
97
|
+
*/
|
|
98
|
+
details(): Promise<GetVNextWorkflowResponse> {
|
|
99
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}`);
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
/**
|
|
103
|
+
* Retrieves all runs for a vNext workflow
|
|
104
|
+
* @param params - Parameters for filtering runs
|
|
105
|
+
* @returns Promise containing vNext workflow runs array
|
|
106
|
+
*/
|
|
107
|
+
runs(params?: GetWorkflowRunsParams): Promise<GetWorkflowRunsResponse> {
|
|
108
|
+
const searchParams = new URLSearchParams();
|
|
109
|
+
if (params?.fromDate) {
|
|
110
|
+
searchParams.set('fromDate', params.fromDate.toISOString());
|
|
111
|
+
}
|
|
112
|
+
if (params?.toDate) {
|
|
113
|
+
searchParams.set('toDate', params.toDate.toISOString());
|
|
114
|
+
}
|
|
115
|
+
if (params?.limit) {
|
|
116
|
+
searchParams.set('limit', String(params.limit));
|
|
117
|
+
}
|
|
118
|
+
if (params?.offset) {
|
|
119
|
+
searchParams.set('offset', String(params.offset));
|
|
120
|
+
}
|
|
121
|
+
if (params?.resourceId) {
|
|
122
|
+
searchParams.set('resourceId', params.resourceId);
|
|
123
|
+
}
|
|
124
|
+
|
|
125
|
+
if (searchParams.size) {
|
|
126
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/runs?${searchParams}`);
|
|
127
|
+
} else {
|
|
128
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/runs`);
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
/**
|
|
133
|
+
* Creates a new vNext workflow run
|
|
134
|
+
* @param params - Optional object containing the optional runId
|
|
135
|
+
* @returns Promise containing the runId of the created run
|
|
136
|
+
*/
|
|
137
|
+
createRun(params?: { runId?: string }): Promise<{ runId: string }> {
|
|
138
|
+
const searchParams = new URLSearchParams();
|
|
139
|
+
|
|
140
|
+
if (!!params?.runId) {
|
|
141
|
+
searchParams.set('runId', params.runId);
|
|
142
|
+
}
|
|
143
|
+
|
|
144
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/create-run?${searchParams.toString()}`, {
|
|
145
|
+
method: 'POST',
|
|
146
|
+
});
|
|
147
|
+
}
|
|
148
|
+
|
|
149
|
+
/**
|
|
150
|
+
* Starts a vNext workflow run synchronously without waiting for the workflow to complete
|
|
151
|
+
* @param params - Object containing the runId, inputData and runtimeContext
|
|
152
|
+
* @returns Promise containing success message
|
|
153
|
+
*/
|
|
154
|
+
start(params: {
|
|
155
|
+
runId: string;
|
|
156
|
+
inputData: Record<string, any>;
|
|
157
|
+
runtimeContext?: RuntimeContext;
|
|
158
|
+
}): Promise<{ message: string }> {
|
|
159
|
+
const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined;
|
|
160
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/start?runId=${params.runId}`, {
|
|
161
|
+
method: 'POST',
|
|
162
|
+
body: { inputData: params?.inputData, runtimeContext },
|
|
163
|
+
});
|
|
164
|
+
}
|
|
165
|
+
|
|
166
|
+
/**
|
|
167
|
+
* Resumes a suspended vNext workflow step synchronously without waiting for the vNext workflow to complete
|
|
168
|
+
* @param params - Object containing the runId, step, resumeData and runtimeContext
|
|
169
|
+
* @returns Promise containing success message
|
|
170
|
+
*/
|
|
171
|
+
resume({
|
|
172
|
+
step,
|
|
173
|
+
runId,
|
|
174
|
+
resumeData,
|
|
175
|
+
...rest
|
|
176
|
+
}: {
|
|
177
|
+
step: string | string[];
|
|
178
|
+
runId: string;
|
|
179
|
+
resumeData?: Record<string, any>;
|
|
180
|
+
runtimeContext?: RuntimeContext;
|
|
181
|
+
}): Promise<{ message: string }> {
|
|
182
|
+
const runtimeContext = rest.runtimeContext ? Object.fromEntries(rest.runtimeContext.entries()) : undefined;
|
|
183
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/resume?runId=${runId}`, {
|
|
184
|
+
method: 'POST',
|
|
185
|
+
stream: true,
|
|
186
|
+
body: {
|
|
187
|
+
step,
|
|
188
|
+
resumeData,
|
|
189
|
+
runtimeContext,
|
|
190
|
+
},
|
|
191
|
+
});
|
|
192
|
+
}
|
|
193
|
+
|
|
194
|
+
/**
|
|
195
|
+
* Starts a vNext workflow run asynchronously and returns a promise that resolves when the vNext workflow is complete
|
|
196
|
+
* @param params - Object containing the optional runId, inputData and runtimeContext
|
|
197
|
+
* @returns Promise containing the vNext workflow execution results
|
|
198
|
+
*/
|
|
199
|
+
startAsync(params: {
|
|
200
|
+
runId?: string;
|
|
201
|
+
inputData: Record<string, any>;
|
|
202
|
+
runtimeContext?: RuntimeContext;
|
|
203
|
+
}): Promise<VNextWorkflowRunResult> {
|
|
204
|
+
const searchParams = new URLSearchParams();
|
|
205
|
+
|
|
206
|
+
if (!!params?.runId) {
|
|
207
|
+
searchParams.set('runId', params.runId);
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined;
|
|
211
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/start-async?${searchParams.toString()}`, {
|
|
212
|
+
method: 'POST',
|
|
213
|
+
body: { inputData: params.inputData, runtimeContext },
|
|
214
|
+
});
|
|
215
|
+
}
|
|
216
|
+
|
|
217
|
+
/**
|
|
218
|
+
* Resumes a suspended vNext workflow step asynchronously and returns a promise that resolves when the vNext workflow is complete
|
|
219
|
+
* @param params - Object containing the runId, step, resumeData and runtimeContext
|
|
220
|
+
* @returns Promise containing the vNext workflow resume results
|
|
221
|
+
*/
|
|
222
|
+
resumeAsync(params: {
|
|
223
|
+
runId: string;
|
|
224
|
+
step: string | string[];
|
|
225
|
+
resumeData?: Record<string, any>;
|
|
226
|
+
runtimeContext?: RuntimeContext;
|
|
227
|
+
}): Promise<VNextWorkflowRunResult> {
|
|
228
|
+
const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined;
|
|
229
|
+
return this.request(`/api/workflows/v-next/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
230
|
+
method: 'POST',
|
|
231
|
+
body: {
|
|
232
|
+
step: params.step,
|
|
233
|
+
resumeData: params.resumeData,
|
|
234
|
+
runtimeContext,
|
|
235
|
+
},
|
|
236
|
+
});
|
|
237
|
+
}
|
|
238
|
+
|
|
239
|
+
/**
|
|
240
|
+
* Watches vNext workflow transitions in real-time
|
|
241
|
+
* @param runId - Optional run ID to filter the watch stream
|
|
242
|
+
* @returns AsyncGenerator that yields parsed records from the vNext workflow watch stream
|
|
243
|
+
*/
|
|
244
|
+
async watch({ runId }: { runId?: string }, onRecord: (record: VNextWorkflowWatchResult) => void) {
|
|
245
|
+
const response: Response = await this.request(`/api/workflows/v-next/${this.workflowId}/watch?runId=${runId}`, {
|
|
246
|
+
stream: true,
|
|
247
|
+
});
|
|
248
|
+
|
|
249
|
+
if (!response.ok) {
|
|
250
|
+
throw new Error(`Failed to watch vNext workflow: ${response.statusText}`);
|
|
251
|
+
}
|
|
252
|
+
|
|
253
|
+
if (!response.body) {
|
|
254
|
+
throw new Error('Response body is null');
|
|
255
|
+
}
|
|
256
|
+
|
|
257
|
+
for await (const record of this.streamProcessor(response.body)) {
|
|
258
|
+
onRecord(record);
|
|
259
|
+
}
|
|
260
|
+
}
|
|
261
|
+
}
|
|
@@ -1,4 +1,10 @@
|
|
|
1
|
-
import type {
|
|
1
|
+
import type {
|
|
2
|
+
GetWorkflowResponse,
|
|
3
|
+
ClientOptions,
|
|
4
|
+
WorkflowRunResult,
|
|
5
|
+
GetWorkflowRunsResponse,
|
|
6
|
+
GetWorkflowRunsParams,
|
|
7
|
+
} from '../types';
|
|
2
8
|
|
|
3
9
|
import { BaseResource } from './base';
|
|
4
10
|
|
|
@@ -20,6 +26,36 @@ export class Workflow extends BaseResource {
|
|
|
20
26
|
return this.request(`/api/workflows/${this.workflowId}`);
|
|
21
27
|
}
|
|
22
28
|
|
|
29
|
+
/**
|
|
30
|
+
* Retrieves all runs for a workflow
|
|
31
|
+
* @param params - Parameters for filtering runs
|
|
32
|
+
* @returns Promise containing workflow runs array
|
|
33
|
+
*/
|
|
34
|
+
runs(params?: GetWorkflowRunsParams): Promise<GetWorkflowRunsResponse> {
|
|
35
|
+
const searchParams = new URLSearchParams();
|
|
36
|
+
if (params?.fromDate) {
|
|
37
|
+
searchParams.set('fromDate', params.fromDate.toISOString());
|
|
38
|
+
}
|
|
39
|
+
if (params?.toDate) {
|
|
40
|
+
searchParams.set('toDate', params.toDate.toISOString());
|
|
41
|
+
}
|
|
42
|
+
if (params?.limit) {
|
|
43
|
+
searchParams.set('limit', String(params.limit));
|
|
44
|
+
}
|
|
45
|
+
if (params?.offset) {
|
|
46
|
+
searchParams.set('offset', String(params.offset));
|
|
47
|
+
}
|
|
48
|
+
if (params?.resourceId) {
|
|
49
|
+
searchParams.set('resourceId', params.resourceId);
|
|
50
|
+
}
|
|
51
|
+
|
|
52
|
+
if (searchParams.size) {
|
|
53
|
+
return this.request(`/api/workflows/${this.workflowId}/runs?${searchParams}`);
|
|
54
|
+
} else {
|
|
55
|
+
return this.request(`/api/workflows/${this.workflowId}/runs`);
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
|
|
23
59
|
/**
|
|
24
60
|
* @deprecated Use `startAsync` instead
|
|
25
61
|
* Executes the workflow with the provided parameters
|
|
@@ -168,7 +204,7 @@ export class Workflow extends BaseResource {
|
|
|
168
204
|
}
|
|
169
205
|
}
|
|
170
206
|
}
|
|
171
|
-
} catch
|
|
207
|
+
} catch {
|
|
172
208
|
// Silently ignore parsing errors to maintain stream processing
|
|
173
209
|
// This allows the stream to continue even if one record is malformed
|
|
174
210
|
}
|
package/src/types.ts
CHANGED
|
@@ -8,9 +8,11 @@ import type {
|
|
|
8
8
|
StorageThreadType,
|
|
9
9
|
BaseLogMessage,
|
|
10
10
|
WorkflowRunResult as CoreWorkflowRunResult,
|
|
11
|
+
WorkflowRuns,
|
|
11
12
|
} from '@mastra/core';
|
|
12
13
|
|
|
13
14
|
import type { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
|
|
15
|
+
import type { NewWorkflow, WatchEvent, WorkflowResult as VNextWorkflowResult } from '@mastra/core/workflows/vNext';
|
|
14
16
|
import type { JSONSchema7 } from 'json-schema';
|
|
15
17
|
import type { ZodSchema } from 'zod';
|
|
16
18
|
|
|
@@ -46,14 +48,17 @@ export interface GetAgentResponse {
|
|
|
46
48
|
|
|
47
49
|
export type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
|
|
48
50
|
messages: string | string[] | CoreMessage[] | AiMessageType[];
|
|
49
|
-
} & Partial<AgentGenerateOptions<T>>;
|
|
51
|
+
} & Partial<Omit<AgentGenerateOptions<T>, 'experimental_generateMessageId'>>;
|
|
50
52
|
|
|
51
53
|
export type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
|
|
52
54
|
messages: string | string[] | CoreMessage[] | AiMessageType[];
|
|
53
|
-
} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
|
|
55
|
+
} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry' | 'experimental_generateMessageId'>;
|
|
54
56
|
|
|
55
57
|
export interface GetEvalsByAgentIdResponse extends GetAgentResponse {
|
|
56
58
|
evals: any[];
|
|
59
|
+
instructions: string;
|
|
60
|
+
name: string;
|
|
61
|
+
id: string;
|
|
57
62
|
}
|
|
58
63
|
|
|
59
64
|
export interface GetToolResponse {
|
|
@@ -72,12 +77,43 @@ export interface GetWorkflowResponse {
|
|
|
72
77
|
workflowId?: string;
|
|
73
78
|
}
|
|
74
79
|
|
|
80
|
+
export interface GetWorkflowRunsParams {
|
|
81
|
+
fromDate?: Date;
|
|
82
|
+
toDate?: Date;
|
|
83
|
+
limit?: number;
|
|
84
|
+
offset?: number;
|
|
85
|
+
resourceId?: string;
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
export type GetWorkflowRunsResponse = WorkflowRuns;
|
|
89
|
+
|
|
75
90
|
export type WorkflowRunResult = {
|
|
76
91
|
activePaths: Record<string, { status: string; suspendPayload?: any; stepPath: string[] }>;
|
|
77
92
|
results: CoreWorkflowRunResult<any, any, any>['results'];
|
|
78
93
|
timestamp: number;
|
|
79
94
|
runId: string;
|
|
80
95
|
};
|
|
96
|
+
|
|
97
|
+
export interface GetVNextWorkflowResponse {
|
|
98
|
+
name: string;
|
|
99
|
+
steps: {
|
|
100
|
+
[key: string]: {
|
|
101
|
+
id: string;
|
|
102
|
+
description: string;
|
|
103
|
+
inputSchema: string;
|
|
104
|
+
outputSchema: string;
|
|
105
|
+
resumeSchema: string;
|
|
106
|
+
suspendSchema: string;
|
|
107
|
+
};
|
|
108
|
+
};
|
|
109
|
+
stepGraph: NewWorkflow['serializedStepGraph'];
|
|
110
|
+
inputSchema: string;
|
|
111
|
+
outputSchema: string;
|
|
112
|
+
}
|
|
113
|
+
|
|
114
|
+
export type VNextWorkflowWatchResult = WatchEvent & { runId: string };
|
|
115
|
+
|
|
116
|
+
export type VNextWorkflowRunResult = VNextWorkflowResult<any, any>;
|
|
81
117
|
export interface UpsertVectorParams {
|
|
82
118
|
indexName: string;
|
|
83
119
|
vectors: number[][];
|
|
@@ -118,7 +154,7 @@ export type SaveMessageToMemoryResponse = MessageType[];
|
|
|
118
154
|
export interface CreateMemoryThreadParams {
|
|
119
155
|
title: string;
|
|
120
156
|
metadata: Record<string, any>;
|
|
121
|
-
|
|
157
|
+
resourceId: string;
|
|
122
158
|
threadId: string;
|
|
123
159
|
agentId: string;
|
|
124
160
|
}
|
|
@@ -135,7 +171,7 @@ export type GetMemoryThreadResponse = StorageThreadType[];
|
|
|
135
171
|
export interface UpdateMemoryThreadParams {
|
|
136
172
|
title: string;
|
|
137
173
|
metadata: Record<string, any>;
|
|
138
|
-
|
|
174
|
+
resourceId: string;
|
|
139
175
|
}
|
|
140
176
|
|
|
141
177
|
export interface GetMemoryThreadMessagesResponse {
|
|
@@ -197,7 +233,7 @@ type Span = {
|
|
|
197
233
|
};
|
|
198
234
|
|
|
199
235
|
export interface GetTelemetryResponse {
|
|
200
|
-
traces:
|
|
236
|
+
traces: Span[];
|
|
201
237
|
}
|
|
202
238
|
|
|
203
239
|
export interface GetTelemetryParams {
|
|
@@ -206,6 +242,8 @@ export interface GetTelemetryParams {
|
|
|
206
242
|
page?: number;
|
|
207
243
|
perPage?: number;
|
|
208
244
|
attribute?: Record<string, string>;
|
|
245
|
+
fromDate?: Date;
|
|
246
|
+
toDate?: Date;
|
|
209
247
|
}
|
|
210
248
|
|
|
211
249
|
export interface GetNetworkResponse {
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { ZodSchema } from 'zod';
|
|
2
|
+
import originalZodToJsonSchema from 'zod-to-json-schema';
|
|
3
|
+
|
|
4
|
+
export function zodToJsonSchema<T extends ZodSchema | any>(zodSchema: T) {
|
|
5
|
+
if (!(zodSchema instanceof ZodSchema)) {
|
|
6
|
+
return zodSchema;
|
|
7
|
+
}
|
|
8
|
+
|
|
9
|
+
return originalZodToJsonSchema(zodSchema, { $refStrategy: 'none' });
|
|
10
|
+
}
|