@mastra/client-js 0.0.0-fix-message-embedding-20250506021742 → 0.0.0-fix-generate-title-20250616171351
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/.turbo/turbo-build.log +19 -0
- package/CHANGELOG.md +469 -2
- package/dist/index.cjs +525 -108
- package/dist/index.d.cts +304 -82
- package/dist/index.d.ts +304 -82
- package/dist/index.js +521 -108
- package/package.json +19 -14
- package/src/adapters/agui.test.ts +19 -6
- package/src/adapters/agui.ts +31 -11
- package/src/client.ts +172 -21
- package/src/example.ts +29 -30
- package/src/index.test.ts +121 -1
- package/src/resources/a2a.ts +88 -0
- package/src/resources/agent.ts +48 -35
- package/src/resources/base.ts +1 -1
- package/src/resources/index.ts +4 -2
- package/src/resources/{vnext-workflow.ts → legacy-workflow.ts} +124 -139
- package/src/resources/mcp-tool.ts +48 -0
- package/src/resources/memory-thread.ts +13 -3
- package/src/resources/network.ts +5 -11
- package/src/resources/tool.ts +9 -2
- package/src/resources/workflow.ts +218 -96
- package/src/types.ts +99 -22
- package/src/utils/index.ts +11 -0
- package/src/utils/process-client-tools.ts +31 -0
- package/src/utils/zod-to-json-schema.ts +10 -0
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
import type { TaskSendParams, TaskQueryParams, TaskIdParams, Task, AgentCard, JSONRPCResponse } from '@mastra/core/a2a';
|
|
2
|
+
import type { ClientOptions } from '../types';
|
|
3
|
+
import { BaseResource } from './base';
|
|
4
|
+
|
|
5
|
+
/**
|
|
6
|
+
* Class for interacting with an agent via the A2A protocol
|
|
7
|
+
*/
|
|
8
|
+
export class A2A extends BaseResource {
|
|
9
|
+
constructor(
|
|
10
|
+
options: ClientOptions,
|
|
11
|
+
private agentId: string,
|
|
12
|
+
) {
|
|
13
|
+
super(options);
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
/**
|
|
17
|
+
* Get the agent card with metadata about the agent
|
|
18
|
+
* @returns Promise containing the agent card information
|
|
19
|
+
*/
|
|
20
|
+
async getCard(): Promise<AgentCard> {
|
|
21
|
+
return this.request(`/.well-known/${this.agentId}/agent.json`);
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
/**
|
|
25
|
+
* Send a message to the agent and get a response
|
|
26
|
+
* @param params - Parameters for the task
|
|
27
|
+
* @returns Promise containing the task response
|
|
28
|
+
*/
|
|
29
|
+
async sendMessage(params: TaskSendParams): Promise<{ task: Task }> {
|
|
30
|
+
const response = await this.request<JSONRPCResponse<Task>>(`/a2a/${this.agentId}`, {
|
|
31
|
+
method: 'POST',
|
|
32
|
+
body: {
|
|
33
|
+
method: 'tasks/send',
|
|
34
|
+
params,
|
|
35
|
+
},
|
|
36
|
+
});
|
|
37
|
+
|
|
38
|
+
return { task: response.result! };
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
/**
|
|
42
|
+
* Get the status and result of a task
|
|
43
|
+
* @param params - Parameters for querying the task
|
|
44
|
+
* @returns Promise containing the task response
|
|
45
|
+
*/
|
|
46
|
+
async getTask(params: TaskQueryParams): Promise<Task> {
|
|
47
|
+
const response = await this.request<JSONRPCResponse<Task>>(`/a2a/${this.agentId}`, {
|
|
48
|
+
method: 'POST',
|
|
49
|
+
body: {
|
|
50
|
+
method: 'tasks/get',
|
|
51
|
+
params,
|
|
52
|
+
},
|
|
53
|
+
});
|
|
54
|
+
|
|
55
|
+
return response.result!;
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
/**
|
|
59
|
+
* Cancel a running task
|
|
60
|
+
* @param params - Parameters identifying the task to cancel
|
|
61
|
+
* @returns Promise containing the task response
|
|
62
|
+
*/
|
|
63
|
+
async cancelTask(params: TaskIdParams): Promise<{ task: Task }> {
|
|
64
|
+
return this.request(`/a2a/${this.agentId}`, {
|
|
65
|
+
method: 'POST',
|
|
66
|
+
body: {
|
|
67
|
+
method: 'tasks/cancel',
|
|
68
|
+
params,
|
|
69
|
+
},
|
|
70
|
+
});
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
/**
|
|
74
|
+
* Send a message and subscribe to streaming updates (not fully implemented)
|
|
75
|
+
* @param params - Parameters for the task
|
|
76
|
+
* @returns Promise containing the task response
|
|
77
|
+
*/
|
|
78
|
+
async sendAndSubscribe(params: TaskSendParams): Promise<Response> {
|
|
79
|
+
return this.request(`/a2a/${this.agentId}`, {
|
|
80
|
+
method: 'POST',
|
|
81
|
+
body: {
|
|
82
|
+
method: 'tasks/sendSubscribe',
|
|
83
|
+
params,
|
|
84
|
+
},
|
|
85
|
+
stream: true,
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
}
|
package/src/resources/agent.ts
CHANGED
|
@@ -1,8 +1,9 @@
|
|
|
1
1
|
import { processDataStream } from '@ai-sdk/ui-utils';
|
|
2
|
-
import type
|
|
2
|
+
import { type GenerateReturn } from '@mastra/core';
|
|
3
3
|
import type { JSONSchema7 } from 'json-schema';
|
|
4
4
|
import { ZodSchema } from 'zod';
|
|
5
|
-
import { zodToJsonSchema } from 'zod-to-json-schema';
|
|
5
|
+
import { zodToJsonSchema } from '../utils/zod-to-json-schema';
|
|
6
|
+
import { processClientTools } from '../utils/process-client-tools';
|
|
6
7
|
|
|
7
8
|
import type {
|
|
8
9
|
GenerateParams,
|
|
@@ -14,28 +15,8 @@ import type {
|
|
|
14
15
|
} from '../types';
|
|
15
16
|
|
|
16
17
|
import { BaseResource } from './base';
|
|
17
|
-
|
|
18
|
-
|
|
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
|
-
execute(params: { data: any }): Promise<any> {
|
|
33
|
-
return this.request(`/api/agents/${this.agentId}/tools/${this.toolId}/execute`, {
|
|
34
|
-
method: 'POST',
|
|
35
|
-
body: params,
|
|
36
|
-
});
|
|
37
|
-
}
|
|
38
|
-
}
|
|
18
|
+
import type { RuntimeContext } from '@mastra/core/runtime-context';
|
|
19
|
+
import { parseClientRuntimeContext } from '../utils';
|
|
39
20
|
|
|
40
21
|
export class AgentVoice extends BaseResource {
|
|
41
22
|
constructor(
|
|
@@ -69,7 +50,7 @@ export class AgentVoice extends BaseResource {
|
|
|
69
50
|
* @param options - Optional provider-specific options
|
|
70
51
|
* @returns Promise containing the transcribed text
|
|
71
52
|
*/
|
|
72
|
-
listen(audio: Blob, options?: Record<string, any>): Promise<
|
|
53
|
+
listen(audio: Blob, options?: Record<string, any>): Promise<{ text: string }> {
|
|
73
54
|
const formData = new FormData();
|
|
74
55
|
formData.append('audio', audio);
|
|
75
56
|
|
|
@@ -90,6 +71,14 @@ export class AgentVoice extends BaseResource {
|
|
|
90
71
|
getSpeakers(): Promise<Array<{ voiceId: string; [key: string]: any }>> {
|
|
91
72
|
return this.request(`/api/agents/${this.agentId}/voice/speakers`);
|
|
92
73
|
}
|
|
74
|
+
|
|
75
|
+
/**
|
|
76
|
+
* Get the listener configuration for the agent's voice provider
|
|
77
|
+
* @returns Promise containing a check if the agent has listening capabilities
|
|
78
|
+
*/
|
|
79
|
+
getListener(): Promise<{ enabled: boolean }> {
|
|
80
|
+
return this.request(`/api/agents/${this.agentId}/voice/listener`);
|
|
81
|
+
}
|
|
93
82
|
}
|
|
94
83
|
|
|
95
84
|
export class Agent extends BaseResource {
|
|
@@ -116,16 +105,24 @@ export class Agent extends BaseResource {
|
|
|
116
105
|
* @param params - Generation parameters including prompt
|
|
117
106
|
* @returns Promise containing the generated response
|
|
118
107
|
*/
|
|
108
|
+
generate<T extends JSONSchema7 | ZodSchema | undefined = undefined>(
|
|
109
|
+
params: GenerateParams<T> & { output?: never; experimental_output?: never },
|
|
110
|
+
): Promise<GenerateReturn<T>>;
|
|
111
|
+
generate<T extends JSONSchema7 | ZodSchema | undefined = undefined>(
|
|
112
|
+
params: GenerateParams<T> & { output: T; experimental_output?: never },
|
|
113
|
+
): Promise<GenerateReturn<T>>;
|
|
114
|
+
generate<T extends JSONSchema7 | ZodSchema | undefined = undefined>(
|
|
115
|
+
params: GenerateParams<T> & { output?: never; experimental_output: T },
|
|
116
|
+
): Promise<GenerateReturn<T>>;
|
|
119
117
|
generate<T extends JSONSchema7 | ZodSchema | undefined = undefined>(
|
|
120
118
|
params: GenerateParams<T>,
|
|
121
119
|
): Promise<GenerateReturn<T>> {
|
|
122
120
|
const processedParams = {
|
|
123
121
|
...params,
|
|
124
|
-
output: params.output
|
|
125
|
-
experimental_output:
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
: params.experimental_output,
|
|
122
|
+
output: params.output ? zodToJsonSchema(params.output) : undefined,
|
|
123
|
+
experimental_output: params.experimental_output ? zodToJsonSchema(params.experimental_output) : undefined,
|
|
124
|
+
runtimeContext: parseClientRuntimeContext(params.runtimeContext),
|
|
125
|
+
clientTools: processClientTools(params.clientTools),
|
|
129
126
|
};
|
|
130
127
|
|
|
131
128
|
return this.request(`/api/agents/${this.agentId}/generate`, {
|
|
@@ -148,11 +145,10 @@ export class Agent extends BaseResource {
|
|
|
148
145
|
> {
|
|
149
146
|
const processedParams = {
|
|
150
147
|
...params,
|
|
151
|
-
output: params.output
|
|
152
|
-
experimental_output:
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
: params.experimental_output,
|
|
148
|
+
output: params.output ? zodToJsonSchema(params.output) : undefined,
|
|
149
|
+
experimental_output: params.experimental_output ? zodToJsonSchema(params.experimental_output) : undefined,
|
|
150
|
+
runtimeContext: parseClientRuntimeContext(params.runtimeContext),
|
|
151
|
+
clientTools: processClientTools(params.clientTools),
|
|
156
152
|
};
|
|
157
153
|
|
|
158
154
|
const response: Response & {
|
|
@@ -186,6 +182,23 @@ export class Agent extends BaseResource {
|
|
|
186
182
|
return this.request(`/api/agents/${this.agentId}/tools/${toolId}`);
|
|
187
183
|
}
|
|
188
184
|
|
|
185
|
+
/**
|
|
186
|
+
* Executes a tool for the agent
|
|
187
|
+
* @param toolId - ID of the tool to execute
|
|
188
|
+
* @param params - Parameters required for tool execution
|
|
189
|
+
* @returns Promise containing the tool execution results
|
|
190
|
+
*/
|
|
191
|
+
executeTool(toolId: string, params: { data: any; runtimeContext?: RuntimeContext }): Promise<any> {
|
|
192
|
+
const body = {
|
|
193
|
+
data: params.data,
|
|
194
|
+
runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : undefined,
|
|
195
|
+
};
|
|
196
|
+
return this.request(`/api/agents/${this.agentId}/tools/${toolId}/execute`, {
|
|
197
|
+
method: 'POST',
|
|
198
|
+
body,
|
|
199
|
+
});
|
|
200
|
+
}
|
|
201
|
+
|
|
189
202
|
/**
|
|
190
203
|
* Retrieves evaluation results for the agent
|
|
191
204
|
* @returns Promise containing agent evaluations
|
package/src/resources/base.ts
CHANGED
|
@@ -21,7 +21,7 @@ export class BaseResource {
|
|
|
21
21
|
|
|
22
22
|
for (let attempt = 0; attempt <= retries; attempt++) {
|
|
23
23
|
try {
|
|
24
|
-
const response = await fetch(`${baseUrl}${path}`, {
|
|
24
|
+
const response = await fetch(`${baseUrl.replace(/\/$/, '')}${path}`, {
|
|
25
25
|
...options,
|
|
26
26
|
headers: {
|
|
27
27
|
...headers,
|
package/src/resources/index.ts
CHANGED
|
@@ -2,7 +2,9 @@ export * from './agent';
|
|
|
2
2
|
export * from './network';
|
|
3
3
|
export * from './memory-thread';
|
|
4
4
|
export * from './vector';
|
|
5
|
-
export * from './workflow';
|
|
5
|
+
export * from './legacy-workflow';
|
|
6
6
|
export * from './tool';
|
|
7
7
|
export * from './base';
|
|
8
|
-
export * from './
|
|
8
|
+
export * from './workflow';
|
|
9
|
+
export * from './a2a';
|
|
10
|
+
export * from './mcp-tool';
|
|
@@ -1,18 +1,16 @@
|
|
|
1
|
-
import type { RuntimeContext } from '@mastra/core/runtime-context';
|
|
2
1
|
import type {
|
|
3
2
|
ClientOptions,
|
|
4
|
-
|
|
3
|
+
LegacyWorkflowRunResult,
|
|
4
|
+
GetLegacyWorkflowRunsResponse,
|
|
5
5
|
GetWorkflowRunsParams,
|
|
6
|
-
|
|
7
|
-
VNextWorkflowRunResult,
|
|
8
|
-
VNextWorkflowWatchResult,
|
|
6
|
+
GetLegacyWorkflowResponse,
|
|
9
7
|
} from '../types';
|
|
10
8
|
|
|
11
9
|
import { BaseResource } from './base';
|
|
12
10
|
|
|
13
11
|
const RECORD_SEPARATOR = '\x1E';
|
|
14
12
|
|
|
15
|
-
export class
|
|
13
|
+
export class LegacyWorkflow extends BaseResource {
|
|
16
14
|
constructor(
|
|
17
15
|
options: ClientOptions,
|
|
18
16
|
private workflowId: string,
|
|
@@ -21,90 +19,19 @@ export class VNextWorkflow extends BaseResource {
|
|
|
21
19
|
}
|
|
22
20
|
|
|
23
21
|
/**
|
|
24
|
-
*
|
|
25
|
-
*
|
|
26
|
-
*
|
|
27
|
-
* @param stream - The readable stream to process
|
|
28
|
-
* @returns An async generator that yields parsed records
|
|
22
|
+
* Retrieves details about the legacy workflow
|
|
23
|
+
* @returns Promise containing legacy workflow details including steps and graphs
|
|
29
24
|
*/
|
|
30
|
-
|
|
31
|
-
|
|
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}`);
|
|
25
|
+
details(): Promise<GetLegacyWorkflowResponse> {
|
|
26
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}`);
|
|
100
27
|
}
|
|
101
28
|
|
|
102
29
|
/**
|
|
103
|
-
* Retrieves all runs for a
|
|
30
|
+
* Retrieves all runs for a legacy workflow
|
|
104
31
|
* @param params - Parameters for filtering runs
|
|
105
|
-
* @returns Promise containing
|
|
32
|
+
* @returns Promise containing legacy workflow runs array
|
|
106
33
|
*/
|
|
107
|
-
runs(params?: GetWorkflowRunsParams): Promise<
|
|
34
|
+
runs(params?: GetWorkflowRunsParams): Promise<GetLegacyWorkflowRunsResponse> {
|
|
108
35
|
const searchParams = new URLSearchParams();
|
|
109
36
|
if (params?.fromDate) {
|
|
110
37
|
searchParams.set('fromDate', params.fromDate.toISOString());
|
|
@@ -123,16 +50,15 @@ export class VNextWorkflow extends BaseResource {
|
|
|
123
50
|
}
|
|
124
51
|
|
|
125
52
|
if (searchParams.size) {
|
|
126
|
-
return this.request(`/api/workflows/
|
|
53
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/runs?${searchParams}`);
|
|
127
54
|
} else {
|
|
128
|
-
return this.request(`/api/workflows/
|
|
55
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/runs`);
|
|
129
56
|
}
|
|
130
57
|
}
|
|
131
58
|
|
|
132
59
|
/**
|
|
133
|
-
* Creates a new
|
|
134
|
-
* @
|
|
135
|
-
* @returns Promise containing the runId of the created run
|
|
60
|
+
* Creates a new legacy workflow run
|
|
61
|
+
* @returns Promise containing the generated run ID
|
|
136
62
|
*/
|
|
137
63
|
createRun(params?: { runId?: string }): Promise<{ runId: string }> {
|
|
138
64
|
const searchParams = new URLSearchParams();
|
|
@@ -141,109 +67,168 @@ export class VNextWorkflow extends BaseResource {
|
|
|
141
67
|
searchParams.set('runId', params.runId);
|
|
142
68
|
}
|
|
143
69
|
|
|
144
|
-
return this.request(`/api/workflows/
|
|
70
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/create-run?${searchParams.toString()}`, {
|
|
145
71
|
method: 'POST',
|
|
146
72
|
});
|
|
147
73
|
}
|
|
148
74
|
|
|
149
75
|
/**
|
|
150
|
-
* Starts a
|
|
151
|
-
* @param params - Object containing the runId
|
|
76
|
+
* Starts a legacy workflow run synchronously without waiting for the workflow to complete
|
|
77
|
+
* @param params - Object containing the runId and triggerData
|
|
152
78
|
* @returns Promise containing success message
|
|
153
79
|
*/
|
|
154
|
-
start(params: {
|
|
155
|
-
runId
|
|
156
|
-
inputData: Record<string, any>;
|
|
157
|
-
runtimeContext?: RuntimeContext;
|
|
158
|
-
}): Promise<{ message: string }> {
|
|
159
|
-
return this.request(`/api/workflows/v-next/${this.workflowId}/start?runId=${params.runId}`, {
|
|
80
|
+
start(params: { runId: string; triggerData: Record<string, any> }): Promise<{ message: string }> {
|
|
81
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/start?runId=${params.runId}`, {
|
|
160
82
|
method: 'POST',
|
|
161
|
-
body:
|
|
83
|
+
body: params?.triggerData,
|
|
162
84
|
});
|
|
163
85
|
}
|
|
164
86
|
|
|
165
87
|
/**
|
|
166
|
-
* Resumes a suspended
|
|
167
|
-
* @param
|
|
168
|
-
* @
|
|
88
|
+
* Resumes a suspended legacy workflow step synchronously without waiting for the workflow to complete
|
|
89
|
+
* @param stepId - ID of the step to resume
|
|
90
|
+
* @param runId - ID of the legacy workflow run
|
|
91
|
+
* @param context - Context to resume the legacy workflow with
|
|
92
|
+
* @returns Promise containing the legacy workflow resume results
|
|
169
93
|
*/
|
|
170
94
|
resume({
|
|
171
|
-
|
|
95
|
+
stepId,
|
|
172
96
|
runId,
|
|
173
|
-
|
|
174
|
-
runtimeContext,
|
|
97
|
+
context,
|
|
175
98
|
}: {
|
|
176
|
-
|
|
99
|
+
stepId: string;
|
|
177
100
|
runId: string;
|
|
178
|
-
|
|
179
|
-
runtimeContext?: RuntimeContext;
|
|
101
|
+
context: Record<string, any>;
|
|
180
102
|
}): Promise<{ message: string }> {
|
|
181
|
-
return this.request(`/api/workflows/
|
|
103
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/resume?runId=${runId}`, {
|
|
182
104
|
method: 'POST',
|
|
183
|
-
stream: true,
|
|
184
105
|
body: {
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
runtimeContext,
|
|
106
|
+
stepId,
|
|
107
|
+
context,
|
|
188
108
|
},
|
|
189
109
|
});
|
|
190
110
|
}
|
|
191
111
|
|
|
192
112
|
/**
|
|
193
|
-
* Starts a
|
|
194
|
-
* @param params - Object containing the optional runId
|
|
195
|
-
* @returns Promise containing the
|
|
113
|
+
* Starts a workflow run asynchronously and returns a promise that resolves when the workflow is complete
|
|
114
|
+
* @param params - Object containing the optional runId and triggerData
|
|
115
|
+
* @returns Promise containing the workflow execution results
|
|
196
116
|
*/
|
|
197
|
-
startAsync(params: {
|
|
198
|
-
runId?: string;
|
|
199
|
-
inputData: Record<string, any>;
|
|
200
|
-
runtimeContext?: RuntimeContext;
|
|
201
|
-
}): Promise<VNextWorkflowRunResult> {
|
|
117
|
+
startAsync(params: { runId?: string; triggerData: Record<string, any> }): Promise<LegacyWorkflowRunResult> {
|
|
202
118
|
const searchParams = new URLSearchParams();
|
|
203
119
|
|
|
204
120
|
if (!!params?.runId) {
|
|
205
121
|
searchParams.set('runId', params.runId);
|
|
206
122
|
}
|
|
207
123
|
|
|
208
|
-
return this.request(`/api/workflows/
|
|
124
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/start-async?${searchParams.toString()}`, {
|
|
209
125
|
method: 'POST',
|
|
210
|
-
body:
|
|
126
|
+
body: params?.triggerData,
|
|
211
127
|
});
|
|
212
128
|
}
|
|
213
129
|
|
|
214
130
|
/**
|
|
215
|
-
* Resumes a suspended
|
|
216
|
-
* @param params - Object containing the runId,
|
|
217
|
-
* @returns Promise containing the
|
|
131
|
+
* Resumes a suspended legacy workflow step asynchronously and returns a promise that resolves when the workflow is complete
|
|
132
|
+
* @param params - Object containing the runId, stepId, and context
|
|
133
|
+
* @returns Promise containing the workflow resume results
|
|
218
134
|
*/
|
|
219
135
|
resumeAsync(params: {
|
|
220
136
|
runId: string;
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
return this.request(`/api/workflows/v-next/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
137
|
+
stepId: string;
|
|
138
|
+
context: Record<string, any>;
|
|
139
|
+
}): Promise<LegacyWorkflowRunResult> {
|
|
140
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
226
141
|
method: 'POST',
|
|
227
142
|
body: {
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
runtimeContext: params.runtimeContext,
|
|
143
|
+
stepId: params.stepId,
|
|
144
|
+
context: params.context,
|
|
231
145
|
},
|
|
232
146
|
});
|
|
233
147
|
}
|
|
234
148
|
|
|
235
149
|
/**
|
|
236
|
-
*
|
|
150
|
+
* Creates an async generator that processes a readable stream and yields records
|
|
151
|
+
* separated by the Record Separator character (\x1E)
|
|
152
|
+
*
|
|
153
|
+
* @param stream - The readable stream to process
|
|
154
|
+
* @returns An async generator that yields parsed records
|
|
155
|
+
*/
|
|
156
|
+
private async *streamProcessor(stream: ReadableStream): AsyncGenerator<LegacyWorkflowRunResult, void, unknown> {
|
|
157
|
+
const reader = stream.getReader();
|
|
158
|
+
|
|
159
|
+
// Track if we've finished reading from the stream
|
|
160
|
+
let doneReading = false;
|
|
161
|
+
// Buffer to accumulate partial chunks
|
|
162
|
+
let buffer = '';
|
|
163
|
+
|
|
164
|
+
try {
|
|
165
|
+
while (!doneReading) {
|
|
166
|
+
// Read the next chunk from the stream
|
|
167
|
+
const { done, value } = await reader.read();
|
|
168
|
+
doneReading = done;
|
|
169
|
+
|
|
170
|
+
// Skip processing if we're done and there's no value
|
|
171
|
+
if (done && !value) continue;
|
|
172
|
+
|
|
173
|
+
try {
|
|
174
|
+
// Decode binary data to text
|
|
175
|
+
const decoded = value ? new TextDecoder().decode(value) : '';
|
|
176
|
+
|
|
177
|
+
// Split the combined buffer and new data by record separator
|
|
178
|
+
const chunks = (buffer + decoded).split(RECORD_SEPARATOR);
|
|
179
|
+
|
|
180
|
+
// The last chunk might be incomplete, so save it for the next iteration
|
|
181
|
+
buffer = chunks.pop() || '';
|
|
182
|
+
|
|
183
|
+
// Process complete chunks
|
|
184
|
+
for (const chunk of chunks) {
|
|
185
|
+
if (chunk) {
|
|
186
|
+
// Only process non-empty chunks
|
|
187
|
+
if (typeof chunk === 'string') {
|
|
188
|
+
try {
|
|
189
|
+
const parsedChunk = JSON.parse(chunk);
|
|
190
|
+
yield parsedChunk;
|
|
191
|
+
} catch {
|
|
192
|
+
// Silently ignore parsing errors to maintain stream processing
|
|
193
|
+
// This allows the stream to continue even if one record is malformed
|
|
194
|
+
}
|
|
195
|
+
}
|
|
196
|
+
}
|
|
197
|
+
}
|
|
198
|
+
} catch {
|
|
199
|
+
// Silently ignore parsing errors to maintain stream processing
|
|
200
|
+
// This allows the stream to continue even if one record is malformed
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
|
|
204
|
+
// Process any remaining data in the buffer after stream is done
|
|
205
|
+
if (buffer) {
|
|
206
|
+
try {
|
|
207
|
+
yield JSON.parse(buffer);
|
|
208
|
+
} catch {
|
|
209
|
+
// Ignore parsing error for final chunk
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
} finally {
|
|
213
|
+
// Always ensure we clean up the reader
|
|
214
|
+
reader.cancel().catch(() => {
|
|
215
|
+
// Ignore cancel errors
|
|
216
|
+
});
|
|
217
|
+
}
|
|
218
|
+
}
|
|
219
|
+
|
|
220
|
+
/**
|
|
221
|
+
* Watches legacy workflow transitions in real-time
|
|
237
222
|
* @param runId - Optional run ID to filter the watch stream
|
|
238
|
-
* @returns AsyncGenerator that yields parsed records from the
|
|
223
|
+
* @returns AsyncGenerator that yields parsed records from the legacy workflow watch stream
|
|
239
224
|
*/
|
|
240
|
-
async watch({ runId }: { runId?: string }, onRecord: (record:
|
|
241
|
-
const response: Response = await this.request(`/api/workflows/
|
|
225
|
+
async watch({ runId }: { runId?: string }, onRecord: (record: LegacyWorkflowRunResult) => void) {
|
|
226
|
+
const response: Response = await this.request(`/api/workflows/legacy/${this.workflowId}/watch?runId=${runId}`, {
|
|
242
227
|
stream: true,
|
|
243
228
|
});
|
|
244
229
|
|
|
245
230
|
if (!response.ok) {
|
|
246
|
-
throw new Error(`Failed to watch
|
|
231
|
+
throw new Error(`Failed to watch legacy workflow: ${response.statusText}`);
|
|
247
232
|
}
|
|
248
233
|
|
|
249
234
|
if (!response.body) {
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import type { RuntimeContext } from '@mastra/core/runtime-context';
|
|
2
|
+
import type { ClientOptions, McpToolInfo } from '../types';
|
|
3
|
+
import { BaseResource } from './base';
|
|
4
|
+
|
|
5
|
+
/**
|
|
6
|
+
* Represents a specific tool available on a specific MCP server.
|
|
7
|
+
* Provides methods to get details and execute the tool.
|
|
8
|
+
*/
|
|
9
|
+
export class MCPTool extends BaseResource {
|
|
10
|
+
private serverId: string;
|
|
11
|
+
private toolId: string;
|
|
12
|
+
|
|
13
|
+
constructor(options: ClientOptions, serverId: string, toolId: string) {
|
|
14
|
+
super(options);
|
|
15
|
+
this.serverId = serverId;
|
|
16
|
+
this.toolId = toolId;
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
/**
|
|
20
|
+
* Retrieves details about this specific tool from the MCP server.
|
|
21
|
+
* @returns Promise containing the tool's information (name, description, schema).
|
|
22
|
+
*/
|
|
23
|
+
details(): Promise<McpToolInfo> {
|
|
24
|
+
return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}`);
|
|
25
|
+
}
|
|
26
|
+
|
|
27
|
+
/**
|
|
28
|
+
* Executes this specific tool on the MCP server.
|
|
29
|
+
* @param params - Parameters for tool execution, including data/args and optional runtimeContext.
|
|
30
|
+
* @returns Promise containing the result of the tool execution.
|
|
31
|
+
*/
|
|
32
|
+
execute(params: { data?: any; runtimeContext?: RuntimeContext }): Promise<any> {
|
|
33
|
+
const body: any = {};
|
|
34
|
+
if (params.data !== undefined) body.data = params.data;
|
|
35
|
+
// If none of data, args the body might be empty or just contain runtimeContext.
|
|
36
|
+
// The handler will look for these, so an empty args object might be appropriate if that's the intent.
|
|
37
|
+
// else body.data = {}; // Or let it be empty if no specific input fields are used
|
|
38
|
+
|
|
39
|
+
if (params.runtimeContext !== undefined) {
|
|
40
|
+
body.runtimeContext = params.runtimeContext;
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}/execute`, {
|
|
44
|
+
method: 'POST',
|
|
45
|
+
body: Object.keys(body).length > 0 ? body : undefined,
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
}
|