@mastra/client-js 0.1.13-alpha.2 → 0.1.13-alpha.3

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @mastra/client-js@0.1.13-alpha.2 build /home/runner/work/mastra/mastra/client-sdks/client-js
+> @mastra/client-js@0.1.13-alpha.3 build /home/runner/work/mastra/mastra/client-sdks/client-js
 > tsup src/index.ts --format esm,cjs --dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
@@ -9,11 +9,11 @@
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-CJS dist/index.cjs 21.09 KB
-CJS ⚡️ Build success in 874ms
-ESM dist/index.js 20.92 KB
-ESM ⚡️ Build success in 874ms
+ESM dist/index.js 21.58 KB
+ESM ⚡️ Build success in 1464ms
+CJS dist/index.cjs 21.76 KB
+CJS ⚡️ Build success in 1475ms
 DTS Build start
-DTS ⚡️ Build success in 11272ms
-DTS dist/index.d.ts  18.78 KB
-DTS dist/index.d.cts 18.78 KB
+DTS ⚡️ Build success in 15172ms
+DTS dist/index.d.ts  19.18 KB
+DTS dist/index.d.cts 19.18 KB

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/client-js
 
+## 0.1.13-alpha.3
+
+### Patch Changes
+
+- 160f88e: Leverage aiSDK processDataStream util in MastraClient
+- Updated dependencies [b3b34f5]
+- Updated dependencies [a4686e8]
+  - @mastra/core@0.7.0-alpha.3
+
 ## 0.1.13-alpha.2
 
 ### Patch Changes

package/dist/index.cjs

@@ -2,6 +2,7 @@
 
 var zod = require('zod');
 var zodToJsonSchema = require('zod-to-json-schema');
+var uiUtils = require('@ai-sdk/ui-utils');
 
 // src/resources/agent.ts
 
