@mastra/client-js 0.10.21-alpha.1 → 0.10.21-alpha.2

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.
@@ -1,4 +1,4 @@
1
- import type { TaskSendParams, TaskQueryParams, TaskIdParams, Task, AgentCard } from '@mastra/core/a2a';
1
+ import type { AgentCard, GetTaskResponse, MessageSendParams, SendMessageResponse, SendStreamingMessageResponse, Task, TaskQueryParams } from '@mastra/core/a2a';
2
2
  import type { ClientOptions } from '../types.js';
3
3
  import { BaseResource } from './base.js';
4
4
  /**
@@ -13,32 +13,29 @@ export declare class A2A extends BaseResource {
13
13
  */
14
14
  getCard(): Promise<AgentCard>;
15
15
  /**
16
- * Send a message to the agent and get a response
16
+ * Send a message to the agent and gets a message or task response
17
17
  * @param params - Parameters for the task
18
- * @returns Promise containing the task response
18
+ * @returns Promise containing the response
19
+ */
20
+ sendMessage(params: MessageSendParams): Promise<SendMessageResponse>;
21
+ /**
22
+ * Sends a message to an agent to initiate/continue a task and subscribes
23
+ * the client to real-time updates for that task via Server-Sent Events (SSE).
24
+ * @param params - Parameters for the task
25
+ * @returns A stream of Server-Sent Events. Each SSE `data` field contains a `SendStreamingMessageResponse`
19
26
  */
20
- sendMessage(params: TaskSendParams): Promise<{
21
- task: Task;
22
- }>;
27
+ sendStreamingMessage(params: MessageSendParams): Promise<AsyncIterable<SendStreamingMessageResponse>>;
23
28
  /**
24
29
  * Get the status and result of a task
25
30
  * @param params - Parameters for querying the task
26
31
  * @returns Promise containing the task response
27
32
  */
28
- getTask(params: TaskQueryParams): Promise<Task>;
33
+ getTask(params: TaskQueryParams): Promise<GetTaskResponse>;
29
34
  /**
30
35
  * Cancel a running task
31
36
  * @param params - Parameters identifying the task to cancel
32
37
  * @returns Promise containing the task response
33
38
  */
34
- cancelTask(params: TaskIdParams): Promise<{
35
- task: Task;
36
- }>;
37
- /**
38
- * Send a message and subscribe to streaming updates (not fully implemented)
39
- * @param params - Parameters for the task
40
- * @returns Promise containing the task response
41
- */
42
- sendAndSubscribe(params: TaskSendParams): Promise<Response>;
39
+ cancelTask(params: TaskQueryParams): Promise<Task>;
43
40
  }
44
41
  //# sourceMappingURL=a2a.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"a2a.d.ts","sourceRoot":"","sources":["../../src/resources/a2a.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,eAAe,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,EAAmB,MAAM,kBAAkB,CAAC;AACxH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC;;GAEG;AACH,qBAAa,GAAI,SAAQ,YAAY;IAGjC,OAAO,CAAC,OAAO;gBADf,OAAO,EAAE,aAAa,EACd,OAAO,EAAE,MAAM;IAKzB;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,SAAS,CAAC;IAInC;;;;OAIG;IACG,WAAW,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,IAAI,CAAA;KAAE,CAAC;IAYlE;;;;OAIG;IACG,OAAO,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IAYrD;;;;OAIG;IACG,UAAU,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,IAAI,CAAA;KAAE,CAAC;IAU/D;;;;OAIG;IACG,gBAAgB,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC;CAUlE"}
