@mastra/client-js 0.0.0-generate-message-id-20250512171942 → 0.0.0-http-transporter-20250702160118
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 +596 -4
- package/README.md +1 -1
- package/dist/index.cjs +1045 -112
- package/dist/index.d.cts +460 -86
- package/dist/index.d.ts +460 -86
- package/dist/index.js +1046 -113
- package/package.json +18 -13
- package/src/client.ts +234 -22
- package/src/example.ts +33 -31
- package/src/index.test.ts +121 -1
- package/src/resources/agent.ts +585 -18
- package/src/resources/base.ts +2 -1
- package/src/resources/index.ts +3 -2
- package/src/resources/{vnext-workflow.ts → legacy-workflow.ts} +124 -143
- package/src/resources/mcp-tool.ts +48 -0
- package/src/resources/memory-thread.ts +13 -3
- package/src/resources/network-memory-thread.ts +63 -0
- package/src/resources/network.ts +2 -3
- package/src/resources/tool.ts +4 -3
- package/src/resources/vNextNetwork.ts +177 -0
- package/src/resources/workflow.ts +241 -96
- package/src/types.ts +182 -22
- package/src/utils/index.ts +11 -0
- package/src/utils/process-client-tools.ts +32 -0
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@mastra/client-js",
|
|
3
|
-
"version": "0.0.0-
|
|
3
|
+
"version": "0.0.0-http-transporter-20250702160118",
|
|
4
4
|
"description": "The official TypeScript library for the Mastra Client API",
|
|
5
5
|
"author": "",
|
|
6
6
|
"type": "module",
|
|
@@ -19,30 +19,35 @@
|
|
|
19
19
|
},
|
|
20
20
|
"./package.json": "./package.json"
|
|
21
21
|
},
|
|
22
|
-
"repository":
|
|
22
|
+
"repository": {
|
|
23
|
+
"type": "git",
|
|
24
|
+
"url": "https://github.com/mastra-ai/mastra.git",
|
|
25
|
+
"directory": "client-sdks/client-js"
|
|
26
|
+
},
|
|
27
|
+
"homepage": "https://github.com/mastra-ai/mastra/tree/main/client-sdks/client-js#readme",
|
|
23
28
|
"license": "Elastic-2.0",
|
|
24
29
|
"dependencies": {
|
|
25
30
|
"@ag-ui/client": "^0.0.27",
|
|
26
|
-
"@ai-sdk/ui-utils": "^1.
|
|
31
|
+
"@ai-sdk/ui-utils": "^1.2.11",
|
|
27
32
|
"json-schema": "^0.4.0",
|
|
28
33
|
"rxjs": "7.8.1",
|
|
29
|
-
"zod": "^3.
|
|
34
|
+
"zod": "^3.25.67",
|
|
30
35
|
"zod-to-json-schema": "^3.24.5",
|
|
31
|
-
"@mastra/core": "0.0.0-
|
|
36
|
+
"@mastra/core": "0.0.0-http-transporter-20250702160118"
|
|
32
37
|
},
|
|
33
38
|
"peerDependencies": {
|
|
34
39
|
"zod": "^3.0.0"
|
|
35
40
|
},
|
|
36
41
|
"devDependencies": {
|
|
37
|
-
"@babel/preset-env": "^7.
|
|
38
|
-
"@babel/preset-typescript": "^7.27.
|
|
39
|
-
"@tsconfig/recommended": "^1.0.
|
|
42
|
+
"@babel/preset-env": "^7.27.2",
|
|
43
|
+
"@babel/preset-typescript": "^7.27.1",
|
|
44
|
+
"@tsconfig/recommended": "^1.0.9",
|
|
40
45
|
"@types/json-schema": "^7.0.15",
|
|
41
|
-
"@types/node": "^20.
|
|
42
|
-
"tsup": "^8.
|
|
43
|
-
"typescript": "^5.8.
|
|
44
|
-
"vitest": "^3.
|
|
45
|
-
"@internal/lint": "0.0.0-
|
|
46
|
+
"@types/node": "^20.19.0",
|
|
47
|
+
"tsup": "^8.5.0",
|
|
48
|
+
"typescript": "^5.8.3",
|
|
49
|
+
"vitest": "^3.2.4",
|
|
50
|
+
"@internal/lint": "0.0.0-http-transporter-20250702160118"
|
|
46
51
|
},
|
|
47
52
|
"scripts": {
|
|
48
53
|
"build": "tsup src/index.ts --format esm,cjs --dts --clean --treeshake=smallest --splitting",
|
package/src/client.ts
CHANGED
|
@@ -1,6 +1,18 @@
|
|
|
1
1
|
import type { AbstractAgent } from '@ag-ui/client';
|
|
2
|
+
import type { ServerDetailInfo } from '@mastra/core/mcp';
|
|
2
3
|
import { AGUIAdapter } from './adapters/agui';
|
|
3
|
-
import {
|
|
4
|
+
import {
|
|
5
|
+
Agent,
|
|
6
|
+
MemoryThread,
|
|
7
|
+
Tool,
|
|
8
|
+
Workflow,
|
|
9
|
+
Vector,
|
|
10
|
+
BaseResource,
|
|
11
|
+
Network,
|
|
12
|
+
A2A,
|
|
13
|
+
MCPTool,
|
|
14
|
+
LegacyWorkflow,
|
|
15
|
+
} from './resources';
|
|
4
16
|
import type {
|
|
5
17
|
ClientOptions,
|
|
6
18
|
CreateMemoryThreadParams,
|
|
@@ -15,11 +27,19 @@ import type {
|
|
|
15
27
|
GetTelemetryParams,
|
|
16
28
|
GetTelemetryResponse,
|
|
17
29
|
GetToolResponse,
|
|
18
|
-
GetVNextWorkflowResponse,
|
|
19
30
|
GetWorkflowResponse,
|
|
20
31
|
SaveMessageToMemoryParams,
|
|
21
32
|
SaveMessageToMemoryResponse,
|
|
33
|
+
McpServerListResponse,
|
|
34
|
+
McpServerToolListResponse,
|
|
35
|
+
GetLegacyWorkflowResponse,
|
|
36
|
+
GetVNextNetworkResponse,
|
|
37
|
+
GetNetworkMemoryThreadParams,
|
|
38
|
+
CreateNetworkMemoryThreadParams,
|
|
39
|
+
SaveNetworkMessageToMemoryParams,
|
|
22
40
|
} from './types';
|
|
41
|
+
import { VNextNetwork } from './resources/vNextNetwork';
|
|
42
|
+
import { NetworkMemoryThread } from './resources/network-memory-thread';
|
|
23
43
|
|
|
24
44
|
export class MastraClient extends BaseResource {
|
|
25
45
|
constructor(options: ClientOptions) {
|
|
@@ -109,6 +129,53 @@ export class MastraClient extends BaseResource {
|
|
|
109
129
|
return this.request(`/api/memory/status?agentId=${agentId}`);
|
|
110
130
|
}
|
|
111
131
|
|
|
132
|
+
/**
|
|
133
|
+
* Retrieves memory threads for a resource
|
|
134
|
+
* @param params - Parameters containing the resource ID
|
|
135
|
+
* @returns Promise containing array of memory threads
|
|
136
|
+
*/
|
|
137
|
+
public getNetworkMemoryThreads(params: GetNetworkMemoryThreadParams): Promise<GetMemoryThreadResponse> {
|
|
138
|
+
return this.request(`/api/memory/network/threads?resourceid=${params.resourceId}&networkId=${params.networkId}`);
|
|
139
|
+
}
|
|
140
|
+
|
|
141
|
+
/**
|
|
142
|
+
* Creates a new memory thread
|
|
143
|
+
* @param params - Parameters for creating the memory thread
|
|
144
|
+
* @returns Promise containing the created memory thread
|
|
145
|
+
*/
|
|
146
|
+
public createNetworkMemoryThread(params: CreateNetworkMemoryThreadParams): Promise<CreateMemoryThreadResponse> {
|
|
147
|
+
return this.request(`/api/memory/network/threads?networkId=${params.networkId}`, { method: 'POST', body: params });
|
|
148
|
+
}
|
|
149
|
+
|
|
150
|
+
/**
|
|
151
|
+
* Gets a memory thread instance by ID
|
|
152
|
+
* @param threadId - ID of the memory thread to retrieve
|
|
153
|
+
* @returns MemoryThread instance
|
|
154
|
+
*/
|
|
155
|
+
public getNetworkMemoryThread(threadId: string, networkId: string) {
|
|
156
|
+
return new NetworkMemoryThread(this.options, threadId, networkId);
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
/**
|
|
160
|
+
* Saves messages to memory
|
|
161
|
+
* @param params - Parameters containing messages to save
|
|
162
|
+
* @returns Promise containing the saved messages
|
|
163
|
+
*/
|
|
164
|
+
public saveNetworkMessageToMemory(params: SaveNetworkMessageToMemoryParams): Promise<SaveMessageToMemoryResponse> {
|
|
165
|
+
return this.request(`/api/memory/network/save-messages?networkId=${params.networkId}`, {
|
|
166
|
+
method: 'POST',
|
|
167
|
+
body: params,
|
|
168
|
+
});
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
/**
|
|
172
|
+
* Gets the status of the memory system
|
|
173
|
+
* @returns Promise containing memory system status
|
|
174
|
+
*/
|
|
175
|
+
public getNetworkMemoryStatus(networkId: string): Promise<{ result: boolean }> {
|
|
176
|
+
return this.request(`/api/memory/network/status?networkId=${networkId}`);
|
|
177
|
+
}
|
|
178
|
+
|
|
112
179
|
/**
|
|
113
180
|
* Retrieves all available tools
|
|
114
181
|
* @returns Promise containing map of tool IDs to tool details
|
|
@@ -126,6 +193,23 @@ export class MastraClient extends BaseResource {
|
|
|
126
193
|
return new Tool(this.options, toolId);
|
|
127
194
|
}
|
|
128
195
|
|
|
196
|
+
/**
|
|
197
|
+
* Retrieves all available legacy workflows
|
|
198
|
+
* @returns Promise containing map of legacy workflow IDs to legacy workflow details
|
|
199
|
+
*/
|
|
200
|
+
public getLegacyWorkflows(): Promise<Record<string, GetLegacyWorkflowResponse>> {
|
|
201
|
+
return this.request('/api/workflows/legacy');
|
|
202
|
+
}
|
|
203
|
+
|
|
204
|
+
/**
|
|
205
|
+
* Gets a legacy workflow instance by ID
|
|
206
|
+
* @param workflowId - ID of the legacy workflow to retrieve
|
|
207
|
+
* @returns Legacy Workflow instance
|
|
208
|
+
*/
|
|
209
|
+
public getLegacyWorkflow(workflowId: string) {
|
|
210
|
+
return new LegacyWorkflow(this.options, workflowId);
|
|
211
|
+
}
|
|
212
|
+
|
|
129
213
|
/**
|
|
130
214
|
* Retrieves all available workflows
|
|
131
215
|
* @returns Promise containing map of workflow IDs to workflow details
|
|
@@ -143,23 +227,6 @@ export class MastraClient extends BaseResource {
|
|
|
143
227
|
return new Workflow(this.options, workflowId);
|
|
144
228
|
}
|
|
145
229
|
|
|
146
|
-
/**
|
|
147
|
-
* Retrieves all available vNext workflows
|
|
148
|
-
* @returns Promise containing map of vNext workflow IDs to vNext workflow details
|
|
149
|
-
*/
|
|
150
|
-
public getVNextWorkflows(): Promise<Record<string, GetVNextWorkflowResponse>> {
|
|
151
|
-
return this.request('/api/workflows/v-next');
|
|
152
|
-
}
|
|
153
|
-
|
|
154
|
-
/**
|
|
155
|
-
* Gets a vNext workflow instance by ID
|
|
156
|
-
* @param workflowId - ID of the vNext workflow to retrieve
|
|
157
|
-
* @returns vNext Workflow instance
|
|
158
|
-
*/
|
|
159
|
-
public getVNextWorkflow(workflowId: string) {
|
|
160
|
-
return new VNextWorkflow(this.options, workflowId);
|
|
161
|
-
}
|
|
162
|
-
|
|
163
230
|
/**
|
|
164
231
|
* Gets a vector instance by name
|
|
165
232
|
* @param vectorName - Name of the vector to retrieve
|
|
@@ -175,7 +242,43 @@ export class MastraClient extends BaseResource {
|
|
|
175
242
|
* @returns Promise containing array of log messages
|
|
176
243
|
*/
|
|
177
244
|
public getLogs(params: GetLogsParams): Promise<GetLogsResponse> {
|
|
178
|
-
|
|
245
|
+
const { transportId, fromDate, toDate, logLevel, filters, page, perPage } = params;
|
|
246
|
+
const _filters = filters ? Object.entries(filters).map(([key, value]) => `${key}:${value}`) : [];
|
|
247
|
+
|
|
248
|
+
const searchParams = new URLSearchParams();
|
|
249
|
+
if (transportId) {
|
|
250
|
+
searchParams.set('transportId', transportId);
|
|
251
|
+
}
|
|
252
|
+
if (fromDate) {
|
|
253
|
+
searchParams.set('fromDate', fromDate.toISOString());
|
|
254
|
+
}
|
|
255
|
+
if (toDate) {
|
|
256
|
+
searchParams.set('toDate', toDate.toISOString());
|
|
257
|
+
}
|
|
258
|
+
if (logLevel) {
|
|
259
|
+
searchParams.set('logLevel', logLevel);
|
|
260
|
+
}
|
|
261
|
+
if (page) {
|
|
262
|
+
searchParams.set('page', String(page));
|
|
263
|
+
}
|
|
264
|
+
if (perPage) {
|
|
265
|
+
searchParams.set('perPage', String(perPage));
|
|
266
|
+
}
|
|
267
|
+
if (_filters) {
|
|
268
|
+
if (Array.isArray(_filters)) {
|
|
269
|
+
for (const filter of _filters) {
|
|
270
|
+
searchParams.append('filters', filter);
|
|
271
|
+
}
|
|
272
|
+
} else {
|
|
273
|
+
searchParams.set('filters', _filters);
|
|
274
|
+
}
|
|
275
|
+
}
|
|
276
|
+
|
|
277
|
+
if (searchParams.size) {
|
|
278
|
+
return this.request(`/api/logs?${searchParams}`);
|
|
279
|
+
} else {
|
|
280
|
+
return this.request(`/api/logs`);
|
|
281
|
+
}
|
|
179
282
|
}
|
|
180
283
|
|
|
181
284
|
/**
|
|
@@ -184,7 +287,47 @@ export class MastraClient extends BaseResource {
|
|
|
184
287
|
* @returns Promise containing array of log messages
|
|
185
288
|
*/
|
|
186
289
|
public getLogForRun(params: GetLogParams): Promise<GetLogsResponse> {
|
|
187
|
-
|
|
290
|
+
const { runId, transportId, fromDate, toDate, logLevel, filters, page, perPage } = params;
|
|
291
|
+
|
|
292
|
+
const _filters = filters ? Object.entries(filters).map(([key, value]) => `${key}:${value}`) : [];
|
|
293
|
+
const searchParams = new URLSearchParams();
|
|
294
|
+
if (runId) {
|
|
295
|
+
searchParams.set('runId', runId);
|
|
296
|
+
}
|
|
297
|
+
if (transportId) {
|
|
298
|
+
searchParams.set('transportId', transportId);
|
|
299
|
+
}
|
|
300
|
+
if (fromDate) {
|
|
301
|
+
searchParams.set('fromDate', fromDate.toISOString());
|
|
302
|
+
}
|
|
303
|
+
if (toDate) {
|
|
304
|
+
searchParams.set('toDate', toDate.toISOString());
|
|
305
|
+
}
|
|
306
|
+
if (logLevel) {
|
|
307
|
+
searchParams.set('logLevel', logLevel);
|
|
308
|
+
}
|
|
309
|
+
if (page) {
|
|
310
|
+
searchParams.set('page', String(page));
|
|
311
|
+
}
|
|
312
|
+
if (perPage) {
|
|
313
|
+
searchParams.set('perPage', String(perPage));
|
|
314
|
+
}
|
|
315
|
+
|
|
316
|
+
if (_filters) {
|
|
317
|
+
if (Array.isArray(_filters)) {
|
|
318
|
+
for (const filter of _filters) {
|
|
319
|
+
searchParams.append('filters', filter);
|
|
320
|
+
}
|
|
321
|
+
} else {
|
|
322
|
+
searchParams.set('filters', _filters);
|
|
323
|
+
}
|
|
324
|
+
}
|
|
325
|
+
|
|
326
|
+
if (searchParams.size) {
|
|
327
|
+
return this.request(`/api/logs/${runId}?${searchParams}`);
|
|
328
|
+
} else {
|
|
329
|
+
return this.request(`/api/logs/${runId}`);
|
|
330
|
+
}
|
|
188
331
|
}
|
|
189
332
|
|
|
190
333
|
/**
|
|
@@ -244,10 +387,18 @@ export class MastraClient extends BaseResource {
|
|
|
244
387
|
* Retrieves all available networks
|
|
245
388
|
* @returns Promise containing map of network IDs to network details
|
|
246
389
|
*/
|
|
247
|
-
public getNetworks(): Promise<
|
|
390
|
+
public getNetworks(): Promise<Array<GetNetworkResponse>> {
|
|
248
391
|
return this.request('/api/networks');
|
|
249
392
|
}
|
|
250
393
|
|
|
394
|
+
/**
|
|
395
|
+
* Retrieves all available vNext networks
|
|
396
|
+
* @returns Promise containing map of vNext network IDs to vNext network details
|
|
397
|
+
*/
|
|
398
|
+
public getVNextNetworks(): Promise<Array<GetVNextNetworkResponse>> {
|
|
399
|
+
return this.request('/api/networks/v-next');
|
|
400
|
+
}
|
|
401
|
+
|
|
251
402
|
/**
|
|
252
403
|
* Gets a network instance by ID
|
|
253
404
|
* @param networkId - ID of the network to retrieve
|
|
@@ -257,6 +408,67 @@ export class MastraClient extends BaseResource {
|
|
|
257
408
|
return new Network(this.options, networkId);
|
|
258
409
|
}
|
|
259
410
|
|
|
411
|
+
/**
|
|
412
|
+
* Gets a vNext network instance by ID
|
|
413
|
+
* @param networkId - ID of the vNext network to retrieve
|
|
414
|
+
* @returns vNext Network instance
|
|
415
|
+
*/
|
|
416
|
+
public getVNextNetwork(networkId: string) {
|
|
417
|
+
return new VNextNetwork(this.options, networkId);
|
|
418
|
+
}
|
|
419
|
+
|
|
420
|
+
/**
|
|
421
|
+
* Retrieves a list of available MCP servers.
|
|
422
|
+
* @param params - Optional parameters for pagination (limit, offset).
|
|
423
|
+
* @returns Promise containing the list of MCP servers and pagination info.
|
|
424
|
+
*/
|
|
425
|
+
public getMcpServers(params?: { limit?: number; offset?: number }): Promise<McpServerListResponse> {
|
|
426
|
+
const searchParams = new URLSearchParams();
|
|
427
|
+
if (params?.limit !== undefined) {
|
|
428
|
+
searchParams.set('limit', String(params.limit));
|
|
429
|
+
}
|
|
430
|
+
if (params?.offset !== undefined) {
|
|
431
|
+
searchParams.set('offset', String(params.offset));
|
|
432
|
+
}
|
|
433
|
+
const queryString = searchParams.toString();
|
|
434
|
+
return this.request(`/api/mcp/v0/servers${queryString ? `?${queryString}` : ''}`);
|
|
435
|
+
}
|
|
436
|
+
|
|
437
|
+
/**
|
|
438
|
+
* Retrieves detailed information for a specific MCP server.
|
|
439
|
+
* @param serverId - The ID of the MCP server to retrieve.
|
|
440
|
+
* @param params - Optional parameters, e.g., specific version.
|
|
441
|
+
* @returns Promise containing the detailed MCP server information.
|
|
442
|
+
*/
|
|
443
|
+
public getMcpServerDetails(serverId: string, params?: { version?: string }): Promise<ServerDetailInfo> {
|
|
444
|
+
const searchParams = new URLSearchParams();
|
|
445
|
+
if (params?.version) {
|
|
446
|
+
searchParams.set('version', params.version);
|
|
447
|
+
}
|
|
448
|
+
const queryString = searchParams.toString();
|
|
449
|
+
return this.request(`/api/mcp/v0/servers/${serverId}${queryString ? `?${queryString}` : ''}`);
|
|
450
|
+
}
|
|
451
|
+
|
|
452
|
+
/**
|
|
453
|
+
* Retrieves a list of tools for a specific MCP server.
|
|
454
|
+
* @param serverId - The ID of the MCP server.
|
|
455
|
+
* @returns Promise containing the list of tools.
|
|
456
|
+
*/
|
|
457
|
+
public getMcpServerTools(serverId: string): Promise<McpServerToolListResponse> {
|
|
458
|
+
return this.request(`/api/mcp/${serverId}/tools`);
|
|
459
|
+
}
|
|
460
|
+
|
|
461
|
+
/**
|
|
462
|
+
* Gets an MCPTool resource instance for a specific tool on an MCP server.
|
|
463
|
+
* This instance can then be used to fetch details or execute the tool.
|
|
464
|
+
* @param serverId - The ID of the MCP server.
|
|
465
|
+
* @param toolId - The ID of the tool.
|
|
466
|
+
* @returns MCPTool instance.
|
|
467
|
+
*/
|
|
468
|
+
public getMcpServerTool(serverId: string, toolId: string): MCPTool {
|
|
469
|
+
return new MCPTool(this.options, serverId, toolId);
|
|
470
|
+
}
|
|
471
|
+
|
|
260
472
|
/**
|
|
261
473
|
* Gets an A2A client for interacting with an agent via the A2A protocol
|
|
262
474
|
* @param agentId - ID of the agent to interact with
|
package/src/example.ts
CHANGED
|
@@ -1,40 +1,42 @@
|
|
|
1
|
-
|
|
1
|
+
import { MastraClient } from './client';
|
|
2
|
+
import z from 'zod';
|
|
2
3
|
// import type { WorkflowRunResult } from './types';
|
|
3
4
|
|
|
4
5
|
// Agent
|
|
6
|
+
(async () => {
|
|
7
|
+
const client = new MastraClient({
|
|
8
|
+
baseUrl: 'http://localhost:4111',
|
|
9
|
+
});
|
|
5
10
|
|
|
6
|
-
|
|
7
|
-
// const client = new MastraClient({
|
|
8
|
-
// baseUrl: 'http://localhost:4111',
|
|
9
|
-
// });
|
|
10
|
-
|
|
11
|
-
// console.log('Starting agent...');
|
|
12
|
-
|
|
13
|
-
// try {
|
|
14
|
-
// const agent = client.getAgent('weatherAgent');
|
|
15
|
-
// const response = await agent.stream({
|
|
16
|
-
// messages: 'what is the weather in new york?',
|
|
17
|
-
// })
|
|
11
|
+
console.log('Starting agent...');
|
|
18
12
|
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
// console.log(file);
|
|
25
|
-
// },
|
|
26
|
-
// onDataPart: (data) => {
|
|
27
|
-
// console.log(data);
|
|
28
|
-
// },
|
|
29
|
-
// onErrorPart: (error) => {
|
|
30
|
-
// console.error(error);
|
|
31
|
-
// },
|
|
32
|
-
// });
|
|
13
|
+
try {
|
|
14
|
+
const agent = client.getAgent('weatherAgent');
|
|
15
|
+
const response = await agent.stream({
|
|
16
|
+
messages: 'what is the weather in new york?',
|
|
17
|
+
});
|
|
33
18
|
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
19
|
+
response.processDataStream({
|
|
20
|
+
onTextPart: text => {
|
|
21
|
+
process.stdout.write(text);
|
|
22
|
+
},
|
|
23
|
+
onFilePart: file => {
|
|
24
|
+
console.log(file);
|
|
25
|
+
},
|
|
26
|
+
onDataPart: data => {
|
|
27
|
+
console.log(data);
|
|
28
|
+
},
|
|
29
|
+
onErrorPart: error => {
|
|
30
|
+
console.error(error);
|
|
31
|
+
},
|
|
32
|
+
onToolCallPart(streamPart) {
|
|
33
|
+
console.log(streamPart);
|
|
34
|
+
},
|
|
35
|
+
});
|
|
36
|
+
} catch (error) {
|
|
37
|
+
console.error(error);
|
|
38
|
+
}
|
|
39
|
+
})();
|
|
38
40
|
|
|
39
41
|
// Workflow
|
|
40
42
|
// (async () => {
|
package/src/index.test.ts
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { describe, expect, beforeEach, it, vi } from 'vitest';
|
|
2
|
-
|
|
3
2
|
import { MastraClient } from './client';
|
|
3
|
+
import type { McpServerListResponse, ServerDetailInfo } from './types';
|
|
4
4
|
|
|
5
5
|
// Mock fetch globally
|
|
6
6
|
global.fetch = vi.fn();
|
|
@@ -236,6 +236,7 @@ describe('MastraClient Resources', () => {
|
|
|
236
236
|
model: 'gpt-4',
|
|
237
237
|
instructions: 'Test instructions',
|
|
238
238
|
tools: {},
|
|
239
|
+
workflows: {},
|
|
239
240
|
};
|
|
240
241
|
mockFetchResponse(mockResponse);
|
|
241
242
|
|
|
@@ -552,6 +553,35 @@ describe('MastraClient Resources', () => {
|
|
|
552
553
|
}),
|
|
553
554
|
);
|
|
554
555
|
});
|
|
556
|
+
|
|
557
|
+
it('should get thread messages with limit', async () => {
|
|
558
|
+
const mockResponse = {
|
|
559
|
+
messages: [
|
|
560
|
+
{
|
|
561
|
+
id: '1',
|
|
562
|
+
content: 'test',
|
|
563
|
+
threadId,
|
|
564
|
+
role: 'user',
|
|
565
|
+
type: 'text',
|
|
566
|
+
resourceId: 'test-resource',
|
|
567
|
+
createdAt: new Date(),
|
|
568
|
+
},
|
|
569
|
+
],
|
|
570
|
+
uiMessages: [],
|
|
571
|
+
};
|
|
572
|
+
mockFetchResponse(mockResponse);
|
|
573
|
+
|
|
574
|
+
const limit = 5;
|
|
575
|
+
const result = await memoryThread.getMessages({ limit });
|
|
576
|
+
|
|
577
|
+
expect(result).toEqual(mockResponse);
|
|
578
|
+
expect(global.fetch).toHaveBeenCalledWith(
|
|
579
|
+
`${clientOptions.baseUrl}/api/memory/threads/${threadId}/messages?agentId=${agentId}&limit=${limit}`,
|
|
580
|
+
expect.objectContaining({
|
|
581
|
+
headers: expect.objectContaining(clientOptions.headers),
|
|
582
|
+
}),
|
|
583
|
+
);
|
|
584
|
+
});
|
|
555
585
|
});
|
|
556
586
|
|
|
557
587
|
describe('Tool Resource', () => {
|
|
@@ -707,4 +737,94 @@ describe('MastraClient Resources', () => {
|
|
|
707
737
|
);
|
|
708
738
|
});
|
|
709
739
|
});
|
|
740
|
+
|
|
741
|
+
describe('MCP Server Registry Client Methods', () => {
|
|
742
|
+
const mockServerInfo1 = {
|
|
743
|
+
id: 'mcp-server-1',
|
|
744
|
+
name: 'Test MCP Server 1',
|
|
745
|
+
version_detail: { version: '1.0.0', release_date: '2023-01-01T00:00:00Z', is_latest: true },
|
|
746
|
+
};
|
|
747
|
+
const mockServerInfo2 = {
|
|
748
|
+
id: 'mcp-server-2',
|
|
749
|
+
name: 'Test MCP Server 2',
|
|
750
|
+
version_detail: { version: '1.1.0', release_date: '2023-02-01T00:00:00Z', is_latest: true },
|
|
751
|
+
};
|
|
752
|
+
|
|
753
|
+
const mockServerDetail1: ServerDetailInfo = {
|
|
754
|
+
...mockServerInfo1,
|
|
755
|
+
description: 'Detailed description for server 1',
|
|
756
|
+
package_canonical: 'npm',
|
|
757
|
+
packages: [{ registry_name: 'npm', name: '@example/server1', version: '1.0.0' }],
|
|
758
|
+
remotes: [{ transport_type: 'sse', url: 'http://localhost/sse1' }],
|
|
759
|
+
};
|
|
760
|
+
|
|
761
|
+
describe('getMcpServers()', () => {
|
|
762
|
+
it('should fetch a list of MCP servers', async () => {
|
|
763
|
+
const mockResponse: McpServerListResponse = {
|
|
764
|
+
servers: [mockServerInfo1, mockServerInfo2],
|
|
765
|
+
total_count: 2,
|
|
766
|
+
next: null,
|
|
767
|
+
};
|
|
768
|
+
mockFetchResponse(mockResponse);
|
|
769
|
+
|
|
770
|
+
const result = await client.getMcpServers();
|
|
771
|
+
expect(result).toEqual(mockResponse);
|
|
772
|
+
expect(global.fetch).toHaveBeenCalledWith(
|
|
773
|
+
`${clientOptions.baseUrl}/api/mcp/v0/servers`,
|
|
774
|
+
expect.objectContaining({
|
|
775
|
+
headers: expect.objectContaining(clientOptions.headers),
|
|
776
|
+
}),
|
|
777
|
+
);
|
|
778
|
+
});
|
|
779
|
+
|
|
780
|
+
it('should fetch MCP servers with limit and offset parameters', async () => {
|
|
781
|
+
const mockResponse: McpServerListResponse = {
|
|
782
|
+
servers: [mockServerInfo1],
|
|
783
|
+
total_count: 2,
|
|
784
|
+
next: '/api/mcp/v0/servers?limit=1&offset=1',
|
|
785
|
+
};
|
|
786
|
+
mockFetchResponse(mockResponse);
|
|
787
|
+
|
|
788
|
+
const result = await client.getMcpServers({ limit: 1, offset: 0 });
|
|
789
|
+
expect(result).toEqual(mockResponse);
|
|
790
|
+
expect(global.fetch).toHaveBeenCalledWith(
|
|
791
|
+
`${clientOptions.baseUrl}/api/mcp/v0/servers?limit=1&offset=0`,
|
|
792
|
+
expect.objectContaining({
|
|
793
|
+
headers: expect.objectContaining(clientOptions.headers),
|
|
794
|
+
}),
|
|
795
|
+
);
|
|
796
|
+
});
|
|
797
|
+
});
|
|
798
|
+
|
|
799
|
+
describe('getMcpServerDetails()', () => {
|
|
800
|
+
const serverId = 'mcp-server-1';
|
|
801
|
+
|
|
802
|
+
it('should fetch details for a specific MCP server', async () => {
|
|
803
|
+
mockFetchResponse(mockServerDetail1);
|
|
804
|
+
|
|
805
|
+
const result = await client.getMcpServerDetails(serverId);
|
|
806
|
+
expect(result).toEqual(mockServerDetail1);
|
|
807
|
+
expect(global.fetch).toHaveBeenCalledWith(
|
|
808
|
+
`${clientOptions.baseUrl}/api/mcp/v0/servers/${serverId}`,
|
|
809
|
+
expect.objectContaining({
|
|
810
|
+
headers: expect.objectContaining(clientOptions.headers),
|
|
811
|
+
}),
|
|
812
|
+
);
|
|
813
|
+
});
|
|
814
|
+
|
|
815
|
+
it('should fetch MCP server details with a version parameter', async () => {
|
|
816
|
+
mockFetchResponse(mockServerDetail1);
|
|
817
|
+
const version = '1.0.0';
|
|
818
|
+
|
|
819
|
+
const result = await client.getMcpServerDetails(serverId, { version });
|
|
820
|
+
expect(result).toEqual(mockServerDetail1);
|
|
821
|
+
expect(global.fetch).toHaveBeenCalledWith(
|
|
822
|
+
`${clientOptions.baseUrl}/api/mcp/v0/servers/${serverId}?version=${version}`,
|
|
823
|
+
expect.objectContaining({
|
|
824
|
+
headers: expect.objectContaining(clientOptions.headers),
|
|
825
|
+
}),
|
|
826
|
+
);
|
|
827
|
+
});
|
|
828
|
+
});
|
|
829
|
+
});
|
|
710
830
|
});
|