@@ -144,19 +145,29 @@ var Agent = class extends BaseResource {
   /**
    * Streams a response from the agent
    * @param params - Stream parameters including prompt
-   * @returns Promise containing the streamed response
+   * @returns Promise containing the enhanced Response object with processDataStream method
    */
-  stream(params) {
+  async stream(params) {
     const processedParams = {
       ...params,
       output: params.output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.output) : params.output,
       experimental_output: params.experimental_output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.experimental_output) : params.experimental_output
     };
-    return this.request(`/api/agents/${this.agentId}/stream`, {
+    const response = await this.request(`/api/agents/${this.agentId}/stream`, {
       method: "POST",
       body: processedParams,
       stream: true
     });
+    if (!response.body) {
+      throw new Error("No response body");
+    }
+    response.processDataStream = async (options = {}) => {
+      await uiUtils.processDataStream({
+        stream: response.body,
+        ...options
+      });
+    };
+    return response;
   }
   /**
    * Gets details about a specific tool available to the agent
@@ -212,19 +223,29 @@ var Network = class extends BaseResource {
   /**
    * Streams a response from the agent
    * @param params - Stream parameters including prompt
-   * @returns Promise containing the streamed response
+   * @returns Promise containing the enhanced Response object with processDataStream method
    */
-  stream(params) {
+  async stream(params) {
     const processedParams = {
       ...params,
       output: params.output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.output) : params.output,
       experimental_output: params.experimental_output instanceof zod.ZodSchema ? zodToJsonSchema.zodToJsonSchema(params.experimental_output) : params.experimental_output
     };
-    return this.request(`/api/networks/${this.networkId}/stream`, {
+    const response = await this.request(`/api/networks/${this.networkId}/stream`, {
       method: "POST",
       body: processedParams,
       stream: true
     });
+    if (!response.body) {
+      throw new Error("No response body");
+    }
+    response.processDataStream = async (options = {}) => {
+      await uiUtils.processDataStream({
+        stream: response.body,
+        ...options
+      });
+    };
+    return response;
   }
 };
 

package/dist/index.d.cts

@@ -1,6 +1,7 @@
 import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
 import { JSONSchema7 } from 'json-schema';
 import { ZodSchema } from 'zod';
+import { processDataStream } from '@ai-sdk/ui-utils';
 import { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
 
 interface ClientOptions {
@@ -34,7 +35,7 @@ type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> =
 } & Partial<AgentGenerateOptions<T>>;
 type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
     messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Partial<AgentStreamOptions<T>>;
+} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
 interface GetEvalsByAgentIdResponse extends GetAgentResponse {
     evals: any[];
 }
@@ -221,9 +222,11 @@ declare class Agent extends BaseResource {
     /**
      * Streams a response from the agent
      * @param params - Stream parameters including prompt
-     * @returns Promise containing the streamed response
+     * @returns Promise containing the enhanced Response object with processDataStream method
      */
-    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response>;
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response & {
+        processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    }>;
     /**
      * Gets details about a specific tool available to the agent
      * @param toolId - ID of the tool to retrieve
@@ -259,9 +262,11 @@ declare class Network extends BaseResource {
     /**
      * Streams a response from the agent
      * @param params - Stream parameters including prompt
-     * @returns Promise containing the streamed response
+     * @returns Promise containing the enhanced Response object with processDataStream method
      */
-    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response>;
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response & {
+        processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    }>;
 }
 
 declare class MemoryThread extends BaseResource {

package/dist/index.d.ts

@@ -1,6 +1,7 @@
 import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
 import { JSONSchema7 } from 'json-schema';
 import { ZodSchema } from 'zod';
+import { processDataStream } from '@ai-sdk/ui-utils';
 import { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
 
 interface ClientOptions {
@@ -34,7 +35,7 @@ type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> =
 } & Partial<AgentGenerateOptions<T>>;
 type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
     messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Partial<AgentStreamOptions<T>>;
+} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
 interface GetEvalsByAgentIdResponse extends GetAgentResponse {
     evals: any[];
 }
@@ -221,9 +222,11 @@ declare class Agent extends BaseResource {
     /**
      * Streams a response from the agent
      * @param params - Stream parameters including prompt
-     * @returns Promise containing the streamed response
+     * @returns Promise containing the enhanced Response object with processDataStream method
      */
-    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response>;
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response & {
+        processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    }>;
     /**
      * Gets details about a specific tool available to the agent
      * @param toolId - ID of the tool to retrieve
@@ -259,9 +262,11 @@ declare class Network extends BaseResource {
     /**
      * Streams a response from the agent
      * @param params - Stream parameters including prompt
-     * @returns Promise containing the streamed response
+     * @returns Promise containing the enhanced Response object with processDataStream method
      */
-    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response>;
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response & {
+        processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    }>;
 }
 
 declare class MemoryThread extends BaseResource {

package/dist/index.js

@@ -1,5 +1,6 @@
 import { ZodSchema } from 'zod';
 import { zodToJsonSchema } from 'zod-to-json-schema';
+import { processDataStream } from '@ai-sdk/ui-utils';
 
 // src/resources/agent.ts
 
@@ -142,19 +143,29 @@ var Agent = class extends BaseResource {
   /**
    * Streams a response from the agent
    * @param params - Stream parameters including prompt
-   * @returns Promise containing the streamed response
+   * @returns Promise containing the enhanced Response object with processDataStream method
    */
-  stream(params) {
+  async stream(params) {
     const processedParams = {
       ...params,
       output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
       experimental_output: params.experimental_output instanceof ZodSchema ? zodToJsonSchema(params.experimental_output) : params.experimental_output
     };
-    return this.request(`/api/agents/${this.agentId}/stream`, {
+    const response = await this.request(`/api/agents/${this.agentId}/stream`, {
       method: "POST",
       body: processedParams,
       stream: true
     });
+    if (!response.body) {
+      throw new Error("No response body");
+    }
+    response.processDataStream = async (options = {}) => {
+      await processDataStream({
+        stream: response.body,
+        ...options
+      });
+    };
+    return response;
   }
   /**
    * Gets details about a specific tool available to the agent
@@ -210,19 +221,29 @@ var Network = class extends BaseResource {
   /**
    * Streams a response from the agent
    * @param params - Stream parameters including prompt
-   * @returns Promise containing the streamed response
+   * @returns Promise containing the enhanced Response object with processDataStream method
    */
-  stream(params) {
+  async stream(params) {
     const processedParams = {
       ...params,
       output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
       experimental_output: params.experimental_output instanceof ZodSchema ? zodToJsonSchema(params.experimental_output) : params.experimental_output
     };
-    return this.request(`/api/networks/${this.networkId}/stream`, {
+    const response = await this.request(`/api/networks/${this.networkId}/stream`, {
       method: "POST",
       body: processedParams,
       stream: true
     });
+    if (!response.body) {
+      throw new Error("No response body");
+    }
+    response.processDataStream = async (options = {}) => {
+      await processDataStream({
+        stream: response.body,
+        ...options
+      });
+    };
+    return response;
   }
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/client-js",
-  "version": "0.1.13-alpha.2",
+  "version": "0.1.13-alpha.3",
   "description": "The official TypeScript library for the Mastra Client API",
   "author": "",
   "type": "module",
@@ -22,10 +22,11 @@
   "repository": "github:mastra-ai/client-js",
   "license": "ISC",
   "dependencies": {
+    "@ai-sdk/ui-utils": "^1.1.19",
     "json-schema": "^0.4.0",
     "zod": "^3.24.2",
     "zod-to-json-schema": "^3.24.3",
-    "@mastra/core": "^0.7.0-alpha.2"
+    "@mastra/core": "^0.7.0-alpha.3"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.26.9",

package/src/example.ts

@@ -1,5 +1,5 @@
-import { MastraClient } from './client';
-import type { WorkflowRunResult } from './types';
+// import { MastraClient } from './client';
+// import type { WorkflowRunResult } from './types';
 
 // Agent
 
@@ -8,38 +8,29 @@ import type { WorkflowRunResult } from './types';
 //     baseUrl: 'http://localhost:4111',
 //   });
 
+//   console.log('Starting agent...');
+
 //   try {
 //     const agent = client.getAgent('weatherAgent');
 //     const response = await agent.stream({
-//       messages: [
-//         {
-//           role: 'user',
-//           content: 'Hello, world!',
-//         },
-//       ],
-//     });
-
-//     const reader = response?.body?.getReader();
-//     const decoder = new TextDecoder();
-//     let buffer = '';
-
-//     while (true) {
-//       if (!reader) break;
-//       const { value, done } = await reader.read();
-//       if (done) break;
-
-//       const chunk = decoder.decode(value);
-//       buffer += chunk;
-
-//       console.log(buffer);
+//       messages: 'what is the weather in new york?',
+//     })
 
-//       const matches = buffer.matchAll(/0:"([^"]*)"/g);
+//     response.processDataStream({
+//       onTextPart: (text) => {
+//         process.stdout.write(text);
+//       },
+//       onFilePart: (file) => {
+//         console.log(file);
+//       },
+//       onDataPart: (data) => {
+//         console.log(data);
+//       },
+//       onErrorPart: (error) => {
+//         console.error(error);
+//       },
+//     });
 
-//       for (const match of matches) {
-//         const content = match[1];
-//         process.stdout.write(`${content}\n`);
-//       }
-//     }
 //   } catch (error) {
 //     console.error(error);
 //   }

package/src/index.test.ts

@@ -318,19 +318,18 @@ describe('MastraClient Resources', () => {
     });
 
     it('should get agent evals', async () => {
-      const mockResponse = {
-        name: 'Test Agent',
-        evals: [{ id: 'eval1' }],
-      };
+      const mockResponse = { data: 'test' };
       mockFetchResponse(mockResponse);
       const result = await agent.evals();
       expect(result).toEqual(mockResponse);
-      expect(global.fetch).toHaveBeenCalledWith(`${clientOptions.baseUrl}/api/agents/test-agent/evals/ci`, {
-        headers: {
-          Authorization: 'Bearer test-key',
-          'Content-Type': 'application/json',
-        },
-      });
+      expect(global.fetch).toHaveBeenCalledWith(
+        `${clientOptions.baseUrl}/api/agents/test-agent/evals/ci`,
+        expect.objectContaining({
+          headers: expect.objectContaining({
+            Authorization: 'Bearer test-key',
+          }),
+        }),
+      );
     });
 
     it('should get live evals', async () => {
@@ -529,29 +528,27 @@ describe('MastraClient Resources', () => {
     });
 
     it('should save messages to memory', async () => {
-      const messages: MessageType[] = [
+      const messages = [
         {
           id: '1',
-          type: 'text',
+          type: 'text' as const,
           content: 'test',
-          role: 'user',
+          role: 'user' as const,
           threadId: 'test-thread',
-          createdAt: new Date(),
+          createdAt: new Date('2025-03-26T10:40:55.116Z'),
         },
       ];
       mockFetchResponse(messages);
-      const result = await client.saveMessageToMemory({ messages, agentId });
+      const result = await client.saveMessageToMemory({ agentId, messages });
       expect(result).toEqual(messages);
       expect(global.fetch).toHaveBeenCalledWith(
         `${clientOptions.baseUrl}/api/memory/save-messages?agentId=${agentId}`,
-        {
+        expect.objectContaining({
           method: 'POST',
-          headers: {
+          headers: expect.objectContaining({
             Authorization: 'Bearer test-key',
-            'Content-Type': 'application/json',
-          },
-          body: JSON.stringify({ messages, agentId }),
-        },
+          }),
+        }),
       );
     });
   });
@@ -584,21 +581,19 @@ describe('MastraClient Resources', () => {
     });
 
     it('should execute tool', async () => {
-      const mockResponse = {
-        result: 'Tool execution result',
-      };
+      const mockResponse = { data: 'test' };
       mockFetchResponse(mockResponse);
-
       const result = await tool.execute({ data: '' });
       expect(result).toEqual(mockResponse);
-      expect(global.fetch).toHaveBeenCalledWith(`${clientOptions.baseUrl}/api/tools/test-tool/execute`, {
-        method: 'POST',
-        headers: {
-          Authorization: 'Bearer test-key',
-          'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({ data: '' }),
-      });
+      expect(global.fetch).toHaveBeenCalledWith(
+        `${clientOptions.baseUrl}/api/tools/test-tool/execute`,
+        expect.objectContaining({
+          method: 'POST',
+          headers: expect.objectContaining({
+            Authorization: 'Bearer test-key',
+          }),
+        }),
+      );
     });
   });
 

package/src/resources/agent.ts

@@ -1,7 +1,8 @@
-import type { GenerateReturn, StreamReturn } from '@mastra/core';
+import type { GenerateReturn } from '@mastra/core';
 import type { JSONSchema7 } from 'json-schema';
 import { ZodSchema } from 'zod';
 import { zodToJsonSchema } from 'zod-to-json-schema';
+import { processDataStream } from '@ai-sdk/ui-utils';
 
 import type {
   GenerateParams,
@@ -28,6 +29,7 @@ export class AgentTool extends BaseResource {
    * @param params - Parameters required for tool execution
    * @returns Promise containing tool execution results
    */
+  /** @deprecated use CreateRun/startRun */
   execute(params: { data: any }): Promise<any> {
     return this.request(`/api/agents/${this.agentId}/tools/${this.toolId}/execute`, {
       method: 'POST',
@@ -136,9 +138,15 @@ export class Agent extends BaseResource {
   /**
    * Streams a response from the agent
    * @param params - Stream parameters including prompt
-   * @returns Promise containing the streamed response
+   * @returns Promise containing the enhanced Response object with processDataStream method
    */
-  stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response> {
+  async stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(
+    params: StreamParams<T>,
+  ): Promise<
+    Response & {
+      processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    }
+  > {
     const processedParams = {
       ...params,
       output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
@@ -148,11 +156,26 @@ export class Agent extends BaseResource {
           : params.experimental_output,
     };
 
-    return this.request(`/api/agents/${this.agentId}/stream`, {
+    const response: Response & {
+      processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    } = await this.request(`/api/agents/${this.agentId}/stream`, {
       method: 'POST',
       body: processedParams,
       stream: true,
     });
+
+    if (!response.body) {
+      throw new Error('No response body');
+    }
+
+    response.processDataStream = async (options = {}) => {
+      await processDataStream({
+        stream: response.body as ReadableStream<Uint8Array>,
+        ...options,
+      });
+    };
+
+    return response;
   }
 
   /**

package/src/resources/network.ts

@@ -6,6 +6,7 @@ import { zodToJsonSchema } from 'zod-to-json-schema';
 import type { GenerateParams, ClientOptions, StreamParams, GetNetworkResponse } from '../types';
 
 import { BaseResource } from './base';
+import { processDataStream } from '@ai-sdk/ui-utils';
 
 export class Network extends BaseResource {
   constructor(
@@ -49,9 +50,15 @@ export class Network extends BaseResource {
   /**
    * Streams a response from the agent
    * @param params - Stream parameters including prompt
-   * @returns Promise containing the streamed response
+   * @returns Promise containing the enhanced Response object with processDataStream method
    */
-  stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response> {
+  async stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(
+    params: StreamParams<T>,
+  ): Promise<
+    Response & {
+      processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    }
+  > {
     const processedParams = {
       ...params,
       output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
@@ -61,10 +68,25 @@ export class Network extends BaseResource {
           : params.experimental_output,
     };
 
-    return this.request(`/api/networks/${this.networkId}/stream`, {
+    const response: Response & {
+      processDataStream: (options?: Omit<Parameters<typeof processDataStream>[0], 'stream'>) => Promise<void>;
+    } = await this.request(`/api/networks/${this.networkId}/stream`, {
       method: 'POST',
       body: processedParams,
       stream: true,
     });
+
+    if (!response.body) {
+      throw new Error('No response body');
+    }
+
+    response.processDataStream = async (options = {}) => {
+      await processDataStream({
+        stream: response.body as ReadableStream<Uint8Array>,
+        ...options,
+      });
+    };
+
+    return response;
   }
 }

package/src/types.ts

@@ -49,7 +49,7 @@ export type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undef
 
 export type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
   messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Partial<AgentStreamOptions<T>>;
+} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
 
 export interface GetEvalsByAgentIdResponse extends GetAgentResponse {
   evals: any[];