@mastra/client-js 0.0.0-switch-to-core-20250424015131 → 0.0.0-vector-query-sources-20250516172905

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/dist/index.cjs CHANGED
@@ -1,10 +1,206 @@
1
1
  'use strict';
2
2
 
3
- var zod = require('zod');
4
- var zodToJsonSchema = require('zod-to-json-schema');
3
+ var client = require('@ag-ui/client');
4
+ var rxjs = require('rxjs');
5
5
  var uiUtils = require('@ai-sdk/ui-utils');
6
+ var zod = require('zod');
7
+ var originalZodToJsonSchema = require('zod-to-json-schema');
6
8
 
7
- // src/resources/agent.ts
9
+ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
10
+
11
+ var originalZodToJsonSchema__default = /*#__PURE__*/_interopDefault(originalZodToJsonSchema);
12
+
13
+ // src/adapters/agui.ts
14
+ var AGUIAdapter = class extends client.AbstractAgent {
15
+ agent;
16
+ resourceId;
17
+ constructor({ agent, agentId, resourceId, ...rest }) {
18
+ super({
19
+ agentId,
20
+ ...rest
21
+ });
22
+ this.agent = agent;
23
+ this.resourceId = resourceId;
24
+ }
25
+ run(input) {
26
+ return new rxjs.Observable((subscriber) => {
27
+ const convertedMessages = convertMessagesToMastraMessages(input.messages);
28
+ subscriber.next({
29
+ type: client.EventType.RUN_STARTED,
30
+ threadId: input.threadId,
31
+ runId: input.runId
32
+ });
33
+ this.agent.stream({
34
+ threadId: input.threadId,
35
+ resourceId: this.resourceId ?? "",
36
+ runId: input.runId,
37
+ messages: convertedMessages,
38
+ clientTools: input.tools.reduce(
39
+ (acc, tool) => {
40
+ acc[tool.name] = {
41
+ id: tool.name,
42
+ description: tool.description,
43
+ inputSchema: tool.parameters
44
+ };
45
+ return acc;
46
+ },
47
+ {}
48
+ )
49
+ }).then((response) => {
50
+ let currentMessageId = void 0;
51
+ let isInTextMessage = false;
52
+ return response.processDataStream({
53
+ onTextPart: (text) => {
54
+ if (currentMessageId === void 0) {
55
+ currentMessageId = generateUUID();
56
+ const message2 = {
57
+ type: client.EventType.TEXT_MESSAGE_START,
58
+ messageId: currentMessageId,
59
+ role: "assistant"
60
+ };
61
+ subscriber.next(message2);
62
+ isInTextMessage = true;
63
+ }
64
+ const message = {
65
+ type: client.EventType.TEXT_MESSAGE_CONTENT,
66
+ messageId: currentMessageId,
67
+ delta: text
68
+ };
69
+ subscriber.next(message);
70
+ },
71
+ onFinishMessagePart: () => {
72
+ if (currentMessageId !== void 0) {
73
+ const message = {
74
+ type: client.EventType.TEXT_MESSAGE_END,
75
+ messageId: currentMessageId
76
+ };
77
+ subscriber.next(message);
78
+ isInTextMessage = false;
79
+ }
80
+ subscriber.next({
81
+ type: client.EventType.RUN_FINISHED,
82
+ threadId: input.threadId,
83
+ runId: input.runId
84
+ });
85
+ subscriber.complete();
86
+ },
87
+ onToolCallPart(streamPart) {
88
+ const parentMessageId = currentMessageId || generateUUID();
89
+ if (isInTextMessage) {
90
+ const message = {
91
+ type: client.EventType.TEXT_MESSAGE_END,
92
+ messageId: parentMessageId
93
+ };
94
+ subscriber.next(message);
95
+ isInTextMessage = false;
96
+ }
97
+ subscriber.next({
98
+ type: client.EventType.TOOL_CALL_START,
99
+ toolCallId: streamPart.toolCallId,
100
+ toolCallName: streamPart.toolName,
101
+ parentMessageId
102
+ });
103
+ subscriber.next({
104
+ type: client.EventType.TOOL_CALL_ARGS,
105
+ toolCallId: streamPart.toolCallId,
106
+ delta: JSON.stringify(streamPart.args),
107
+ parentMessageId
108
+ });
109
+ subscriber.next({
110
+ type: client.EventType.TOOL_CALL_END,
111
+ toolCallId: streamPart.toolCallId,
112
+ parentMessageId
113
+ });
114
+ }
115
+ });
116
+ }).catch((error) => {
117
+ console.error("error", error);
118
+ subscriber.error(error);
119
+ });
120
+ return () => {
121
+ };
122
+ });
123
+ }
124
+ };
125
+ function generateUUID() {
126
+ if (typeof crypto !== "undefined") {
127
+ if (typeof crypto.randomUUID === "function") {
128
+ return crypto.randomUUID();
129
+ }
130
+ if (typeof crypto.getRandomValues === "function") {
131
+ const buffer = new Uint8Array(16);
132
+ crypto.getRandomValues(buffer);
133
+ buffer[6] = buffer[6] & 15 | 64;
134
+ buffer[8] = buffer[8] & 63 | 128;
135
+ let hex = "";
136
+ for (let i = 0; i < 16; i++) {
137
+ hex += buffer[i].toString(16).padStart(2, "0");
138
+ if (i === 3 || i === 5 || i === 7 || i === 9) hex += "-";
139
+ }
140
+ return hex;
141
+ }
142
+ }
143
+ return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
144
+ const r = Math.random() * 16 | 0;
145
+ const v = c === "x" ? r : r & 3 | 8;
146
+ return v.toString(16);
147
+ });
148
+ }
149
+ function convertMessagesToMastraMessages(messages) {
150
+ const result = [];
151
+ for (const message of messages) {
152
+ if (message.role === "assistant") {
153
+ const parts = message.content ? [{ type: "text", text: message.content }] : [];
154
+ for (const toolCall of message.toolCalls ?? []) {
155
+ parts.push({
156
+ type: "tool-call",
157
+ toolCallId: toolCall.id,
158
+ toolName: toolCall.function.name,
159
+ args: JSON.parse(toolCall.function.arguments)
160
+ });
161
+ }
162
+ result.push({
163
+ role: "assistant",
164
+ content: parts
165
+ });
166
+ if (message.toolCalls?.length) {
167
+ result.push({
168
+ role: "tool",
169
+ content: message.toolCalls.map((toolCall) => ({
170
+ type: "tool-result",
171
+ toolCallId: toolCall.id,
172
+ toolName: toolCall.function.name,
173
+ result: JSON.parse(toolCall.function.arguments)
174
+ }))
175
+ });
176
+ }
177
+ } else if (message.role === "user") {
178
+ result.push({
179
+ role: "user",
180
+ content: message.content || ""
181
+ });
182
+ } else if (message.role === "tool") {
183
+ result.push({
184
+ role: "tool",
185
+ content: [
186
+ {
187
+ type: "tool-result",
188
+ toolCallId: message.toolCallId,
189
+ toolName: "unknown",
190
+ result: message.content
191
+ }
192
+ ]
193
+ });
194
+ }
195
+ }
196
+ return result;
197
+ }
198
+ function zodToJsonSchema(zodSchema) {
199
+ if (!(zodSchema instanceof zod.ZodSchema)) {
200
+ return zodSchema;
201
+ }
202
+ return originalZodToJsonSchema__default.default(zodSchema, { $refStrategy: "none" });
203
+ }
8
204
 