1
+ {"version":3,"file":"a2a.d.ts","sourceRoot":"","sources":["../../src/resources/a2a.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,mBAAmB,EACnB,4BAA4B,EAC5B,IAAI,EACJ,eAAe,EAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC;;GAEG;AACH,qBAAa,GAAI,SAAQ,YAAY;IAGjC,OAAO,CAAC,OAAO;gBADf,OAAO,EAAE,aAAa,EACd,OAAO,EAAE,MAAM;IAKzB;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,SAAS,CAAC;IAInC;;;;OAIG;IACG,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAY1E;;;;;OAKG;IACG,oBAAoB,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,aAAa,CAAC,4BAA4B,CAAC,CAAC;IAY3G;;;;OAIG;IACG,OAAO,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IAYhE;;;;OAIG;IACG,UAAU,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;CASzD"}
@@ -1,4 +1,5 @@
1
1
  import { processDataStream } from '@ai-sdk/ui-utils';
2
+ import { type MastraLanguageModel } from '@mastra/core';
2
3
  import { type GenerateReturn } from '@mastra/core/llm';
3
4
  import type { JSONSchema7 } from 'json-schema';
4
5
  import { ZodSchema } from 'zod';
@@ -108,5 +109,15 @@ export declare class Agent extends BaseResource {
108
109
  * @returns Promise containing live agent evaluations
109
110
  */
110
111
  liveEvals(): Promise<GetEvalsByAgentIdResponse>;
112
+ /**
113
+ * Updates the model for the agent
114
+ * @param params - Parameters for updating the model
115
+ * @returns Promise containing the updated model
116
+ */
117
+ updateModel(params: {
118
+ model: MastraLanguageModel;
119
+ }): Promise<{
120
+ message: string;
121
+ }>;
111
122
  }
112
123
  //# sourceMappingURL=agent.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/resources/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,iBAAiB,EAQlB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,KAAK,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAKhC,OAAO,KAAK,EACV,cAAc,EACd,gBAAgB,EAChB,yBAAyB,EACzB,eAAe,EACf,aAAa,EACb,YAAY,EACb,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAInE,qBAAa,UAAW,SAAQ,YAAY;IAGxC,OAAO,CAAC,OAAO;gBADf,OAAO,EAAE,aAAa,EACd,OAAO,EAAE,MAAM;IAMzB;;;;;OAKG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,OAAO,CAAC,QAAQ,CAAC;IAWhG;;;;;OAKG;IACH,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IAc7E;;;OAGG;IACH,WAAW,IAAI,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC,CAAC;IAItE;;;OAGG;IACH,WAAW,IAAI,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;CAG7C;AAED,qBAAa,KAAM,SAAQ,YAAY;IAKnC,OAAO,CAAC,OAAO;IAJjB,SAAgB,KAAK,EAAE,UAAU,CAAC;gBAGhC,OAAO,EAAE,aAAa,EACd,OAAO,EAAE,MAAM;IAMzB;;;OAGG;IACH,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC;;;;OAIG;IACG,QAAQ,CACZ,MAAM,EAAE,cAAc,CAAC,SAAS,CAAC,GAAG;QAAE,MAAM,CAAC,EAAE,KAAK,CAAC;QAAC,mBAAmB,CAAC,EAAE,KAAK,CAAA;KAAE,GAClF,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IAC/C,QAAQ,CAAC,MAAM,SAAS,WAAW,GAAG,SAAS,EACnD,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,GAAG;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,mBAAmB,CAAC,EAAE,KAAK,CAAA;KAAE,GAC/E,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAC5C,QAAQ,CAAC,gBAAgB,SAAS,WAAW,GAAG,SAAS,EAC7D,MAAM,EAAE,cAAc,CAAC,gBAAgB,CAAC,GAAG;QAAE,MAAM,CAAC,EAAE,KAAK,CAAC;QAAC,mBAAmB,EAAE,gBAAgB,CAAA;KAAE,GACnG,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;YA6E9C,mBAAmB;IAyVjC;;;;OAIG;IACG,MAAM,CAAC,CAAC,SAAS,WAAW,GAAG,SAAS,GAAG,SAAS,GAAG,SAAS,EACpE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,GACtB,OAAO,CACR,QAAQ,GAAG;QACT,iBAAiB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACzG,CACF;IAmCD;;OAEG;YACW,qBAAqB;IAgJnC;;;;OAIG;IACH,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAIjD;;;;;OAKG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;QAAE,IAAI,EAAE,GAAG,CAAC;QAAC,cAAc,CAAC,EAAE,cAAc,CAAA;KAAE,GAAG,OAAO,CAAC,GAAG,CAAC;IAWjG;;;OAGG;IACH,KAAK,IAAI,OAAO,CAAC,yBAAyB,CAAC;IAI3C;;;OAGG;IACH,SAAS,IAAI,OAAO,CAAC,yBAAyB,CAAC;CAGhD"}
1
+ {"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/resources/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,iBAAiB,EAQlB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAA0B,KAAK,mBAAmB,EAAE,MAAM,cAAc,CAAC;AAChF,OAAO,EAAE,KAAK,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAKhC,OAAO,KAAK,EACV,cAAc,EACd,gBAAgB,EAChB,yBAAyB,EACzB,eAAe,EACf,aAAa,EACb,YAAY,EACb,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAInE,qBAAa,UAAW,SAAQ,YAAY;IAGxC,OAAO,CAAC,OAAO;gBADf,OAAO,EAAE,aAAa,EACd,OAAO,EAAE,MAAM;IAMzB;;;;;OAKG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,OAAO,CAAC,QAAQ,CAAC;IAWhG;;;;;OAKG;IACH,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IAc7E;;;OAGG;IACH,WAAW,IAAI,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC,CAAC;IAItE;;;OAGG;IACH,WAAW,IAAI,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;CAG7C;AAED,qBAAa,KAAM,SAAQ,YAAY;IAKnC,OAAO,CAAC,OAAO;IAJjB,SAAgB,KAAK,EAAE,UAAU,CAAC;gBAGhC,OAAO,EAAE,aAAa,EACd,OAAO,EAAE,MAAM;IAMzB;;;OAGG;IACH,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC;;;;OAIG;IACG,QAAQ,CACZ,MAAM,EAAE,cAAc,CAAC,SAAS,CAAC,GAAG;QAAE,MAAM,CAAC,EAAE,KAAK,CAAC;QAAC,mBAAmB,CAAC,EAAE,KAAK,CAAA;KAAE,GAClF,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IAC/C,QAAQ,CAAC,MAAM,SAAS,WAAW,GAAG,SAAS,EACnD,MAAM,EAAE,cAAc,CAAC,MAAM,CAAC,GAAG;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,mBAAmB,CAAC,EAAE,KAAK,CAAA;KAAE,GAC/E,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAC5C,QAAQ,CAAC,gBAAgB,SAAS,WAAW,GAAG,SAAS,EAC7D,MAAM,EAAE,cAAc,CAAC,gBAAgB,CAAC,GAAG;QAAE,MAAM,CAAC,EAAE,KAAK,CAAC;QAAC,mBAAmB,EAAE,gBAAgB,CAAA;KAAE,GACnG,OAAO,CAAC,cAAc,CAAC,GAAG,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;YA6E9C,mBAAmB;IAyVjC;;;;OAIG;IACG,MAAM,CAAC,CAAC,SAAS,WAAW,GAAG,SAAS,GAAG,SAAS,GAAG,SAAS,EACpE,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,GACtB,OAAO,CACR,QAAQ,GAAG;QACT,iBAAiB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACzG,CACF;IAmCD;;OAEG;YACW,qBAAqB;IAgJnC;;;;OAIG;IACH,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAIjD;;;;;OAKG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE;QAAE,IAAI,EAAE,GAAG,CAAC;QAAC,cAAc,CAAC,EAAE,cAAc,CAAA;KAAE,GAAG,OAAO,CAAC,GAAG,CAAC;IAWjG;;;OAGG;IACH,KAAK,IAAI,OAAO,CAAC,yBAAyB,CAAC;IAI3C;;;OAGG;IACH,SAAS,IAAI,OAAO,CAAC,yBAAyB,CAAC;IAI/C;;;;OAIG;IACH,WAAW,CAAC,MAAM,EAAE;QAAE,KAAK,EAAE,mBAAmB,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;CAMlF"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@mastra/client-js",
3
- "version": "0.10.21-alpha.1",
3
+ "version": "0.10.21-alpha.2",
4
4
  "description": "The official TypeScript library for the Mastra Client API",
5
5
  "author": "",
6
6
  "type": "module",
@@ -34,7 +34,7 @@
34
34
  "rxjs": "7.8.1",
35
35
  "zod": "^3.25.67",
36
36
  "zod-to-json-schema": "^3.24.5",
37
- "@mastra/core": "0.13.2-alpha.1"
37
+ "@mastra/core": "0.13.2-alpha.2"
38
38
  },
39
39
  "peerDependencies": {
40
40
  "zod": "^3.0.0"
package/src/client.ts CHANGED
@@ -620,4 +620,12 @@ export class MastraClient extends BaseResource {
620
620
  body: params,
621
621
  });
622
622
  }
