@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/dist/index.js
CHANGED
|
@@ -1,8 +1,10 @@
|
|
|
1
1
|
import { AbstractAgent, EventType } from '@ag-ui/client';
|
|
2
2
|
import { Observable } from 'rxjs';
|
|
3
|
-
import { processDataStream } from '@ai-sdk/ui-utils';
|
|
3
|
+
import { processDataStream, parsePartialJson } from '@ai-sdk/ui-utils';
|
|
4
4
|
import { ZodSchema } from 'zod';
|
|
5
5
|
import originalZodToJsonSchema from 'zod-to-json-schema';
|
|
6
|
+
import { isVercelTool } from '@mastra/core/tools';
|
|
7
|
+
import { RuntimeContext } from '@mastra/core/runtime-context';
|
|
6
8
|
|
|
7
9
|
// src/adapters/agui.ts
|
|
8
10
|
var AGUIAdapter = class extends AbstractAgent {
|
|
@@ -195,6 +197,33 @@ function zodToJsonSchema(zodSchema) {
|
|
|
195
197
|
}
|
|
196
198
|
return originalZodToJsonSchema(zodSchema, { $refStrategy: "none" });
|
|
197
199
|
}
|
|
200
|
+
function processClientTools(clientTools) {
|
|
201
|
+
if (!clientTools) {
|
|
202
|
+
return void 0;
|
|
203
|
+
}
|
|
204
|
+
return Object.fromEntries(
|
|
205
|
+
Object.entries(clientTools).map(([key, value]) => {
|
|
206
|
+
if (isVercelTool(value)) {
|
|
207
|
+
return [
|
|
208
|
+
key,
|
|
209
|
+
{
|
|
210
|
+
...value,
|
|
211
|
+
parameters: value.parameters ? zodToJsonSchema(value.parameters) : void 0
|
|
212
|
+
}
|
|
213
|
+
];
|
|
214
|
+
} else {
|
|
215
|
+
return [
|
|
216
|
+
key,
|
|
217
|
+
{
|
|
218
|
+
...value,
|
|
219
|
+
inputSchema: value.inputSchema ? zodToJsonSchema(value.inputSchema) : void 0,
|
|
220
|
+
outputSchema: value.outputSchema ? zodToJsonSchema(value.outputSchema) : void 0
|
|
221
|
+
}
|
|
222
|
+
];
|
|
223
|
+
}
|
|
224
|
+
})
|
|
225
|
+
);
|
|
226
|
+
}
|
|
198
227
|
|
|
199
228
|
// src/resources/base.ts
|
|
200
229
|
var BaseResource = class {
|
|
@@ -214,9 +243,10 @@ var BaseResource = class {
|
|
|
214
243
|
let delay = backoffMs;
|
|
215
244
|
for (let attempt = 0; attempt <= retries; attempt++) {
|
|
216
245
|
try {
|
|
217
|
-
const response = await fetch(`${baseUrl}${path}`, {
|
|
246
|
+
const response = await fetch(`${baseUrl.replace(/\/$/, "")}${path}`, {
|
|
218
247
|
...options,
|
|
219
248
|
headers: {
|
|
249
|
+
...options.method === "POST" || options.method === "PUT" ? { "content-type": "application/json" } : {},
|
|
220
250
|
...headers,
|
|
221
251
|
...options.headers
|
|
222
252
|
// TODO: Bring this back once we figure out what we/users need to do to make this work with cross-origin requests
|
|
@@ -254,8 +284,15 @@ var BaseResource = class {
|
|
|
254
284
|
throw lastError || new Error("Request failed");
|
|
255
285
|
}
|
|
256
286
|
};
|
|
257
|
-
|
|
258
|
-
|
|
287
|
+
function parseClientRuntimeContext(runtimeContext) {
|
|
288
|
+
if (runtimeContext) {
|
|
289
|
+
if (runtimeContext instanceof RuntimeContext) {
|
|
290
|
+
return Object.fromEntries(runtimeContext.entries());
|
|
291
|
+
}
|
|
292
|
+
return runtimeContext;
|
|
293
|
+
}
|
|
294
|
+
return void 0;
|
|
295
|
+
}
|
|
259
296
|
var AgentVoice = class extends BaseResource {
|
|
260
297
|
constructor(options, agentId) {
|
|
261
298
|
super(options);
|
|
@@ -302,6 +339,13 @@ var AgentVoice = class extends BaseResource {
|
|
|
302
339
|
getSpeakers() {
|
|
303
340
|
return this.request(`/api/agents/${this.agentId}/voice/speakers`);
|
|
304
341
|
}
|
|
342
|
+
/**
|
|
343
|
+
* Get the listener configuration for the agent's voice provider
|
|
344
|
+
* @returns Promise containing a check if the agent has listening capabilities
|
|
345
|
+
*/
|
|
346
|
+
getListener() {
|
|
347
|
+
return this.request(`/api/agents/${this.agentId}/voice/listener`);
|
|
348
|
+
}
|
|
305
349
|
};
|
|
306
350
|
var Agent = class extends BaseResource {
|
|
307
351
|
constructor(options, agentId) {
|
|
@@ -317,22 +361,318 @@ var Agent = class extends BaseResource {
|
|
|
317
361
|
details() {
|
|
318
362
|
return this.request(`/api/agents/${this.agentId}`);
|
|
319
363
|
}
|
|
320
|
-
|
|
321
|
-
* Generates a response from the agent
|
|
322
|
-
* @param params - Generation parameters including prompt
|
|
323
|
-
* @returns Promise containing the generated response
|
|
324
|
-
*/
|
|
325
|
-
generate(params) {
|
|
364
|
+
async generate(params) {
|
|
326
365
|
const processedParams = {
|
|
327
366
|
...params,
|
|
328
|
-
output: zodToJsonSchema(params.output),
|
|
329
|
-
experimental_output: zodToJsonSchema(params.experimental_output),
|
|
330
|
-
runtimeContext:
|
|
367
|
+
output: params.output ? zodToJsonSchema(params.output) : void 0,
|
|
368
|
+
experimental_output: params.experimental_output ? zodToJsonSchema(params.experimental_output) : void 0,
|
|
369
|
+
runtimeContext: parseClientRuntimeContext(params.runtimeContext),
|
|
370
|
+
clientTools: processClientTools(params.clientTools)
|
|
331
371
|
};
|
|
332
|
-
|
|
372
|
+
const { runId, resourceId, threadId, runtimeContext } = processedParams;
|
|
373
|
+
const response = await this.request(`/api/agents/${this.agentId}/generate`, {
|
|
333
374
|
method: "POST",
|
|
334
375
|
body: processedParams
|
|
335
376
|
});
|
|
377
|
+
if (response.finishReason === "tool-calls") {
|
|
378
|
+
for (const toolCall of response.toolCalls) {
|
|
379
|
+
const clientTool = params.clientTools?.[toolCall.toolName];
|
|
380
|
+
if (clientTool && clientTool.execute) {
|
|
381
|
+
const result = await clientTool.execute(
|
|
382
|
+
{ context: toolCall?.args, runId, resourceId, threadId, runtimeContext },
|
|
383
|
+
{
|
|
384
|
+
messages: response.messages,
|
|
385
|
+
toolCallId: toolCall?.toolCallId
|
|
386
|
+
}
|
|
387
|
+
);
|
|
388
|
+
const updatedMessages = [
|
|
389
|
+
{
|
|
390
|
+
role: "user",
|
|
391
|
+
content: params.messages
|
|
392
|
+
},
|
|
393
|
+
...response.response.messages,
|
|
394
|
+
{
|
|
395
|
+
role: "tool",
|
|
396
|
+
content: [
|
|
397
|
+
{
|
|
398
|
+
type: "tool-result",
|
|
399
|
+
toolCallId: toolCall.toolCallId,
|
|
400
|
+
toolName: toolCall.toolName,
|
|
401
|
+
result
|
|
402
|
+
}
|
|
403
|
+
]
|
|
404
|
+
}
|
|
405
|
+
];
|
|
406
|
+
return this.generate({
|
|
407
|
+
...params,
|
|
408
|
+
messages: updatedMessages
|
|
409
|
+
});
|
|
410
|
+
}
|
|
411
|
+
}
|
|
412
|
+
}
|
|
413
|
+
return response;
|
|
414
|
+
}
|
|
415
|
+
async processChatResponse({
|
|
416
|
+
stream,
|
|
417
|
+
update,
|
|
418
|
+
onToolCall,
|
|
419
|
+
onFinish,
|
|
420
|
+
getCurrentDate = () => /* @__PURE__ */ new Date(),
|
|
421
|
+
lastMessage
|
|
422
|
+
}) {
|
|
423
|
+
const replaceLastMessage = lastMessage?.role === "assistant";
|
|
424
|
+
let step = replaceLastMessage ? 1 + // find max step in existing tool invocations:
|
|
425
|
+
(lastMessage.toolInvocations?.reduce((max, toolInvocation) => {
|
|
426
|
+
return Math.max(max, toolInvocation.step ?? 0);
|
|
427
|
+
}, 0) ?? 0) : 0;
|
|
428
|
+
const message = replaceLastMessage ? structuredClone(lastMessage) : {
|
|
429
|
+
id: crypto.randomUUID(),
|
|
430
|
+
createdAt: getCurrentDate(),
|
|
431
|
+
role: "assistant",
|
|
432
|
+
content: "",
|
|
433
|
+
parts: []
|
|
434
|
+
};
|
|
435
|
+
let currentTextPart = void 0;
|
|
436
|
+
let currentReasoningPart = void 0;
|
|
437
|
+
let currentReasoningTextDetail = void 0;
|
|
438
|
+
function updateToolInvocationPart(toolCallId, invocation) {
|
|
439
|
+
const part = message.parts.find(
|
|
440
|
+
(part2) => part2.type === "tool-invocation" && part2.toolInvocation.toolCallId === toolCallId
|
|
441
|
+
);
|
|
442
|
+
if (part != null) {
|
|
443
|
+
part.toolInvocation = invocation;
|
|
444
|
+
} else {
|
|
445
|
+
message.parts.push({
|
|
446
|
+
type: "tool-invocation",
|
|
447
|
+
toolInvocation: invocation
|
|
448
|
+
});
|
|
449
|
+
}
|
|
450
|
+
}
|
|
451
|
+
const data = [];
|
|
452
|
+
let messageAnnotations = replaceLastMessage ? lastMessage?.annotations : void 0;
|
|
453
|
+
const partialToolCalls = {};
|
|
454
|
+
let usage = {
|
|
455
|
+
completionTokens: NaN,
|
|
456
|
+
promptTokens: NaN,
|
|
457
|
+
totalTokens: NaN
|
|
458
|
+
};
|
|
459
|
+
let finishReason = "unknown";
|
|
460
|
+
function execUpdate() {
|
|
461
|
+
const copiedData = [...data];
|
|
462
|
+
if (messageAnnotations?.length) {
|
|
463
|
+
message.annotations = messageAnnotations;
|
|
464
|
+
}
|
|
465
|
+
const copiedMessage = {
|
|
466
|
+
// deep copy the message to ensure that deep changes (msg attachments) are updated
|
|
467
|
+
// with SolidJS. SolidJS uses referential integration of sub-objects to detect changes.
|
|
468
|
+
...structuredClone(message),
|
|
469
|
+
// add a revision id to ensure that the message is updated with SWR. SWR uses a
|
|
470
|
+
// hashing approach by default to detect changes, but it only works for shallow
|
|
471
|
+
// changes. This is why we need to add a revision id to ensure that the message
|
|
472
|
+
// is updated with SWR (without it, the changes get stuck in SWR and are not
|
|
473
|
+
// forwarded to rendering):
|
|
474
|
+
revisionId: crypto.randomUUID()
|
|
475
|
+
};
|
|
476
|
+
update({
|
|
477
|
+
message: copiedMessage,
|
|
478
|
+
data: copiedData,
|
|
479
|
+
replaceLastMessage
|
|
480
|
+
});
|
|
481
|
+
}
|
|
482
|
+
await processDataStream({
|
|
483
|
+
stream,
|
|
484
|
+
onTextPart(value) {
|
|
485
|
+
if (currentTextPart == null) {
|
|
486
|
+
currentTextPart = {
|
|
487
|
+
type: "text",
|
|
488
|
+
text: value
|
|
489
|
+
};
|
|
490
|
+
message.parts.push(currentTextPart);
|
|
491
|
+
} else {
|
|
492
|
+
currentTextPart.text += value;
|
|
493
|
+
}
|
|
494
|
+
message.content += value;
|
|
495
|
+
execUpdate();
|
|
496
|
+
},
|
|
497
|
+
onReasoningPart(value) {
|
|
498
|
+
if (currentReasoningTextDetail == null) {
|
|
499
|
+
currentReasoningTextDetail = { type: "text", text: value };
|
|
500
|
+
if (currentReasoningPart != null) {
|
|
501
|
+
currentReasoningPart.details.push(currentReasoningTextDetail);
|
|
502
|
+
}
|
|
503
|
+
} else {
|
|
504
|
+
currentReasoningTextDetail.text += value;
|
|
505
|
+
}
|
|
506
|
+
if (currentReasoningPart == null) {
|
|
507
|
+
currentReasoningPart = {
|
|
508
|
+
type: "reasoning",
|
|
509
|
+
reasoning: value,
|
|
510
|
+
details: [currentReasoningTextDetail]
|
|
511
|
+
};
|
|
512
|
+
message.parts.push(currentReasoningPart);
|
|
513
|
+
} else {
|
|
514
|
+
currentReasoningPart.reasoning += value;
|
|
515
|
+
}
|
|
516
|
+
message.reasoning = (message.reasoning ?? "") + value;
|
|
517
|
+
execUpdate();
|
|
518
|
+
},
|
|
519
|
+
onReasoningSignaturePart(value) {
|
|
520
|
+
if (currentReasoningTextDetail != null) {
|
|
521
|
+
currentReasoningTextDetail.signature = value.signature;
|
|
522
|
+
}
|
|
523
|
+
},
|
|
524
|
+
onRedactedReasoningPart(value) {
|
|
525
|
+
if (currentReasoningPart == null) {
|
|
526
|
+
currentReasoningPart = {
|
|
527
|
+
type: "reasoning",
|
|
528
|
+
reasoning: "",
|
|
529
|
+
details: []
|
|
530
|
+
};
|
|
531
|
+
message.parts.push(currentReasoningPart);
|
|
532
|
+
}
|
|
533
|
+
currentReasoningPart.details.push({
|
|
534
|
+
type: "redacted",
|
|
535
|
+
data: value.data
|
|
536
|
+
});
|
|
537
|
+
currentReasoningTextDetail = void 0;
|
|
538
|
+
execUpdate();
|
|
539
|
+
},
|
|
540
|
+
onFilePart(value) {
|
|
541
|
+
message.parts.push({
|
|
542
|
+
type: "file",
|
|
543
|
+
mimeType: value.mimeType,
|
|
544
|
+
data: value.data
|
|
545
|
+
});
|
|
546
|
+
execUpdate();
|
|
547
|
+
},
|
|
548
|
+
onSourcePart(value) {
|
|
549
|
+
message.parts.push({
|
|
550
|
+
type: "source",
|
|
551
|
+
source: value
|
|
552
|
+
});
|
|
553
|
+
execUpdate();
|
|
554
|
+
},
|
|
555
|
+
onToolCallStreamingStartPart(value) {
|
|
556
|
+
if (message.toolInvocations == null) {
|
|
557
|
+
message.toolInvocations = [];
|
|
558
|
+
}
|
|
559
|
+
partialToolCalls[value.toolCallId] = {
|
|
560
|
+
text: "",
|
|
561
|
+
step,
|
|
562
|
+
toolName: value.toolName,
|
|
563
|
+
index: message.toolInvocations.length
|
|
564
|
+
};
|
|
565
|
+
const invocation = {
|
|
566
|
+
state: "partial-call",
|
|
567
|
+
step,
|
|
568
|
+
toolCallId: value.toolCallId,
|
|
569
|
+
toolName: value.toolName,
|
|
570
|
+
args: void 0
|
|
571
|
+
};
|
|
572
|
+
message.toolInvocations.push(invocation);
|
|
573
|
+
updateToolInvocationPart(value.toolCallId, invocation);
|
|
574
|
+
execUpdate();
|
|
575
|
+
},
|
|
576
|
+
onToolCallDeltaPart(value) {
|
|
577
|
+
const partialToolCall = partialToolCalls[value.toolCallId];
|
|
578
|
+
partialToolCall.text += value.argsTextDelta;
|
|
579
|
+
const { value: partialArgs } = parsePartialJson(partialToolCall.text);
|
|
580
|
+
const invocation = {
|
|
581
|
+
state: "partial-call",
|
|
582
|
+
step: partialToolCall.step,
|
|
583
|
+
toolCallId: value.toolCallId,
|
|
584
|
+
toolName: partialToolCall.toolName,
|
|
585
|
+
args: partialArgs
|
|
586
|
+
};
|
|
587
|
+
message.toolInvocations[partialToolCall.index] = invocation;
|
|
588
|
+
updateToolInvocationPart(value.toolCallId, invocation);
|
|
589
|
+
execUpdate();
|
|
590
|
+
},
|
|
591
|
+
async onToolCallPart(value) {
|
|
592
|
+
const invocation = {
|
|
593
|
+
state: "call",
|
|
594
|
+
step,
|
|
595
|
+
...value
|
|
596
|
+
};
|
|
597
|
+
if (partialToolCalls[value.toolCallId] != null) {
|
|
598
|
+
message.toolInvocations[partialToolCalls[value.toolCallId].index] = invocation;
|
|
599
|
+
} else {
|
|
600
|
+
if (message.toolInvocations == null) {
|
|
601
|
+
message.toolInvocations = [];
|
|
602
|
+
}
|
|
603
|
+
message.toolInvocations.push(invocation);
|
|
604
|
+
}
|
|
605
|
+
updateToolInvocationPart(value.toolCallId, invocation);
|
|
606
|
+
execUpdate();
|
|
607
|
+
if (onToolCall) {
|
|
608
|
+
const result = await onToolCall({ toolCall: value });
|
|
609
|
+
if (result != null) {
|
|
610
|
+
const invocation2 = {
|
|
611
|
+
state: "result",
|
|
612
|
+
step,
|
|
613
|
+
...value,
|
|
614
|
+
result
|
|
615
|
+
};
|
|
616
|
+
message.toolInvocations[message.toolInvocations.length - 1] = invocation2;
|
|
617
|
+
updateToolInvocationPart(value.toolCallId, invocation2);
|
|
618
|
+
execUpdate();
|
|
619
|
+
}
|
|
620
|
+
}
|
|
621
|
+
},
|
|
622
|
+
onToolResultPart(value) {
|
|
623
|
+
const toolInvocations = message.toolInvocations;
|
|
624
|
+
if (toolInvocations == null) {
|
|
625
|
+
throw new Error("tool_result must be preceded by a tool_call");
|
|
626
|
+
}
|
|
627
|
+
const toolInvocationIndex = toolInvocations.findIndex((invocation2) => invocation2.toolCallId === value.toolCallId);
|
|
628
|
+
if (toolInvocationIndex === -1) {
|
|
629
|
+
throw new Error("tool_result must be preceded by a tool_call with the same toolCallId");
|
|
630
|
+
}
|
|
631
|
+
const invocation = {
|
|
632
|
+
...toolInvocations[toolInvocationIndex],
|
|
633
|
+
state: "result",
|
|
634
|
+
...value
|
|
635
|
+
};
|
|
636
|
+
toolInvocations[toolInvocationIndex] = invocation;
|
|
637
|
+
updateToolInvocationPart(value.toolCallId, invocation);
|
|
638
|
+
execUpdate();
|
|
639
|
+
},
|
|
640
|
+
onDataPart(value) {
|
|
641
|
+
data.push(...value);
|
|
642
|
+
execUpdate();
|
|
643
|
+
},
|
|
644
|
+
onMessageAnnotationsPart(value) {
|
|
645
|
+
if (messageAnnotations == null) {
|
|
646
|
+
messageAnnotations = [...value];
|
|
647
|
+
} else {
|
|
648
|
+
messageAnnotations.push(...value);
|
|
649
|
+
}
|
|
650
|
+
execUpdate();
|
|
651
|
+
},
|
|
652
|
+
onFinishStepPart(value) {
|
|
653
|
+
step += 1;
|
|
654
|
+
currentTextPart = value.isContinued ? currentTextPart : void 0;
|
|
655
|
+
currentReasoningPart = void 0;
|
|
656
|
+
currentReasoningTextDetail = void 0;
|
|
657
|
+
},
|
|
658
|
+
onStartStepPart(value) {
|
|
659
|
+
if (!replaceLastMessage) {
|
|
660
|
+
message.id = value.messageId;
|
|
661
|
+
}
|
|
662
|
+
message.parts.push({ type: "step-start" });
|
|
663
|
+
execUpdate();
|
|
664
|
+
},
|
|
665
|
+
onFinishMessagePart(value) {
|
|
666
|
+
finishReason = value.finishReason;
|
|
667
|
+
if (value.usage != null) {
|
|
668
|
+
usage = value.usage;
|
|
669
|
+
}
|
|
670
|
+
},
|
|
671
|
+
onErrorPart(error) {
|
|
672
|
+
throw new Error(error);
|
|
673
|
+
}
|
|
674
|
+
});
|
|
675
|
+
onFinish?.({ message, finishReason, usage });
|
|
336
676
|
}
|
|
337
677
|
/**
|
|
338
678
|
* Streams a response from the agent
|
|
@@ -342,10 +682,30 @@ var Agent = class extends BaseResource {
|
|
|
342
682
|
async stream(params) {
|
|
343
683
|
const processedParams = {
|
|
344
684
|
...params,
|
|
345
|
-
output: zodToJsonSchema(params.output),
|
|
346
|
-
experimental_output: zodToJsonSchema(params.experimental_output),
|
|
347
|
-
runtimeContext:
|
|
685
|
+
output: params.output ? zodToJsonSchema(params.output) : void 0,
|
|
686
|
+
experimental_output: params.experimental_output ? zodToJsonSchema(params.experimental_output) : void 0,
|
|
687
|
+
runtimeContext: parseClientRuntimeContext(params.runtimeContext),
|
|
688
|
+
clientTools: processClientTools(params.clientTools)
|
|
348
689
|
};
|
|
690
|
+
const { readable, writable } = new TransformStream();
|
|
691
|
+
const response = await this.processStreamResponse(processedParams, writable);
|
|
692
|
+
const streamResponse = new Response(readable, {
|
|
693
|
+
status: response.status,
|
|
694
|
+
statusText: response.statusText,
|
|
695
|
+
headers: response.headers
|
|
696
|
+
});
|
|
697
|
+
streamResponse.processDataStream = async (options = {}) => {
|
|
698
|
+
await processDataStream({
|
|
699
|
+
stream: streamResponse.body,
|
|
700
|
+
...options
|
|
701
|
+
});
|
|
702
|
+
};
|
|
703
|
+
return streamResponse;
|
|
704
|
+
}
|
|
705
|
+
/**
|
|
706
|
+
* Processes the stream response and handles tool calls
|
|
707
|
+
*/
|
|
708
|
+
async processStreamResponse(processedParams, writable) {
|
|
349
709
|
const response = await this.request(`/api/agents/${this.agentId}/stream`, {
|
|
350
710
|
method: "POST",
|
|
351
711
|
body: processedParams,
|
|
@@ -354,12 +714,95 @@ var Agent = class extends BaseResource {
|
|
|
354
714
|
if (!response.body) {
|
|
355
715
|
throw new Error("No response body");
|
|
356
716
|
}
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
|
|
717
|
+
try {
|
|
718
|
+
let toolCalls = [];
|
|
719
|
+
let messages = [];
|
|
720
|
+
const [streamForWritable, streamForProcessing] = response.body.tee();
|
|
721
|
+
streamForWritable.pipeTo(writable, {
|
|
722
|
+
preventClose: true
|
|
723
|
+
}).catch((error) => {
|
|
724
|
+
console.error("Error piping to writable stream:", error);
|
|
361
725
|
});
|
|
362
|
-
|
|
726
|
+
this.processChatResponse({
|
|
727
|
+
stream: streamForProcessing,
|
|
728
|
+
update: ({ message }) => {
|
|
729
|
+
messages.push(message);
|
|
730
|
+
},
|
|
731
|
+
onFinish: async ({ finishReason, message }) => {
|
|
732
|
+
if (finishReason === "tool-calls") {
|
|
733
|
+
const toolCall = [...message?.parts ?? []].reverse().find((part) => part.type === "tool-invocation")?.toolInvocation;
|
|
734
|
+
if (toolCall) {
|
|
735
|
+
toolCalls.push(toolCall);
|
|
736
|
+
}
|
|
737
|
+
for (const toolCall2 of toolCalls) {
|
|
738
|
+
const clientTool = processedParams.clientTools?.[toolCall2.toolName];
|
|
739
|
+
if (clientTool && clientTool.execute) {
|
|
740
|
+
const result = await clientTool.execute(
|
|
741
|
+
{
|
|
742
|
+
context: toolCall2?.args,
|
|
743
|
+
runId: processedParams.runId,
|
|
744
|
+
resourceId: processedParams.resourceId,
|
|
745
|
+
threadId: processedParams.threadId,
|
|
746
|
+
runtimeContext: processedParams.runtimeContext
|
|
747
|
+
},
|
|
748
|
+
{
|
|
749
|
+
messages: response.messages,
|
|
750
|
+
toolCallId: toolCall2?.toolCallId
|
|
751
|
+
}
|
|
752
|
+
);
|
|
753
|
+
const lastMessage = JSON.parse(JSON.stringify(messages[messages.length - 1]));
|
|
754
|
+
const toolInvocationPart = lastMessage?.parts?.find(
|
|
755
|
+
(part) => part.type === "tool-invocation" && part.toolInvocation?.toolCallId === toolCall2.toolCallId
|
|
756
|
+
);
|
|
757
|
+
if (toolInvocationPart) {
|
|
758
|
+
toolInvocationPart.toolInvocation = {
|
|
759
|
+
...toolInvocationPart.toolInvocation,
|
|
760
|
+
state: "result",
|
|
761
|
+
result
|
|
762
|
+
};
|
|
763
|
+
}
|
|
764
|
+
const toolInvocation = lastMessage?.toolInvocations?.find(
|
|
765
|
+
(toolInvocation2) => toolInvocation2.toolCallId === toolCall2.toolCallId
|
|
766
|
+
);
|
|
767
|
+
if (toolInvocation) {
|
|
768
|
+
toolInvocation.state = "result";
|
|
769
|
+
toolInvocation.result = result;
|
|
770
|
+
}
|
|
771
|
+
const writer = writable.getWriter();
|
|
772
|
+
try {
|
|
773
|
+
await writer.write(
|
|
774
|
+
new TextEncoder().encode(
|
|
775
|
+
"a:" + JSON.stringify({
|
|
776
|
+
toolCallId: toolCall2.toolCallId,
|
|
777
|
+
result
|
|
778
|
+
}) + "\n"
|
|
779
|
+
)
|
|
780
|
+
);
|
|
781
|
+
} finally {
|
|
782
|
+
writer.releaseLock();
|
|
783
|
+
}
|
|
784
|
+
const originalMessages = processedParams.messages;
|
|
785
|
+
const messageArray = Array.isArray(originalMessages) ? originalMessages : [originalMessages];
|
|
786
|
+
this.processStreamResponse(
|
|
787
|
+
{
|
|
788
|
+
...processedParams,
|
|
789
|
+
messages: [...messageArray, ...messages, lastMessage]
|
|
790
|
+
},
|
|
791
|
+
writable
|
|
792
|
+
);
|
|
793
|
+
}
|
|
794
|
+
}
|
|
795
|
+
} else {
|
|
796
|
+
setTimeout(() => {
|
|
797
|
+
writable.close();
|
|
798
|
+
}, 0);
|
|
799
|
+
}
|
|
800
|
+
},
|
|
801
|
+
lastMessage: void 0
|
|
802
|
+
});
|
|
803
|
+
} catch (error) {
|
|
804
|
+
console.error("Error processing stream response:", error);
|
|
805
|
+
}
|
|
363
806
|
return response;
|
|
364
807
|
}
|
|
365
808
|
/**
|
|
@@ -494,10 +937,15 @@ var MemoryThread = class extends BaseResource {
|
|
|
494
937
|
}
|
|
495
938
|
/**
|
|
496
939
|
* Retrieves messages associated with the thread
|
|
940
|
+
* @param params - Optional parameters including limit for number of messages to retrieve
|
|
497
941
|
* @returns Promise containing thread messages and UI messages
|
|
498
942
|
*/
|
|
499
|
-
getMessages() {
|
|
500
|
-
|
|
943
|
+
getMessages(params) {
|
|
944
|
+
const query = new URLSearchParams({
|
|
945
|
+
agentId: this.agentId,
|
|
946
|
+
...params?.limit ? { limit: params.limit.toString() } : {}
|
|
947
|
+
});
|
|
948
|
+
return this.request(`/api/memory/threads/${this.threadId}/messages?${query.toString()}`);
|
|
501
949
|
}
|
|
502
950
|
};
|
|
503
951
|
|
|
@@ -567,24 +1015,24 @@ var Vector = class extends BaseResource {
|
|
|
567
1015
|
}
|
|
568
1016
|
};
|
|
569
1017
|
|
|
570
|
-
// src/resources/workflow.ts
|
|
1018
|
+
// src/resources/legacy-workflow.ts
|
|
571
1019
|
var RECORD_SEPARATOR = "";
|
|
572
|
-
var
|
|
1020
|
+
var LegacyWorkflow = class extends BaseResource {
|
|
573
1021
|
constructor(options, workflowId) {
|
|
574
1022
|
super(options);
|
|
575
1023
|
this.workflowId = workflowId;
|
|
576
1024
|
}
|
|
577
1025
|
/**
|
|
578
|
-
* Retrieves details about the workflow
|
|
579
|
-
* @returns Promise containing workflow details including steps and graphs
|
|
1026
|
+
* Retrieves details about the legacy workflow
|
|
1027
|
+
* @returns Promise containing legacy workflow details including steps and graphs
|
|
580
1028
|
*/
|
|
581
1029
|
details() {
|
|
582
|
-
return this.request(`/api/workflows/${this.workflowId}`);
|
|
1030
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}`);
|
|
583
1031
|
}
|
|
584
1032
|
/**
|
|
585
|
-
* Retrieves all runs for a workflow
|
|
1033
|
+
* Retrieves all runs for a legacy workflow
|
|
586
1034
|
* @param params - Parameters for filtering runs
|
|
587
|
-
* @returns Promise containing workflow runs array
|
|
1035
|
+
* @returns Promise containing legacy workflow runs array
|
|
588
1036
|
*/
|
|
589
1037
|
runs(params) {
|
|
590
1038
|
const searchParams = new URLSearchParams();
|
|
@@ -604,25 +1052,13 @@ var Workflow = class extends BaseResource {
|
|
|
604
1052
|
searchParams.set("resourceId", params.resourceId);
|
|
605
1053
|
}
|
|
606
1054
|
if (searchParams.size) {
|
|
607
|
-
return this.request(`/api/workflows/${this.workflowId}/runs?${searchParams}`);
|
|
1055
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/runs?${searchParams}`);
|
|
608
1056
|
} else {
|
|
609
|
-
return this.request(`/api/workflows/${this.workflowId}/runs`);
|
|
1057
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/runs`);
|
|
610
1058
|
}
|
|
611
1059
|
}
|
|
612
1060
|
/**
|
|
613
|
-
*
|
|
614
|
-
* Executes the workflow with the provided parameters
|
|
615
|
-
* @param params - Parameters required for workflow execution
|
|
616
|
-
* @returns Promise containing the workflow execution results
|
|
617
|
-
*/
|
|
618
|
-
execute(params) {
|
|
619
|
-
return this.request(`/api/workflows/${this.workflowId}/execute`, {
|
|
620
|
-
method: "POST",
|
|
621
|
-
body: params
|
|
622
|
-
});
|
|
623
|
-
}
|
|
624
|
-
/**
|
|
625
|
-
* Creates a new workflow run
|
|
1061
|
+
* Creates a new legacy workflow run
|
|
626
1062
|
* @returns Promise containing the generated run ID
|
|
627
1063
|
*/
|
|
628
1064
|
createRun(params) {
|
|
@@ -630,34 +1066,34 @@ var Workflow = class extends BaseResource {
|
|
|
630
1066
|
if (!!params?.runId) {
|
|
631
1067
|
searchParams.set("runId", params.runId);
|
|
632
1068
|
}
|
|
633
|
-
return this.request(`/api/workflows/${this.workflowId}/
|
|
1069
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/create-run?${searchParams.toString()}`, {
|
|
634
1070
|
method: "POST"
|
|
635
1071
|
});
|
|
636
1072
|
}
|
|
637
1073
|
/**
|
|
638
|
-
* Starts a workflow run synchronously without waiting for the workflow to complete
|
|
1074
|
+
* Starts a legacy workflow run synchronously without waiting for the workflow to complete
|
|
639
1075
|
* @param params - Object containing the runId and triggerData
|
|
640
1076
|
* @returns Promise containing success message
|
|
641
1077
|
*/
|
|
642
1078
|
start(params) {
|
|
643
|
-
return this.request(`/api/workflows/${this.workflowId}/start?runId=${params.runId}`, {
|
|
1079
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/start?runId=${params.runId}`, {
|
|
644
1080
|
method: "POST",
|
|
645
1081
|
body: params?.triggerData
|
|
646
1082
|
});
|
|
647
1083
|
}
|
|
648
1084
|
/**
|
|
649
|
-
* Resumes a suspended workflow step synchronously without waiting for the workflow to complete
|
|
1085
|
+
* Resumes a suspended legacy workflow step synchronously without waiting for the workflow to complete
|
|
650
1086
|
* @param stepId - ID of the step to resume
|
|
651
|
-
* @param runId - ID of the workflow run
|
|
652
|
-
* @param context - Context to resume the workflow with
|
|
653
|
-
* @returns Promise containing the workflow resume results
|
|
1087
|
+
* @param runId - ID of the legacy workflow run
|
|
1088
|
+
* @param context - Context to resume the legacy workflow with
|
|
1089
|
+
* @returns Promise containing the legacy workflow resume results
|
|
654
1090
|
*/
|
|
655
1091
|
resume({
|
|
656
1092
|
stepId,
|
|
657
1093
|
runId,
|
|
658
1094
|
context
|
|
659
1095
|
}) {
|
|
660
|
-
return this.request(`/api/workflows/${this.workflowId}/resume?runId=${runId}`, {
|
|
1096
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/resume?runId=${runId}`, {
|
|
661
1097
|
method: "POST",
|
|
662
1098
|
body: {
|
|
663
1099
|
stepId,
|
|
@@ -675,18 +1111,18 @@ var Workflow = class extends BaseResource {
|
|
|
675
1111
|
if (!!params?.runId) {
|
|
676
1112
|
searchParams.set("runId", params.runId);
|
|
677
1113
|
}
|
|
678
|
-
return this.request(`/api/workflows/${this.workflowId}/start-async?${searchParams.toString()}`, {
|
|
1114
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/start-async?${searchParams.toString()}`, {
|
|
679
1115
|
method: "POST",
|
|
680
1116
|
body: params?.triggerData
|
|
681
1117
|
});
|
|
682
1118
|
}
|
|
683
1119
|
/**
|
|
684
|
-
* Resumes a suspended workflow step asynchronously and returns a promise that resolves when the workflow is complete
|
|
1120
|
+
* Resumes a suspended legacy workflow step asynchronously and returns a promise that resolves when the workflow is complete
|
|
685
1121
|
* @param params - Object containing the runId, stepId, and context
|
|
686
1122
|
* @returns Promise containing the workflow resume results
|
|
687
1123
|
*/
|
|
688
1124
|
resumeAsync(params) {
|
|
689
|
-
return this.request(`/api/workflows/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
1125
|
+
return this.request(`/api/workflows/legacy/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
690
1126
|
method: "POST",
|
|
691
1127
|
body: {
|
|
692
1128
|
stepId: params.stepId,
|
|
@@ -740,16 +1176,16 @@ var Workflow = class extends BaseResource {
|
|
|
740
1176
|
}
|
|
741
1177
|
}
|
|
742
1178
|
/**
|
|
743
|
-
* Watches workflow transitions in real-time
|
|
1179
|
+
* Watches legacy workflow transitions in real-time
|
|
744
1180
|
* @param runId - Optional run ID to filter the watch stream
|
|
745
|
-
* @returns AsyncGenerator that yields parsed records from the workflow watch stream
|
|
1181
|
+
* @returns AsyncGenerator that yields parsed records from the legacy workflow watch stream
|
|
746
1182
|
*/
|
|
747
1183
|
async watch({ runId }, onRecord) {
|
|
748
|
-
const response = await this.request(`/api/workflows/${this.workflowId}/watch?runId=${runId}`, {
|
|
1184
|
+
const response = await this.request(`/api/workflows/legacy/${this.workflowId}/watch?runId=${runId}`, {
|
|
749
1185
|
stream: true
|
|
750
1186
|
});
|
|
751
1187
|
if (!response.ok) {
|
|
752
|
-
throw new Error(`Failed to watch workflow: ${response.statusText}`);
|
|
1188
|
+
throw new Error(`Failed to watch legacy workflow: ${response.statusText}`);
|
|
753
1189
|
}
|
|
754
1190
|
if (!response.body) {
|
|
755
1191
|
throw new Error("Response body is null");
|
|
@@ -761,7 +1197,7 @@ var Workflow = class extends BaseResource {
|
|
|
761
1197
|
};
|
|
762
1198
|
|
|
763
1199
|
// src/resources/tool.ts
|
|
764
|
-
var
|
|
1200
|
+
var Tool2 = class extends BaseResource {
|
|
765
1201
|
constructor(options, toolId) {
|
|
766
1202
|
super(options);
|
|
767
1203
|
this.toolId = toolId;
|
|
@@ -785,7 +1221,7 @@ var Tool = class extends BaseResource {
|
|
|
785
1221
|
}
|
|
786
1222
|
const body = {
|
|
787
1223
|
data: params.data,
|
|
788
|
-
runtimeContext:
|
|
1224
|
+
runtimeContext: parseClientRuntimeContext(params.runtimeContext)
|
|
789
1225
|
};
|
|
790
1226
|
return this.request(`/api/tools/${this.toolId}/execute?${url.toString()}`, {
|
|
791
1227
|
method: "POST",
|
|
@@ -793,14 +1229,16 @@ var Tool = class extends BaseResource {
|
|
|
793
1229
|
});
|
|
794
1230
|
}
|
|
795
1231
|
};
|
|
1232
|
+
|
|
1233
|
+
// src/resources/workflow.ts
|
|
796
1234
|
var RECORD_SEPARATOR2 = "";
|
|
797
|
-
var
|
|
1235
|
+
var Workflow = class extends BaseResource {
|
|
798
1236
|
constructor(options, workflowId) {
|
|
799
1237
|
super(options);
|
|
800
1238
|
this.workflowId = workflowId;
|
|
801
1239
|
}
|
|
802
1240
|
/**
|
|
803
|
-
* Creates an async generator that processes a readable stream and yields
|
|
1241
|
+
* Creates an async generator that processes a readable stream and yields workflow records
|
|
804
1242
|
* separated by the Record Separator character (\x1E)
|
|
805
1243
|
*
|
|
806
1244
|
* @param stream - The readable stream to process
|
|
@@ -845,16 +1283,16 @@ var VNextWorkflow = class extends BaseResource {
|
|
|
845
1283
|
}
|
|
846
1284
|
}
|
|
847
1285
|
/**
|
|
848
|
-
* Retrieves details about the
|
|
849
|
-
* @returns Promise containing
|
|
1286
|
+
* Retrieves details about the workflow
|
|
1287
|
+
* @returns Promise containing workflow details including steps and graphs
|
|
850
1288
|
*/
|
|
851
1289
|
details() {
|
|
852
|
-
return this.request(`/api/workflows
|
|
1290
|
+
return this.request(`/api/workflows/${this.workflowId}`);
|
|
853
1291
|
}
|
|
854
1292
|
/**
|
|
855
|
-
* Retrieves all runs for a
|
|
1293
|
+
* Retrieves all runs for a workflow
|
|
856
1294
|
* @param params - Parameters for filtering runs
|
|
857
|
-
* @returns Promise containing
|
|
1295
|
+
* @returns Promise containing workflow runs array
|
|
858
1296
|
*/
|
|
859
1297
|
runs(params) {
|
|
860
1298
|
const searchParams = new URLSearchParams();
|
|
@@ -864,23 +1302,60 @@ var VNextWorkflow = class extends BaseResource {
|
|
|
864
1302
|
if (params?.toDate) {
|
|
865
1303
|
searchParams.set("toDate", params.toDate.toISOString());
|
|
866
1304
|
}
|
|
867
|
-
if (params?.limit) {
|
|
1305
|
+
if (params?.limit !== null && params?.limit !== void 0 && !isNaN(Number(params?.limit))) {
|
|
868
1306
|
searchParams.set("limit", String(params.limit));
|
|
869
1307
|
}
|
|
870
|
-
if (params?.offset) {
|
|
1308
|
+
if (params?.offset !== null && params?.offset !== void 0 && !isNaN(Number(params?.offset))) {
|
|
871
1309
|
searchParams.set("offset", String(params.offset));
|
|
872
1310
|
}
|
|
873
1311
|
if (params?.resourceId) {
|
|
874
1312
|
searchParams.set("resourceId", params.resourceId);
|
|
875
1313
|
}
|
|
876
1314
|
if (searchParams.size) {
|
|
877
|
-
return this.request(`/api/workflows
|
|
1315
|
+
return this.request(`/api/workflows/${this.workflowId}/runs?${searchParams}`);
|
|
878
1316
|
} else {
|
|
879
|
-
return this.request(`/api/workflows
|
|
1317
|
+
return this.request(`/api/workflows/${this.workflowId}/runs`);
|
|
880
1318
|
}
|
|
881
1319
|
}
|
|
882
1320
|
/**
|
|
883
|
-
*
|
|
1321
|
+
* Retrieves a specific workflow run by its ID
|
|
1322
|
+
* @param runId - The ID of the workflow run to retrieve
|
|
1323
|
+
* @returns Promise containing the workflow run details
|
|
1324
|
+
*/
|
|
1325
|
+
runById(runId) {
|
|
1326
|
+
return this.request(`/api/workflows/${this.workflowId}/runs/${runId}`);
|
|
1327
|
+
}
|
|
1328
|
+
/**
|
|
1329
|
+
* Retrieves the execution result for a specific workflow run by its ID
|
|
1330
|
+
* @param runId - The ID of the workflow run to retrieve the execution result for
|
|
1331
|
+
* @returns Promise containing the workflow run execution result
|
|
1332
|
+
*/
|
|
1333
|
+
runExecutionResult(runId) {
|
|
1334
|
+
return this.request(`/api/workflows/${this.workflowId}/runs/${runId}/execution-result`);
|
|
1335
|
+
}
|
|
1336
|
+
/**
|
|
1337
|
+
* Cancels a specific workflow run by its ID
|
|
1338
|
+
* @param runId - The ID of the workflow run to cancel
|
|
1339
|
+
* @returns Promise containing a success message
|
|
1340
|
+
*/
|
|
1341
|
+
cancelRun(runId) {
|
|
1342
|
+
return this.request(`/api/workflows/${this.workflowId}/runs/${runId}/cancel`, {
|
|
1343
|
+
method: "POST"
|
|
1344
|
+
});
|
|
1345
|
+
}
|
|
1346
|
+
/**
|
|
1347
|
+
* Sends an event to a specific workflow run by its ID
|
|
1348
|
+
* @param params - Object containing the runId, event and data
|
|
1349
|
+
* @returns Promise containing a success message
|
|
1350
|
+
*/
|
|
1351
|
+
sendRunEvent(params) {
|
|
1352
|
+
return this.request(`/api/workflows/${this.workflowId}/runs/${params.runId}/send-event`, {
|
|
1353
|
+
method: "POST",
|
|
1354
|
+
body: { event: params.event, data: params.data }
|
|
1355
|
+
});
|
|
1356
|
+
}
|
|
1357
|
+
/**
|
|
1358
|
+
* Creates a new workflow run
|
|
884
1359
|
* @param params - Optional object containing the optional runId
|
|
885
1360
|
* @returns Promise containing the runId of the created run
|
|
886
1361
|
*/
|
|
@@ -889,24 +1364,24 @@ var VNextWorkflow = class extends BaseResource {
|
|
|
889
1364
|
if (!!params?.runId) {
|
|
890
1365
|
searchParams.set("runId", params.runId);
|
|
891
1366
|
}
|
|
892
|
-
return this.request(`/api/workflows
|
|
1367
|
+
return this.request(`/api/workflows/${this.workflowId}/create-run?${searchParams.toString()}`, {
|
|
893
1368
|
method: "POST"
|
|
894
1369
|
});
|
|
895
1370
|
}
|
|
896
1371
|
/**
|
|
897
|
-
* Starts a
|
|
1372
|
+
* Starts a workflow run synchronously without waiting for the workflow to complete
|
|
898
1373
|
* @param params - Object containing the runId, inputData and runtimeContext
|
|
899
1374
|
* @returns Promise containing success message
|
|
900
1375
|
*/
|
|
901
1376
|
start(params) {
|
|
902
|
-
const runtimeContext =
|
|
903
|
-
return this.request(`/api/workflows
|
|
1377
|
+
const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
|
|
1378
|
+
return this.request(`/api/workflows/${this.workflowId}/start?runId=${params.runId}`, {
|
|
904
1379
|
method: "POST",
|
|
905
1380
|
body: { inputData: params?.inputData, runtimeContext }
|
|
906
1381
|
});
|
|
907
1382
|
}
|
|
908
1383
|
/**
|
|
909
|
-
* Resumes a suspended
|
|
1384
|
+
* Resumes a suspended workflow step synchronously without waiting for the workflow to complete
|
|
910
1385
|
* @param params - Object containing the runId, step, resumeData and runtimeContext
|
|
911
1386
|
* @returns Promise containing success message
|
|
912
1387
|
*/
|
|
@@ -916,8 +1391,8 @@ var VNextWorkflow = class extends BaseResource {
|
|
|
916
1391
|
resumeData,
|
|
917
1392
|
...rest
|
|
918
1393
|
}) {
|
|
919
|
-
const runtimeContext =
|
|
920
|
-
return this.request(`/api/workflows
|
|
1394
|
+
const runtimeContext = parseClientRuntimeContext(rest.runtimeContext);
|
|
1395
|
+
return this.request(`/api/workflows/${this.workflowId}/resume?runId=${runId}`, {
|
|
921
1396
|
method: "POST",
|
|
922
1397
|
stream: true,
|
|
923
1398
|
body: {
|
|
@@ -928,29 +1403,76 @@ var VNextWorkflow = class extends BaseResource {
|
|
|
928
1403
|
});
|
|
929
1404
|
}
|
|
930
1405
|
/**
|
|
931
|
-
* Starts a
|
|
1406
|
+
* Starts a workflow run asynchronously and returns a promise that resolves when the workflow is complete
|
|
932
1407
|
* @param params - Object containing the optional runId, inputData and runtimeContext
|
|
933
|
-
* @returns Promise containing the
|
|
1408
|
+
* @returns Promise containing the workflow execution results
|
|
934
1409
|
*/
|
|
935
1410
|
startAsync(params) {
|
|
936
1411
|
const searchParams = new URLSearchParams();
|
|
937
1412
|
if (!!params?.runId) {
|
|
938
1413
|
searchParams.set("runId", params.runId);
|
|
939
1414
|
}
|
|
940
|
-
const runtimeContext =
|
|
941
|
-
return this.request(`/api/workflows
|
|
1415
|
+
const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
|
|
1416
|
+
return this.request(`/api/workflows/${this.workflowId}/start-async?${searchParams.toString()}`, {
|
|
942
1417
|
method: "POST",
|
|
943
1418
|
body: { inputData: params.inputData, runtimeContext }
|
|
944
1419
|
});
|
|
945
1420
|
}
|
|
946
1421
|
/**
|
|
947
|
-
*
|
|
1422
|
+
* Starts a workflow run and returns a stream
|
|
1423
|
+
* @param params - Object containing the optional runId, inputData and runtimeContext
|
|
1424
|
+
* @returns Promise containing the workflow execution results
|
|
1425
|
+
*/
|
|
1426
|
+
async stream(params) {
|
|
1427
|
+
const searchParams = new URLSearchParams();
|
|
1428
|
+
if (!!params?.runId) {
|
|
1429
|
+
searchParams.set("runId", params.runId);
|
|
1430
|
+
}
|
|
1431
|
+
const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
|
|
1432
|
+
const response = await this.request(
|
|
1433
|
+
`/api/workflows/${this.workflowId}/stream?${searchParams.toString()}`,
|
|
1434
|
+
{
|
|
1435
|
+
method: "POST",
|
|
1436
|
+
body: { inputData: params.inputData, runtimeContext },
|
|
1437
|
+
stream: true
|
|
1438
|
+
}
|
|
1439
|
+
);
|
|
1440
|
+
if (!response.ok) {
|
|
1441
|
+
throw new Error(`Failed to stream vNext workflow: ${response.statusText}`);
|
|
1442
|
+
}
|
|
1443
|
+
if (!response.body) {
|
|
1444
|
+
throw new Error("Response body is null");
|
|
1445
|
+
}
|
|
1446
|
+
const transformStream = new TransformStream({
|
|
1447
|
+
start() {
|
|
1448
|
+
},
|
|
1449
|
+
async transform(chunk, controller) {
|
|
1450
|
+
try {
|
|
1451
|
+
const decoded = new TextDecoder().decode(chunk);
|
|
1452
|
+
const chunks = decoded.split(RECORD_SEPARATOR2);
|
|
1453
|
+
for (const chunk2 of chunks) {
|
|
1454
|
+
if (chunk2) {
|
|
1455
|
+
try {
|
|
1456
|
+
const parsedChunk = JSON.parse(chunk2);
|
|
1457
|
+
controller.enqueue(parsedChunk);
|
|
1458
|
+
} catch {
|
|
1459
|
+
}
|
|
1460
|
+
}
|
|
1461
|
+
}
|
|
1462
|
+
} catch {
|
|
1463
|
+
}
|
|
1464
|
+
}
|
|
1465
|
+
});
|
|
1466
|
+
return response.body.pipeThrough(transformStream);
|
|
1467
|
+
}
|
|
1468
|
+
/**
|
|
1469
|
+
* Resumes a suspended workflow step asynchronously and returns a promise that resolves when the workflow is complete
|
|
948
1470
|
* @param params - Object containing the runId, step, resumeData and runtimeContext
|
|
949
|
-
* @returns Promise containing the
|
|
1471
|
+
* @returns Promise containing the workflow resume results
|
|
950
1472
|
*/
|
|
951
1473
|
resumeAsync(params) {
|
|
952
|
-
const runtimeContext =
|
|
953
|
-
return this.request(`/api/workflows
|
|
1474
|
+
const runtimeContext = parseClientRuntimeContext(params.runtimeContext);
|
|
1475
|
+
return this.request(`/api/workflows/${this.workflowId}/resume-async?runId=${params.runId}`, {
|
|
954
1476
|
method: "POST",
|
|
955
1477
|
body: {
|
|
956
1478
|
step: params.step,
|
|
@@ -960,24 +1482,51 @@ var VNextWorkflow = class extends BaseResource {
|
|
|
960
1482
|
});
|
|
961
1483
|
}
|
|
962
1484
|
/**
|
|
963
|
-
* Watches
|
|
1485
|
+
* Watches workflow transitions in real-time
|
|
964
1486
|
* @param runId - Optional run ID to filter the watch stream
|
|
965
|
-
* @returns AsyncGenerator that yields parsed records from the
|
|
1487
|
+
* @returns AsyncGenerator that yields parsed records from the workflow watch stream
|
|
966
1488
|
*/
|
|
967
1489
|
async watch({ runId }, onRecord) {
|
|
968
|
-
const response = await this.request(`/api/workflows
|
|
1490
|
+
const response = await this.request(`/api/workflows/${this.workflowId}/watch?runId=${runId}`, {
|
|
969
1491
|
stream: true
|
|
970
1492
|
});
|
|
971
1493
|
if (!response.ok) {
|
|
972
|
-
throw new Error(`Failed to watch
|
|
1494
|
+
throw new Error(`Failed to watch workflow: ${response.statusText}`);
|
|
973
1495
|
}
|
|
974
1496
|
if (!response.body) {
|
|
975
1497
|
throw new Error("Response body is null");
|
|
976
1498
|
}
|
|
977
1499
|
for await (const record of this.streamProcessor(response.body)) {
|
|
978
|
-
|
|
1500
|
+
if (typeof record === "string") {
|
|
1501
|
+
onRecord(JSON.parse(record));
|
|
1502
|
+
} else {
|
|
1503
|
+
onRecord(record);
|
|
1504
|
+
}
|
|
979
1505
|
}
|
|
980
1506
|
}
|
|
1507
|
+
/**
|
|
1508
|
+
* Creates a new ReadableStream from an iterable or async iterable of objects,
|
|
1509
|
+
* serializing each as JSON and separating them with the record separator (\x1E).
|
|
1510
|
+
*
|
|
1511
|
+
* @param records - An iterable or async iterable of objects to stream
|
|
1512
|
+
* @returns A ReadableStream emitting the records as JSON strings separated by the record separator
|
|
1513
|
+
*/
|
|
1514
|
+
static createRecordStream(records) {
|
|
1515
|
+
const encoder = new TextEncoder();
|
|
1516
|
+
return new ReadableStream({
|
|
1517
|
+
async start(controller) {
|
|
1518
|
+
try {
|
|
1519
|
+
for await (const record of records) {
|
|
1520
|
+
const json = JSON.stringify(record) + RECORD_SEPARATOR2;
|
|
1521
|
+
controller.enqueue(encoder.encode(json));
|
|
1522
|
+
}
|
|
1523
|
+
controller.close();
|
|
1524
|
+
} catch (err) {
|
|
1525
|
+
controller.error(err);
|
|
1526
|
+
}
|
|
1527
|
+
}
|
|
1528
|
+
});
|
|
1529
|
+
}
|
|
981
1530
|
};
|
|
982
1531
|
|
|
983
1532
|
// src/resources/a2a.ts
|
|
@@ -1054,6 +1603,214 @@ var A2A = class extends BaseResource {
|
|
|
1054
1603
|
}
|
|
1055
1604
|
};
|
|
1056
1605
|
|
|
1606
|
+
// src/resources/mcp-tool.ts
|
|
1607
|
+
var MCPTool = class extends BaseResource {
|
|
1608
|
+
serverId;
|
|
1609
|
+
toolId;
|
|
1610
|
+
constructor(options, serverId, toolId) {
|
|
1611
|
+
super(options);
|
|
1612
|
+
this.serverId = serverId;
|
|
1613
|
+
this.toolId = toolId;
|
|
1614
|
+
}
|
|
1615
|
+
/**
|
|
1616
|
+
* Retrieves details about this specific tool from the MCP server.
|
|
1617
|
+
* @returns Promise containing the tool's information (name, description, schema).
|
|
1618
|
+
*/
|
|
1619
|
+
details() {
|
|
1620
|
+
return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}`);
|
|
1621
|
+
}
|
|
1622
|
+
/**
|
|
1623
|
+
* Executes this specific tool on the MCP server.
|
|
1624
|
+
* @param params - Parameters for tool execution, including data/args and optional runtimeContext.
|
|
1625
|
+
* @returns Promise containing the result of the tool execution.
|
|
1626
|
+
*/
|
|
1627
|
+
execute(params) {
|
|
1628
|
+
const body = {};
|
|
1629
|
+
if (params.data !== void 0) body.data = params.data;
|
|
1630
|
+
if (params.runtimeContext !== void 0) {
|
|
1631
|
+
body.runtimeContext = params.runtimeContext;
|
|
1632
|
+
}
|
|
1633
|
+
return this.request(`/api/mcp/${this.serverId}/tools/${this.toolId}/execute`, {
|
|
1634
|
+
method: "POST",
|
|
1635
|
+
body: Object.keys(body).length > 0 ? body : void 0
|
|
1636
|
+
});
|
|
1637
|
+
}
|
|
1638
|
+
};
|
|
1639
|
+
|
|
1640
|
+
// src/resources/vNextNetwork.ts
|
|
1641
|
+
var RECORD_SEPARATOR3 = "";
|
|
1642
|
+
var VNextNetwork = class extends BaseResource {
|
|
1643
|
+
constructor(options, networkId) {
|
|
1644
|
+
super(options);
|
|
1645
|
+
this.networkId = networkId;
|
|
1646
|
+
}
|
|
1647
|
+
/**
|
|
1648
|
+
* Retrieves details about the network
|
|
1649
|
+
* @returns Promise containing vNext network details
|
|
1650
|
+
*/
|
|
1651
|
+
details() {
|
|
1652
|
+
return this.request(`/api/networks/v-next/${this.networkId}`);
|
|
1653
|
+
}
|
|
1654
|
+
/**
|
|
1655
|
+
* Generates a response from the v-next network
|
|
1656
|
+
* @param params - Generation parameters including message
|
|
1657
|
+
* @returns Promise containing the generated response
|
|
1658
|
+
*/
|
|
1659
|
+
generate(params) {
|
|
1660
|
+
return this.request(`/api/networks/v-next/${this.networkId}/generate`, {
|
|
1661
|
+
method: "POST",
|
|
1662
|
+
body: params
|
|
1663
|
+
});
|
|
1664
|
+
}
|
|
1665
|
+
/**
|
|
1666
|
+
* Generates a response from the v-next network using multiple primitives
|
|
1667
|
+
* @param params - Generation parameters including message
|
|
1668
|
+
* @returns Promise containing the generated response
|
|
1669
|
+
*/
|
|
1670
|
+
loop(params) {
|
|
1671
|
+
return this.request(`/api/networks/v-next/${this.networkId}/loop`, {
|
|
1672
|
+
method: "POST",
|
|
1673
|
+
body: params
|
|
1674
|
+
});
|
|
1675
|
+
}
|
|
1676
|
+
async *streamProcessor(stream) {
|
|
1677
|
+
const reader = stream.getReader();
|
|
1678
|
+
let doneReading = false;
|
|
1679
|
+
let buffer = "";
|
|
1680
|
+
try {
|
|
1681
|
+
while (!doneReading) {
|
|
1682
|
+
const { done, value } = await reader.read();
|
|
1683
|
+
doneReading = done;
|
|
1684
|
+
if (done && !value) continue;
|
|
1685
|
+
try {
|
|
1686
|
+
const decoded = value ? new TextDecoder().decode(value) : "";
|
|
1687
|
+
const chunks = (buffer + decoded).split(RECORD_SEPARATOR3);
|
|
1688
|
+
buffer = chunks.pop() || "";
|
|
1689
|
+
for (const chunk of chunks) {
|
|
1690
|
+
if (chunk) {
|
|
1691
|
+
if (typeof chunk === "string") {
|
|
1692
|
+
try {
|
|
1693
|
+
const parsedChunk = JSON.parse(chunk);
|
|
1694
|
+
yield parsedChunk;
|
|
1695
|
+
} catch {
|
|
1696
|
+
}
|
|
1697
|
+
}
|
|
1698
|
+
}
|
|
1699
|
+
}
|
|
1700
|
+
} catch {
|
|
1701
|
+
}
|
|
1702
|
+
}
|
|
1703
|
+
if (buffer) {
|
|
1704
|
+
try {
|
|
1705
|
+
yield JSON.parse(buffer);
|
|
1706
|
+
} catch {
|
|
1707
|
+
}
|
|
1708
|
+
}
|
|
1709
|
+
} finally {
|
|
1710
|
+
reader.cancel().catch(() => {
|
|
1711
|
+
});
|
|
1712
|
+
}
|
|
1713
|
+
}
|
|
1714
|
+
/**
|
|
1715
|
+
* Streams a response from the v-next network
|
|
1716
|
+
* @param params - Stream parameters including message
|
|
1717
|
+
* @returns Promise containing the results
|
|
1718
|
+
*/
|
|
1719
|
+
async stream(params, onRecord) {
|
|
1720
|
+
const response = await this.request(`/api/networks/v-next/${this.networkId}/stream`, {
|
|
1721
|
+
method: "POST",
|
|
1722
|
+
body: params,
|
|
1723
|
+
stream: true
|
|
1724
|
+
});
|
|
1725
|
+
if (!response.ok) {
|
|
1726
|
+
throw new Error(`Failed to stream vNext network: ${response.statusText}`);
|
|
1727
|
+
}
|
|
1728
|
+
if (!response.body) {
|
|
1729
|
+
throw new Error("Response body is null");
|
|
1730
|
+
}
|
|
1731
|
+
for await (const record of this.streamProcessor(response.body)) {
|
|
1732
|
+
if (typeof record === "string") {
|
|
1733
|
+
onRecord(JSON.parse(record));
|
|
1734
|
+
} else {
|
|
1735
|
+
onRecord(record);
|
|
1736
|
+
}
|
|
1737
|
+
}
|
|
1738
|
+
}
|
|
1739
|
+
/**
|
|
1740
|
+
* Streams a response from the v-next network loop
|
|
1741
|
+
* @param params - Stream parameters including message
|
|
1742
|
+
* @returns Promise containing the results
|
|
1743
|
+
*/
|
|
1744
|
+
async loopStream(params, onRecord) {
|
|
1745
|
+
const response = await this.request(`/api/networks/v-next/${this.networkId}/loop-stream`, {
|
|
1746
|
+
method: "POST",
|
|
1747
|
+
body: params,
|
|
1748
|
+
stream: true
|
|
1749
|
+
});
|
|
1750
|
+
if (!response.ok) {
|
|
1751
|
+
throw new Error(`Failed to stream vNext network loop: ${response.statusText}`);
|
|
1752
|
+
}
|
|
1753
|
+
if (!response.body) {
|
|
1754
|
+
throw new Error("Response body is null");
|
|
1755
|
+
}
|
|
1756
|
+
for await (const record of this.streamProcessor(response.body)) {
|
|
1757
|
+
if (typeof record === "string") {
|
|
1758
|
+
onRecord(JSON.parse(record));
|
|
1759
|
+
} else {
|
|
1760
|
+
onRecord(record);
|
|
1761
|
+
}
|
|
1762
|
+
}
|
|
1763
|
+
}
|
|
1764
|
+
};
|
|
1765
|
+
|
|
1766
|
+
// src/resources/network-memory-thread.ts
|
|
1767
|
+
var NetworkMemoryThread = class extends BaseResource {
|
|
1768
|
+
constructor(options, threadId, networkId) {
|
|
1769
|
+
super(options);
|
|
1770
|
+
this.threadId = threadId;
|
|
1771
|
+
this.networkId = networkId;
|
|
1772
|
+
}
|
|
1773
|
+
/**
|
|
1774
|
+
* Retrieves the memory thread details
|
|
1775
|
+
* @returns Promise containing thread details including title and metadata
|
|
1776
|
+
*/
|
|
1777
|
+
get() {
|
|
1778
|
+
return this.request(`/api/memory/network/threads/${this.threadId}?networkId=${this.networkId}`);
|
|
1779
|
+
}
|
|
1780
|
+
/**
|
|
1781
|
+
* Updates the memory thread properties
|
|
1782
|
+
* @param params - Update parameters including title and metadata
|
|
1783
|
+
* @returns Promise containing updated thread details
|
|
1784
|
+
*/
|
|
1785
|
+
update(params) {
|
|
1786
|
+
return this.request(`/api/memory/network/threads/${this.threadId}?networkId=${this.networkId}`, {
|
|
1787
|
+
method: "PATCH",
|
|
1788
|
+
body: params
|
|
1789
|
+
});
|
|
1790
|
+
}
|
|
1791
|
+
/**
|
|
1792
|
+
* Deletes the memory thread
|
|
1793
|
+
* @returns Promise containing deletion result
|
|
1794
|
+
*/
|
|
1795
|
+
delete() {
|
|
1796
|
+
return this.request(`/api/memory/network/threads/${this.threadId}?networkId=${this.networkId}`, {
|
|
1797
|
+
method: "DELETE"
|
|
1798
|
+
});
|
|
1799
|
+
}
|
|
1800
|
+
/**
|
|
1801
|
+
* Retrieves messages associated with the thread
|
|
1802
|
+
* @param params - Optional parameters including limit for number of messages to retrieve
|
|
1803
|
+
* @returns Promise containing thread messages and UI messages
|
|
1804
|
+
*/
|
|
1805
|
+
getMessages(params) {
|
|
1806
|
+
const query = new URLSearchParams({
|
|
1807
|
+
networkId: this.networkId,
|
|
1808
|
+
...params?.limit ? { limit: params.limit.toString() } : {}
|
|
1809
|
+
});
|
|
1810
|
+
return this.request(`/api/memory/network/threads/${this.threadId}/messages?${query.toString()}`);
|
|
1811
|
+
}
|
|
1812
|
+
};
|
|
1813
|
+
|
|
1057
1814
|
// src/client.ts
|
|
1058
1815
|
var MastraClient = class extends BaseResource {
|
|
1059
1816
|
constructor(options) {
|
|
@@ -1131,6 +1888,48 @@ var MastraClient = class extends BaseResource {
|
|
|
1131
1888
|
getMemoryStatus(agentId) {
|
|
1132
1889
|
return this.request(`/api/memory/status?agentId=${agentId}`);
|
|
1133
1890
|
}
|
|
1891
|
+
/**
|
|
1892
|
+
* Retrieves memory threads for a resource
|
|
1893
|
+
* @param params - Parameters containing the resource ID
|
|
1894
|
+
* @returns Promise containing array of memory threads
|
|
1895
|
+
*/
|
|
1896
|
+
getNetworkMemoryThreads(params) {
|
|
1897
|
+
return this.request(`/api/memory/network/threads?resourceid=${params.resourceId}&networkId=${params.networkId}`);
|
|
1898
|
+
}
|
|
1899
|
+
/**
|
|
1900
|
+
* Creates a new memory thread
|
|
1901
|
+
* @param params - Parameters for creating the memory thread
|
|
1902
|
+
* @returns Promise containing the created memory thread
|
|
1903
|
+
*/
|
|
1904
|
+
createNetworkMemoryThread(params) {
|
|
1905
|
+
return this.request(`/api/memory/network/threads?networkId=${params.networkId}`, { method: "POST", body: params });
|
|
1906
|
+
}
|
|
1907
|
+
/**
|
|
1908
|
+
* Gets a memory thread instance by ID
|
|
1909
|
+
* @param threadId - ID of the memory thread to retrieve
|
|
1910
|
+
* @returns MemoryThread instance
|
|
1911
|
+
*/
|
|
1912
|
+
getNetworkMemoryThread(threadId, networkId) {
|
|
1913
|
+
return new NetworkMemoryThread(this.options, threadId, networkId);
|
|
1914
|
+
}
|
|
1915
|
+
/**
|
|
1916
|
+
* Saves messages to memory
|
|
1917
|
+
* @param params - Parameters containing messages to save
|
|
1918
|
+
* @returns Promise containing the saved messages
|
|
1919
|
+
*/
|
|
1920
|
+
saveNetworkMessageToMemory(params) {
|
|
1921
|
+
return this.request(`/api/memory/network/save-messages?networkId=${params.networkId}`, {
|
|
1922
|
+
method: "POST",
|
|
1923
|
+
body: params
|
|
1924
|
+
});
|
|
1925
|
+
}
|
|
1926
|
+
/**
|
|
1927
|
+
* Gets the status of the memory system
|
|
1928
|
+
* @returns Promise containing memory system status
|
|
1929
|
+
*/
|
|
1930
|
+
getNetworkMemoryStatus(networkId) {
|
|
1931
|
+
return this.request(`/api/memory/network/status?networkId=${networkId}`);
|
|
1932
|
+
}
|
|
1134
1933
|
/**
|
|
1135
1934
|
* Retrieves all available tools
|
|
1136
1935
|
* @returns Promise containing map of tool IDs to tool details
|
|
@@ -1144,7 +1943,22 @@ var MastraClient = class extends BaseResource {
|
|
|
1144
1943
|
* @returns Tool instance
|
|
1145
1944
|
*/
|
|
1146
1945
|
getTool(toolId) {
|
|
1147
|
-
return new
|
|
1946
|
+
return new Tool2(this.options, toolId);
|
|
1947
|
+
}
|
|
1948
|
+
/**
|
|
1949
|
+
* Retrieves all available legacy workflows
|
|
1950
|
+
* @returns Promise containing map of legacy workflow IDs to legacy workflow details
|
|
1951
|
+
*/
|
|
1952
|
+
getLegacyWorkflows() {
|
|
1953
|
+
return this.request("/api/workflows/legacy");
|
|
1954
|
+
}
|
|
1955
|
+
/**
|
|
1956
|
+
* Gets a legacy workflow instance by ID
|
|
1957
|
+
* @param workflowId - ID of the legacy workflow to retrieve
|
|
1958
|
+
* @returns Legacy Workflow instance
|
|
1959
|
+
*/
|
|
1960
|
+
getLegacyWorkflow(workflowId) {
|
|
1961
|
+
return new LegacyWorkflow(this.options, workflowId);
|
|
1148
1962
|
}
|
|
1149
1963
|
/**
|
|
1150
1964
|
* Retrieves all available workflows
|
|
@@ -1161,21 +1975,6 @@ var MastraClient = class extends BaseResource {
|
|
|
1161
1975
|
getWorkflow(workflowId) {
|
|
1162
1976
|
return new Workflow(this.options, workflowId);
|
|
1163
1977
|
}
|
|
1164
|
-
/**
|
|
1165
|
-
* Retrieves all available vNext workflows
|
|
1166
|
-
* @returns Promise containing map of vNext workflow IDs to vNext workflow details
|
|
1167
|
-
*/
|
|
1168
|
-
getVNextWorkflows() {
|
|
1169
|
-
return this.request("/api/workflows/v-next");
|
|
1170
|
-
}
|
|
1171
|
-
/**
|
|
1172
|
-
* Gets a vNext workflow instance by ID
|
|
1173
|
-
* @param workflowId - ID of the vNext workflow to retrieve
|
|
1174
|
-
* @returns vNext Workflow instance
|
|
1175
|
-
*/
|
|
1176
|
-
getVNextWorkflow(workflowId) {
|
|
1177
|
-
return new VNextWorkflow(this.options, workflowId);
|
|
1178
|
-
}
|
|
1179
1978
|
/**
|
|
1180
1979
|
* Gets a vector instance by name
|
|
1181
1980
|
* @param vectorName - Name of the vector to retrieve
|
|
@@ -1190,7 +1989,41 @@ var MastraClient = class extends BaseResource {
|
|
|
1190
1989
|
* @returns Promise containing array of log messages
|
|
1191
1990
|
*/
|
|
1192
1991
|
getLogs(params) {
|
|
1193
|
-
|
|
1992
|
+
const { transportId, fromDate, toDate, logLevel, filters, page, perPage } = params;
|
|
1993
|
+
const _filters = filters ? Object.entries(filters).map(([key, value]) => `${key}:${value}`) : [];
|
|
1994
|
+
const searchParams = new URLSearchParams();
|
|
1995
|
+
if (transportId) {
|
|
1996
|
+
searchParams.set("transportId", transportId);
|
|
1997
|
+
}
|
|
1998
|
+
if (fromDate) {
|
|
1999
|
+
searchParams.set("fromDate", fromDate.toISOString());
|
|
2000
|
+
}
|
|
2001
|
+
if (toDate) {
|
|
2002
|
+
searchParams.set("toDate", toDate.toISOString());
|
|
2003
|
+
}
|
|
2004
|
+
if (logLevel) {
|
|
2005
|
+
searchParams.set("logLevel", logLevel);
|
|
2006
|
+
}
|
|
2007
|
+
if (page) {
|
|
2008
|
+
searchParams.set("page", String(page));
|
|
2009
|
+
}
|
|
2010
|
+
if (perPage) {
|
|
2011
|
+
searchParams.set("perPage", String(perPage));
|
|
2012
|
+
}
|
|
2013
|
+
if (_filters) {
|
|
2014
|
+
if (Array.isArray(_filters)) {
|
|
2015
|
+
for (const filter of _filters) {
|
|
2016
|
+
searchParams.append("filters", filter);
|
|
2017
|
+
}
|
|
2018
|
+
} else {
|
|
2019
|
+
searchParams.set("filters", _filters);
|
|
2020
|
+
}
|
|
2021
|
+
}
|
|
2022
|
+
if (searchParams.size) {
|
|
2023
|
+
return this.request(`/api/logs?${searchParams}`);
|
|
2024
|
+
} else {
|
|
2025
|
+
return this.request(`/api/logs`);
|
|
2026
|
+
}
|
|
1194
2027
|
}
|
|
1195
2028
|
/**
|
|
1196
2029
|
* Gets logs for a specific run
|
|
@@ -1198,7 +2031,44 @@ var MastraClient = class extends BaseResource {
|
|
|
1198
2031
|
* @returns Promise containing array of log messages
|
|
1199
2032
|
*/
|
|
1200
2033
|
getLogForRun(params) {
|
|
1201
|
-
|
|
2034
|
+
const { runId, transportId, fromDate, toDate, logLevel, filters, page, perPage } = params;
|
|
2035
|
+
const _filters = filters ? Object.entries(filters).map(([key, value]) => `${key}:${value}`) : [];
|
|
2036
|
+
const searchParams = new URLSearchParams();
|
|
2037
|
+
if (runId) {
|
|
2038
|
+
searchParams.set("runId", runId);
|
|
2039
|
+
}
|
|
2040
|
+
if (transportId) {
|
|
2041
|
+
searchParams.set("transportId", transportId);
|
|
2042
|
+
}
|
|
2043
|
+
if (fromDate) {
|
|
2044
|
+
searchParams.set("fromDate", fromDate.toISOString());
|
|
2045
|
+
}
|
|
2046
|
+
if (toDate) {
|
|
2047
|
+
searchParams.set("toDate", toDate.toISOString());
|
|
2048
|
+
}
|
|
2049
|
+
if (logLevel) {
|
|
2050
|
+
searchParams.set("logLevel", logLevel);
|
|
2051
|
+
}
|
|
2052
|
+
if (page) {
|
|
2053
|
+
searchParams.set("page", String(page));
|
|
2054
|
+
}
|
|
2055
|
+
if (perPage) {
|
|
2056
|
+
searchParams.set("perPage", String(perPage));
|
|
2057
|
+
}
|
|
2058
|
+
if (_filters) {
|
|
2059
|
+
if (Array.isArray(_filters)) {
|
|
2060
|
+
for (const filter of _filters) {
|
|
2061
|
+
searchParams.append("filters", filter);
|
|
2062
|
+
}
|
|
2063
|
+
} else {
|
|
2064
|
+
searchParams.set("filters", _filters);
|
|
2065
|
+
}
|
|
2066
|
+
}
|
|
2067
|
+
if (searchParams.size) {
|
|
2068
|
+
return this.request(`/api/logs/${runId}?${searchParams}`);
|
|
2069
|
+
} else {
|
|
2070
|
+
return this.request(`/api/logs/${runId}`);
|
|
2071
|
+
}
|
|
1202
2072
|
}
|
|
1203
2073
|
/**
|
|
1204
2074
|
* List of all log transports
|
|
@@ -1256,6 +2126,13 @@ var MastraClient = class extends BaseResource {
|
|
|
1256
2126
|
getNetworks() {
|
|
1257
2127
|
return this.request("/api/networks");
|
|
1258
2128
|
}
|
|
2129
|
+
/**
|
|
2130
|
+
* Retrieves all available vNext networks
|
|
2131
|
+
* @returns Promise containing map of vNext network IDs to vNext network details
|
|
2132
|
+
*/
|
|
2133
|
+
getVNextNetworks() {
|
|
2134
|
+
return this.request("/api/networks/v-next");
|
|
2135
|
+
}
|
|
1259
2136
|
/**
|
|
1260
2137
|
* Gets a network instance by ID
|
|
1261
2138
|
* @param networkId - ID of the network to retrieve
|
|
@@ -1264,6 +2141,62 @@ var MastraClient = class extends BaseResource {
|
|
|
1264
2141
|
getNetwork(networkId) {
|
|
1265
2142
|
return new Network(this.options, networkId);
|
|
1266
2143
|
}
|
|
2144
|
+
/**
|
|
2145
|
+
* Gets a vNext network instance by ID
|
|
2146
|
+
* @param networkId - ID of the vNext network to retrieve
|
|
2147
|
+
* @returns vNext Network instance
|
|
2148
|
+
*/
|
|
2149
|
+
getVNextNetwork(networkId) {
|
|
2150
|
+
return new VNextNetwork(this.options, networkId);
|
|
2151
|
+
}
|
|
2152
|
+
/**
|
|
2153
|
+
* Retrieves a list of available MCP servers.
|
|
2154
|
+
* @param params - Optional parameters for pagination (limit, offset).
|
|
2155
|
+
* @returns Promise containing the list of MCP servers and pagination info.
|
|
2156
|
+
*/
|
|
2157
|
+
getMcpServers(params) {
|
|
2158
|
+
const searchParams = new URLSearchParams();
|
|
2159
|
+
if (params?.limit !== void 0) {
|
|
2160
|
+
searchParams.set("limit", String(params.limit));
|
|
2161
|
+
}
|
|
2162
|
+
if (params?.offset !== void 0) {
|
|
2163
|
+
searchParams.set("offset", String(params.offset));
|
|
2164
|
+
}
|
|
2165
|
+
const queryString = searchParams.toString();
|
|
2166
|
+
return this.request(`/api/mcp/v0/servers${queryString ? `?${queryString}` : ""}`);
|
|
2167
|
+
}
|
|
2168
|
+
/**
|
|
2169
|
+
* Retrieves detailed information for a specific MCP server.
|
|
2170
|
+
* @param serverId - The ID of the MCP server to retrieve.
|
|
2171
|
+
* @param params - Optional parameters, e.g., specific version.
|
|
2172
|
+
* @returns Promise containing the detailed MCP server information.
|
|
2173
|
+
*/
|
|
2174
|
+
getMcpServerDetails(serverId, params) {
|
|
2175
|
+
const searchParams = new URLSearchParams();
|
|
2176
|
+
if (params?.version) {
|
|
2177
|
+
searchParams.set("version", params.version);
|
|
2178
|
+
}
|
|
2179
|
+
const queryString = searchParams.toString();
|
|
2180
|
+
return this.request(`/api/mcp/v0/servers/${serverId}${queryString ? `?${queryString}` : ""}`);
|
|
2181
|
+
}
|
|
2182
|
+
/**
|
|
2183
|
+
* Retrieves a list of tools for a specific MCP server.
|
|
2184
|
+
* @param serverId - The ID of the MCP server.
|
|
2185
|
+
* @returns Promise containing the list of tools.
|
|
2186
|
+
*/
|
|
2187
|
+
getMcpServerTools(serverId) {
|
|
2188
|
+
return this.request(`/api/mcp/${serverId}/tools`);
|
|
2189
|
+
}
|
|
2190
|
+
/**
|
|
2191
|
+
* Gets an MCPTool resource instance for a specific tool on an MCP server.
|
|
2192
|
+
* This instance can then be used to fetch details or execute the tool.
|
|
2193
|
+
* @param serverId - The ID of the MCP server.
|
|
2194
|
+
* @param toolId - The ID of the tool.
|
|
2195
|
+
* @returns MCPTool instance.
|
|
2196
|
+
*/
|
|
2197
|
+
getMcpServerTool(serverId, toolId) {
|
|
2198
|
+
return new MCPTool(this.options, serverId, toolId);
|
|
2199
|
+
}
|
|
1267
2200
|
/**
|
|
1268
2201
|
* Gets an A2A client for interacting with an agent via the A2A protocol
|
|
1269
2202
|
* @param agentId - ID of the agent to interact with
|