9
205
  // src/resources/base.ts
10
206
  var BaseResource = class {
@@ -24,7 +220,7 @@ var BaseResource = class {
24
220
  let delay = backoffMs;
25
221
  for (let attempt = 0; attempt <= retries; attempt++) {
26
222
  try {
27
- const response = await fetch(`${baseUrl}${path}`, {
223
+ const response = await fetch(`${baseUrl.replace(/\/$/, "")}${path}`, {
28
224
  ...options,
29
225
  headers: {
30
226
  ...headers,
@@ -135,8 +331,9 @@ var Agent = class extends BaseResource {
135
331
  generate(params) {
136
332
  const processedParams = {
137
333
  ...params,
138
- output: params.output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.output) : params.output,
139
- experimental_output: params.experimental_output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.experimental_output) : params.experimental_output
334
+ output: params.output ? zodToJsonSchema(params.output) : void 0,
335
+ experimental_output: params.experimental_output ? zodToJsonSchema(params.experimental_output) : void 0,
336
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0
140
337
  };
141
338
  return this.request(`/api/agents/${this.agentId}/generate`, {
142
339
  method: "POST",
@@ -151,8 +348,9 @@ var Agent = class extends BaseResource {
151
348
  async stream(params) {
152
349
  const processedParams = {
153
350
  ...params,
154
- output: params.output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.output) : params.output,
155
- experimental_output: params.experimental_output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.experimental_output) : params.experimental_output
351
+ output: params.output ? zodToJsonSchema(params.output) : void 0,
352
+ experimental_output: params.experimental_output ? zodToJsonSchema(params.experimental_output) : void 0,
353
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0
156
354
  };
157
355
  const response = await this.request(`/api/agents/${this.agentId}/stream`, {
158
356
  method: "POST",
@@ -178,6 +376,22 @@ var Agent = class extends BaseResource {
178
376
  getTool(toolId) {
179
377
  return this.request(`/api/agents/${this.agentId}/tools/${toolId}`);
180
378
  }
379
+ /**
380
+ * Executes a tool for the agent
381
+ * @param toolId - ID of the tool to execute
382
+ * @param params - Parameters required for tool execution
383
+ * @returns Promise containing the tool execution results
384
+ */
385
+ executeTool(toolId, params) {
386
+ const body = {
387
+ data: params.data,
388
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0
389
+ };
390
+ return this.request(`/api/agents/${this.agentId}/tools/${toolId}/execute`, {
391
+ method: "POST",
392
+ body
393
+ });
394
+ }
181
395
  /**
182
396
  * Retrieves evaluation results for the agent
183
397
  * @returns Promise containing agent evaluations
@@ -213,8 +427,8 @@ var Network = class extends BaseResource {
213
427
  generate(params) {
214
428
  const processedParams = {
215
429
  ...params,
216
- output: params.output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.output) : params.output,
217
- experimental_output: params.experimental_output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.experimental_output) : params.experimental_output
430
+ output: zodToJsonSchema(params.output),
431
+ experimental_output: zodToJsonSchema(params.experimental_output)
218
432
  };
219
433
  return this.request(`/api/networks/${this.networkId}/generate`, {
220
434
  method: "POST",
@@ -229,8 +443,8 @@ var Network = class extends BaseResource {
229
443
  async stream(params) {
230
444
  const processedParams = {
231
445
  ...params,
232
- output: params.output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.output) : params.output,
233
- experimental_output: params.experimental_output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.experimental_output) : params.experimental_output
446
+ output: zodToJsonSchema(params.output),
447
+ experimental_output: zodToJsonSchema(params.experimental_output)
234
448
  };
235
449
  const response = await this.request(`/api/networks/${this.networkId}/stream`, {
236
450
  method: "POST",
@@ -286,10 +500,15 @@ var MemoryThread = class extends BaseResource {
286
500
  }
287
501
  /**
288
502
  * Retrieves messages associated with the thread
503
+ * @param params - Optional parameters including limit for number of messages to retrieve
289
504
  * @returns Promise containing thread messages and UI messages
290
505
  */
291
- getMessages() {
292
- return this.request(`/api/memory/threads/${this.threadId}/messages?agentId=${this.agentId}`);
506
+ getMessages(params) {
507
+ const query = new URLSearchParams({
508
+ agentId: this.agentId,
509
+ ...params?.limit ? { limit: params.limit.toString() } : {}
510
+ });
511
+ return this.request(`/api/memory/threads/${this.threadId}/messages?${query.toString()}`);
293
512
  }
294
513
  };
295
514
 
@@ -373,6 +592,34 @@ var Workflow = class extends BaseResource {
373
592
  details() {
374
593
  return this.request(`/api/workflows/${this.workflowId}`);
375
594
  }
595
+ /**
596
+ * Retrieves all runs for a workflow
597
+ * @param params - Parameters for filtering runs
598
+ * @returns Promise containing workflow runs array
599
+ */
600
+ runs(params) {
601
+ const searchParams = new URLSearchParams();
602
+ if (params?.fromDate) {
603
+ searchParams.set("fromDate", params.fromDate.toISOString());
604
+ }
605
+ if (params?.toDate) {
606
+ searchParams.set("toDate", params.toDate.toISOString());
607
+ }
608
+ if (params?.limit) {
609
+ searchParams.set("limit", String(params.limit));
610
+ }
611
+ if (params?.offset) {
612
+ searchParams.set("offset", String(params.offset));
613
+ }
614
+ if (params?.resourceId) {
615
+ searchParams.set("resourceId", params.resourceId);
616
+ }
617
+ if (searchParams.size) {
618
+ return this.request(`/api/workflows/${this.workflowId}/runs?${searchParams}`);
619
+ } else {
620
+ return this.request(`/api/workflows/${this.workflowId}/runs`);
621
+ }
622
+ }
376
623
  /**
377
624
  * @deprecated Use `startAsync` instead
378
625
  * Executes the workflow with the provided parameters
@@ -489,7 +736,7 @@ var Workflow = class extends BaseResource {
489
736
  }
490
737
  }
491
738
  }
492
- } catch (error) {
739
+ } catch {
493
740
  }
494
741
  }
495
742
  if (buffer) {
@@ -543,9 +790,313 @@ var Tool = class extends BaseResource {
543
790
  * @returns Promise containing the tool execution results
544
791
  */
545
792
  execute(params) {
546
- return this.request(`/api/tools/${this.toolId}/execute`, {
793
+ const url = new URLSearchParams();
794
+ if (params.runId) {
795
+ url.set("runId", params.runId);
796
+ }
797
+ const body = {
798
+ data: params.data,
799
+ runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0
800
+ };
801
+ return this.request(`/api/tools/${this.toolId}/execute?${url.toString()}`, {
547
802
  method: "POST",
548
- body: params
803
+ body
804
+ });
805
+ }
806
+ };
807
+
808
+ // src/resources/vnext-workflow.ts
809
+ var RECORD_SEPARATOR2 = "";
810
+ var VNextWorkflow = class extends BaseResource {
811
+ constructor(options, workflowId) {
812
+ super(options);
813
+ this.workflowId = workflowId;
814
+ }
815
+ /**
816
+ * Creates an async generator that processes a readable stream and yields vNext workflow records
817
+ * separated by the Record Separator character (\x1E)
818
+ *
819
+ * @param stream - The readable stream to process
820
+ * @returns An async generator that yields parsed records
821
+ */
822
+ async *streamProcessor(stream) {
823
+ const reader = stream.getReader();
824
+ let doneReading = false;
825
+ let buffer = "";
826
+ try {
827
+ while (!doneReading) {
828
+ const { done, value } = await reader.read();
829
+ doneReading = done;
830
+ if (done && !value) continue;
831
+ try {
832
+ const decoded = value ? new TextDecoder().decode(value) : "";
833
+ const chunks = (buffer + decoded).split(RECORD_SEPARATOR2);
834
+ buffer = chunks.pop() || "";
835
+ for (const chunk of chunks) {
836
+ if (chunk) {
837
+ if (typeof chunk === "string") {
838
+ try {
839
+ const parsedChunk = JSON.parse(chunk);
840
+ yield parsedChunk;
841
+ } catch {
842
+ }
843
+ }
844
+ }
845
+ }
846
+ } catch {
847
+ }
848
+ }
849
+ if (buffer) {
850
+ try {
851
+ yield JSON.parse(buffer);
852
+ } catch {
853
+ }
854
+ }
855
+ } finally {
856
+ reader.cancel().catch(() => {
857
+ });
858
+ }
859
+ }
860
+ /**
861
+ * Retrieves details about the vNext workflow
862
+ * @returns Promise containing vNext workflow details including steps and graphs
863
+ */
864
+ details() {
865
+ return this.request(`/api/workflows/v-next/${this.workflowId}`);
866
+ }
867
+ /**
868
+ * Retrieves all runs for a vNext workflow
869
+ * @param params - Parameters for filtering runs
870
+ * @returns Promise containing vNext workflow runs array
871
+ */
872
+ runs(params) {
873
+ const searchParams = new URLSearchParams();
874
+ if (params?.fromDate) {
875
+ searchParams.set("fromDate", params.fromDate.toISOString());
876
+ }
877
+ if (params?.toDate) {
878
+ searchParams.set("toDate", params.toDate.toISOString());
879
+ }
880
+ if (params?.limit) {
881
+ searchParams.set("limit", String(params.limit));
882
+ }
883
+ if (params?.offset) {
884
+ searchParams.set("offset", String(params.offset));
885
+ }
886
+ if (params?.resourceId) {
887
+ searchParams.set("resourceId", params.resourceId);
888
+ }
889
+ if (searchParams.size) {
890
+ return this.request(`/api/workflows/v-next/${this.workflowId}/runs?${searchParams}`);
891
+ } else {
892
+ return this.request(`/api/workflows/v-next/${this.workflowId}/runs`);
893
+ }
894
+ }
895
+ /**
896
+ * Creates a new vNext workflow run
897
+ * @param params - Optional object containing the optional runId
898
+ * @returns Promise containing the runId of the created run
899
+ */
900
+ createRun(params) {
901
+ const searchParams = new URLSearchParams();
902
+ if (!!params?.runId) {
903
+ searchParams.set("runId", params.runId);
904
+ }
905
+ return this.request(`/api/workflows/v-next/${this.workflowId}/create-run?${searchParams.toString()}`, {
906
+ method: "POST"
907
+ });
908
+ }
909
+ /**
910
+ * Starts a vNext workflow run synchronously without waiting for the workflow to complete
911
+ * @param params - Object containing the runId, inputData and runtimeContext
912
+ * @returns Promise containing success message
913
+ */
914
+ start(params) {
915
+ const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0;
916
+ return this.request(`/api/workflows/v-next/${this.workflowId}/start?runId=${params.runId}`, {
917
+ method: "POST",
918
+ body: { inputData: params?.inputData, runtimeContext }
919
+ });
920
+ }
921
+ /**
922
+ * Resumes a suspended vNext workflow step synchronously without waiting for the vNext workflow to complete
923
+ * @param params - Object containing the runId, step, resumeData and runtimeContext
924
+ * @returns Promise containing success message
925
+ */
926
+ resume({
927
+ step,
928
+ runId,
929
+ resumeData,
930
+ ...rest
931
+ }) {
932
+ const runtimeContext = rest.runtimeContext ? Object.fromEntries(rest.runtimeContext.entries()) : void 0;
933
+ return this.request(`/api/workflows/v-next/${this.workflowId}/resume?runId=${runId}`, {
934
+ method: "POST",
935
+ stream: true,
936
+ body: {
937
+ step,
938
+ resumeData,
939
+ runtimeContext
940
+ }
941
+ });
942
+ }
943
+ /**
944
+ * Starts a vNext workflow run asynchronously and returns a promise that resolves when the vNext workflow is complete
945
+ * @param params - Object containing the optional runId, inputData and runtimeContext
946
+ * @returns Promise containing the vNext workflow execution results
947
+ */
948
+ startAsync(params) {
949
+ const searchParams = new URLSearchParams();
950
+ if (!!params?.runId) {
951
+ searchParams.set("runId", params.runId);
952
+ }
953
+ const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0;
954
+ return this.request(`/api/workflows/v-next/${this.workflowId}/start-async?${searchParams.toString()}`, {
955
+ method: "POST",
956
+ body: { inputData: params.inputData, runtimeContext }
957
+ });
958
+ }
959
+ /**
960
+ * Resumes a suspended vNext workflow step asynchronously and returns a promise that resolves when the vNext workflow is complete
961
+ * @param params - Object containing the runId, step, resumeData and runtimeContext
962
+ * @returns Promise containing the vNext workflow resume results
963
+ */
964
+ resumeAsync(params) {
965
+ const runtimeContext = params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0;
966
+ return this.request(`/api/workflows/v-next/${this.workflowId}/resume-async?runId=${params.runId}`, {
967
+ method: "POST",
968
+ body: {
969
+ step: params.step,
970
+ resumeData: params.resumeData,
971
+ runtimeContext
972
+ }
973
+ });
974
+ }
975
+ /**
976
+ * Watches vNext workflow transitions in real-time
977
+ * @param runId - Optional run ID to filter the watch stream
978
+ * @returns AsyncGenerator that yields parsed records from the vNext workflow watch stream
979
+ */
980
+ async watch({ runId }, onRecord) {
981
+ const response = await this.request(`/api/workflows/v-next/${this.workflowId}/watch?runId=${runId}`, {
982
+ stream: true
983
+ });
984
+ if (!response.ok) {
985
+ throw new Error(`Failed to watch vNext workflow: ${response.statusText}`);
986
+ }
987
+ if (!response.body) {
988
+ throw new Error("Response body is null");
989
+ }
990
+ for await (const record of this.streamProcessor(response.body)) {
991
+ onRecord(record);
992
+ }
993
+ }
994
+ };
995
+
996
+ // src/resources/a2a.ts
997
+ var A2A = class extends BaseResource {
998
+ constructor(options, agentId) {
999
+ super(options);
1000
+ this.agentId = agentId;
1001
+ }
1002
+ /**
1003
+ * Get the agent card with metadata about the agent
1004
+ * @returns Promise containing the agent card information
1005
+ */
1006
+ async getCard() {
1007
+ return this.request(`/.well-known/${this.agentId}/agent.json`);
1008
+ }
1009
+ /**
1010
+ * Send a message to the agent and get a response
1011
+ * @param params - Parameters for the task
1012
+ * @returns Promise containing the task response
1013
+ */
1014
+ async sendMessage(params) {
1015
+ const response = await this.request(`/a2a/${this.agentId}`, {
1016
+ method: "POST",
1017
+ body: {
1018
+ method: "tasks/send",
1019
+ params
1020
+ }
1021
+ });
1022
+ return { task: response.result };
1023
+ }
1024
+ /**
1025
+ * Get the status and result of a task
1026
+ * @param params - Parameters for querying the task
1027
+ * @returns Promise containing the task response
1028
+ */
1029
+ async getTask(params) {
1030
+ const response = await this.request(`/a2a/${this.agentId}`, {
1031
+ method: "POST",
1032
+ body: {
1033
+ method: "tasks/get",
1034
+ params
1035
+ }
1036
+ });
1037
+ return response.result;
1038
+ }
1039
+ /**
1040
+ * Cancel a running task
1041
+ * @param params - Parameters identifying the task to cancel
1042
+ * @returns Promise containing the task response
1043
+ */
1044
+ async cancelTask(params) {
1045
+ return this.request(`/a2a/${this.agentId}`, {
1046
+ method: "POST",
1047
+ body: {
1048
+ method: "tasks/cancel",
1049
+ params
1050
+ }
1051
+ });
1052
+ }
1053
+ /**
1054
+ * Send a message and subscribe to streaming updates (not fully implemented)
1055
+ * @param params - Parameters for the task
1056
+ * @returns Promise containing the task response
1057
+ */
1058
+ async sendAndSubscribe(params) {
1059
+ return this.request(`/a2a/${this.agentId}`, {
1060
+ method: "POST",
1061
+ body: {
1062
+ method: "tasks/sendSubscribe",
1063
+ params
1064
+ },
1065
+ stream: true
1066
+ });
1067
+ }
1068
+ };
1069
+
1070
+ // src/resources/mcp-tool.ts
1071
+ var MCPTool = class extends BaseResource {
1072
+ serverId;
1073
+ toolId;
1074
+ constructor(options, serverId, toolId) {
1075
+ super(options);
1076
+ this.serverId = serverId;
1077
+ this.toolId = toolId;
1078
+ }
1079
+ /**
1080
+ * Retrieves details about this specific tool from the MCP server.
1081
+ * @returns Promise containing the tool's information (name, description, schema).
1082
+ */
1083
+ details() {
1084
+ return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}`);
1085
+ }
1086
+ /**
1087
+ * Executes this specific tool on the MCP server.
1088
+ * @param params - Parameters for tool execution, including data/args and optional runtimeContext.
1089
+ * @returns Promise containing the result of the tool execution.
1090
+ */
1091
+ execute(params) {
1092
+ const body = {};
1093
+ if (params.data !== void 0) body.data = params.data;
1094
+ if (params.runtimeContext !== void 0) {
1095
+ body.runtimeContext = params.runtimeContext;
1096
+ }
1097
+ return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}/execute`, {
1098
+ method: "POST",
1099
+ body: Object.keys(body).length > 0 ? body : void 0
549
1100
  });
550
1101
  }
551
1102
  };
@@ -562,6 +1113,21 @@ var MastraClient = class extends BaseResource {
562
1113
  getAgents() {
563
1114
  return this.request("/api/agents");
564
1115
  }
1116
+ async getAGUI({ resourceId }) {
1117
+ const agents = await this.getAgents();
1118
+ return Object.entries(agents).reduce(
1119
+ (acc, [agentId]) => {
1120
+ const agent = this.getAgent(agentId);
1121
+ acc[agentId] = new AGUIAdapter({
1122
+ agentId,
1123
+ agent,
1124
+ resourceId
1125
+ });
1126
+ return acc;
1127
+ },
1128
+ {}
1129
+ );
1130
+ }
565
1131
  /**
566
1132
  * Gets an agent instance by ID
567
1133
  * @param agentId - ID of the agent to retrieve
@@ -642,6 +1208,21 @@ var MastraClient = class extends BaseResource {
642
1208
  getWorkflow(workflowId) {
643
1209
  return new Workflow(this.options, workflowId);
644
1210
  }
1211
+ /**
1212
+ * Retrieves all available vNext workflows
1213
+ * @returns Promise containing map of vNext workflow IDs to vNext workflow details
1214
+ */
1215
+ getVNextWorkflows() {
1216
+ return this.request("/api/workflows/v-next");
1217
+ }
1218
+ /**
1219
+ * Gets a vNext workflow instance by ID
1220
+ * @param workflowId - ID of the vNext workflow to retrieve
1221
+ * @returns vNext Workflow instance
1222
+ */
1223
+ getVNextWorkflow(workflowId) {
1224
+ return new VNextWorkflow(this.options, workflowId);
1225
+ }
645
1226
  /**
646
1227
  * Gets a vector instance by name
647
1228
  * @param vectorName - Name of the vector to retrieve
@@ -679,7 +1260,7 @@ var MastraClient = class extends BaseResource {
679
1260
  * @returns Promise containing telemetry data
680
1261
  */
681
1262
  getTelemetry(params) {
682
- const { name, scope, page, perPage, attribute } = params || {};
1263
+ const { name, scope, page, perPage, attribute, fromDate, toDate } = params || {};
683
1264
  const _attribute = attribute ? Object.entries(attribute).map(([key, value]) => `${key}:${value}`) : [];
684
1265
  const searchParams = new URLSearchParams();
685
1266
  if (name) {
@@ -703,6 +1284,12 @@ var MastraClient = class extends BaseResource {
703
1284
  searchParams.set("attribute", _attribute);
704
1285
  }
705
1286
  }
1287
+ if (fromDate) {
1288
+ searchParams.set("fromDate", fromDate.toISOString());
1289
+ }
1290
+ if (toDate) {
1291
+ searchParams.set("toDate", toDate.toISOString());
1292
+ }
706
1293
  if (searchParams.size) {
707
1294
  return this.request(`/api/telemetry?${searchParams}`);
708
1295
  } else {
@@ -724,6 +1311,62 @@ var MastraClient = class extends BaseResource {
724
1311
  getNetwork(networkId) {
725
1312
  return new Network(this.options, networkId);
726
1313
  }
1314
+ /**
1315
+ * Retrieves a list of available MCP servers.
1316
+ * @param params - Optional parameters for pagination (limit, offset).
1317
+ * @returns Promise containing the list of MCP servers and pagination info.
1318
+ */
1319
+ getMcpServers(params) {
1320
+ const searchParams = new URLSearchParams();
1321
+ if (params?.limit !== void 0) {
1322
+ searchParams.set("limit", String(params.limit));
1323
+ }
1324
+ if (params?.offset !== void 0) {
1325
+ searchParams.set("offset", String(params.offset));
1326
+ }
1327
+ const queryString = searchParams.toString();
1328
+ return this.request(`/api/mcp/v0/servers${queryString ? `?${queryString}` : ""}`);
1329
+ }
1330
+ /**
1331
+ * Retrieves detailed information for a specific MCP server.
1332
+ * @param serverId - The ID of the MCP server to retrieve.
1333
+ * @param params - Optional parameters, e.g., specific version.
1334
+ * @returns Promise containing the detailed MCP server information.
1335
+ */
1336
+ getMcpServerDetails(serverId, params) {
1337
+ const searchParams = new URLSearchParams();
1338
+ if (params?.version) {
1339
+ searchParams.set("version", params.version);
1340
+ }
1341
+ const queryString = searchParams.toString();
1342
+ return this.request(`/api/mcp/v0/servers/${serverId}${queryString ? `?${queryString}` : ""}`);
1343
+ }
1344
+ /**
1345
+ * Retrieves a list of tools for a specific MCP server.
1346
+ * @param serverId - The ID of the MCP server.
1347
+ * @returns Promise containing the list of tools.
1348
+ */
1349
+ getMcpServerTools(serverId) {
1350
+ return this.request(`/api/mcp/${serverId}/tools`);
1351
+ }
1352
+ /**
1353
+ * Gets an MCPTool resource instance for a specific tool on an MCP server.
1354
+ * This instance can then be used to fetch details or execute the tool.
1355
+ * @param serverId - The ID of the MCP server.
1356
+ * @param toolId - The ID of the tool.
1357
+ * @returns MCPTool instance.
1358
+ */
1359
+ getMcpServerTool(serverId, toolId) {
1360
+ return new MCPTool(this.options, serverId, toolId);
1361
+ }
1362
+ /**
1363
+ * Gets an A2A client for interacting with an agent via the A2A protocol
1364
+ * @param agentId - ID of the agent to interact with
1365
+ * @returns A2A client instance
1366
+ */
1367
+ getA2A(agentId) {
1368
+ return new A2A(this.options, agentId);
1369
+ }
727
1370
  };
728
1371
 
729
1372
  exports.MastraClient = MastraClient;