623
+
624
+ /**
625
+ * Retrieves model providers with available keys
626
+ * @returns Promise containing model providers with available keys
627
+ */
628
+ getModelProviders(): Promise<string[]> {
629
+ return this.request(`/api/model-providers`);
630
+ }
623
631
  }
@@ -1,4 +1,12 @@
1
- import type { TaskSendParams, TaskQueryParams, TaskIdParams, Task, AgentCard, JSONRPCResponse } from '@mastra/core/a2a';
1
+ import type {
2
+ AgentCard,
3
+ GetTaskResponse,
4
+ MessageSendParams,
5
+ SendMessageResponse,
6
+ SendStreamingMessageResponse,
7
+ Task,
8
+ TaskQueryParams,
9
+ } from '@mastra/core/a2a';
2
10
  import type { ClientOptions } from '../types';
3
11
  import { BaseResource } from './base';
4
12
 
@@ -18,71 +26,73 @@ export class A2A extends BaseResource {
18
26
  * @returns Promise containing the agent card information
19
27
  */
20
28
  async getCard(): Promise<AgentCard> {
21
- return this.request(`/.well-known/${this.agentId}/agent.json`);
29
+ return this.request(`/.well-known/${this.agentId}/agent-card.json`);
22
30
  }
23
31
 
24
32
  /**
25
- * Send a message to the agent and get a response
33
+ * Send a message to the agent and gets a message or task response
26
34
  * @param params - Parameters for the task
27
- * @returns Promise containing the task response
35
+ * @returns Promise containing the response
28
36
  */
29
- async sendMessage(params: TaskSendParams): Promise<{ task: Task }> {
30
- const response = await this.request<JSONRPCResponse<Task>>(`/a2a/${this.agentId}`, {
37
+ async sendMessage(params: MessageSendParams): Promise<SendMessageResponse> {
38
+ const response = await this.request<SendMessageResponse>(`/a2a/${this.agentId}`, {
31
39
  method: 'POST',
32
40
  body: {
33
- method: 'tasks/send',
41
+ method: 'message/send',
34
42
  params,
35
43
  },
36
44
  });
37
45
 
38
- return { task: response.result! };
46
+ return response;
39
47
  }
40
48
 
41
49
  /**
42
- * Get the status and result of a task
43
- * @param params - Parameters for querying the task
44
- * @returns Promise containing the task response
50
+ * Sends a message to an agent to initiate/continue a task and subscribes
51
+ * the client to real-time updates for that task via Server-Sent Events (SSE).
52
+ * @param params - Parameters for the task
53
+ * @returns A stream of Server-Sent Events. Each SSE `data` field contains a `SendStreamingMessageResponse`
45
54
  */
46
- async getTask(params: TaskQueryParams): Promise<Task> {
47
- const response = await this.request<JSONRPCResponse<Task>>(`/a2a/${this.agentId}`, {
55
+ async sendStreamingMessage(params: MessageSendParams): Promise<AsyncIterable<SendStreamingMessageResponse>> {
56
+ const response = await this.request<AsyncIterable<SendStreamingMessageResponse>>(`/a2a/${this.agentId}`, {
48
57
  method: 'POST',
49
58
  body: {
50
- method: 'tasks/get',
59
+ method: 'message/stream',
51
60
  params,
52
61
  },
53
62
  });
54
63
 
55
- return response.result!;
64
+ return response;
56
65
  }
57
66
 
58
67
  /**
59
- * Cancel a running task
60
- * @param params - Parameters identifying the task to cancel
68
+ * Get the status and result of a task
69
+ * @param params - Parameters for querying the task
61
70
  * @returns Promise containing the task response
62
71
  */
63
- async cancelTask(params: TaskIdParams): Promise<{ task: Task }> {
64
- return this.request(`/a2a/${this.agentId}`, {
72
+ async getTask(params: TaskQueryParams): Promise<GetTaskResponse> {
73
+ const response = await this.request<GetTaskResponse>(`/a2a/${this.agentId}`, {
65
74
  method: 'POST',
66
75
  body: {
67
- method: 'tasks/cancel',
76
+ method: 'tasks/get',
68
77
  params,
69
78
  },
70
79
  });
80
+
81
+ return response;
71
82
  }
72
83
 
73
84
  /**
74
- * Send a message and subscribe to streaming updates (not fully implemented)
75
- * @param params - Parameters for the task
85
+ * Cancel a running task
86
+ * @param params - Parameters identifying the task to cancel
76
87
  * @returns Promise containing the task response
77
88
  */
78
- async sendAndSubscribe(params: TaskSendParams): Promise<Response> {
89
+ async cancelTask(params: TaskQueryParams): Promise<Task> {
79
90
  return this.request(`/a2a/${this.agentId}`, {
80
91
  method: 'POST',
81
92
  body: {
82
- method: 'tasks/sendSubscribe',
93
+ method: 'tasks/cancel',
83
94
  params,
84
95
  },
85
- stream: true,
86
96
  });
87
97
  }
88
98
  }
@@ -9,7 +9,7 @@ import {
9
9
  type UIMessage,
10
10
  type UseChatOptions,
11
11
  } from '@ai-sdk/ui-utils';
12
- import { Tool, type CoreMessage } from '@mastra/core';
12
+ import { Tool, type CoreMessage, type MastraLanguageModel } from '@mastra/core';
13
13
  import { type GenerateReturn } from '@mastra/core/llm';
14
14
  import type { JSONSchema7 } from 'json-schema';
15
15
  import { ZodSchema } from 'zod';
@@ -782,4 +782,16 @@ export class Agent extends BaseResource {
782
782
  liveEvals(): Promise<GetEvalsByAgentIdResponse> {
783
783
  return this.request(`/api/agents/${this.agentId}/evals/live`);
784
784
  }
785
+
786
+ /**
787
+ * Updates the model for the agent
788
+ * @param params - Parameters for updating the model
789
+ * @returns Promise containing the updated model
790
+ */
791
+ updateModel(params: { model: MastraLanguageModel }): Promise<{ message: string }> {
792
+ return this.request(`/api/agents/${this.agentId}/model`, {
793
+ method: 'POST',
794
+ body: params,
795
+ });
796
+ }
785
797
  }