@mastra/client-js 0.1.0-alpha.7 → 0.1.0-alpha.9

package/dist/index.d.mts

@@ -18,6 +18,7 @@ interface RequestOptions {
     method?: string;
     headers?: Record<string, string>;
     body?: any;
+    stream?: boolean;
 }
 interface GetAgentResponse {
     name: string;
@@ -144,7 +145,7 @@ declare class Agent extends BaseResource {
      * @param params - Stream parameters including prompt
      * @returns Promise containing the streamed response
      */
-    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): T extends undefined ? Promise<ReadableStream<Uint8Array>> : Promise<Response>;
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response>;
     /**
      * Gets details about a specific tool available to the agent
      * @param toolId - ID of the tool to retrieve

package/dist/index.mjs

@@ -43,10 +43,7 @@ var BaseResource = class {
           }
           throw new Error(errorMessage);
         }
-        if (response.headers.get("Content-Type")?.includes("text/event-stream")) {
-          return response.body;
-        }
-        if (response.headers.get("Content-Type")?.includes("text/x-unknown")) {
+        if (options.stream) {
           return response;
         }
         const data = await response.json();
@@ -100,12 +97,12 @@ var Agent = class extends BaseResource {
   stream(params) {
     const processedParams = {
       ...params,
-      output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
-      stream: true
+      output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output
     };
     return this.request(`/api/agents/${this.agentId}/stream`, {
       method: "POST",
-      body: processedParams
+      body: processedParams,
+      stream: true
     });
   }
   /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/client-js",
-  "version": "0.1.0-alpha.7",
+  "version": "0.1.0-alpha.9",
   "description": "The official TypeScript library for the Mastra Client API",
   "author": "",
   "types": "dist/index.d.mts",

package/src/example.ts

@@ -16,17 +16,30 @@ import { MastraClient } from "./client";
                 createdAt: new Date(),
                 threadId: '1',
                 type: 'text',
-            }],
+            }]
         })
 
 
-        const reader = response?.getReader()
+        const reader = response?.body?.getReader()
+        const decoder = new TextDecoder()
+        let buffer = ''
+
         while (true) {
-            const { done, value } = await reader.read();
+            if (!reader) break;
+            const { value, done } = await reader.read()
             if (done) break;
 
-            const text = new TextDecoder().decode(value);
-            console.log(text);
+            const chunk = decoder.decode(value);
+            buffer += chunk;
+
+            console.log(buffer);
+
+            const matches = buffer.matchAll(/0:"([^"]*)"/g);
+
+            for (const match of matches) {
+                const content = match[1];
+                process.stdout.write(`${content}\n`);
+            }
         }
     } catch (error) {
         console.error(error);

package/src/index.test.ts

@@ -266,8 +266,8 @@ describe('MastraClient Resources', () => {
                 }]
             });
 
-            expect(response).toBeInstanceOf(ReadableStream);
-            const reader = response?.getReader();
+            expect(response.body).toBeInstanceOf(ReadableStream);
+            const reader = response?.body?.getReader();
             expect(reader).toBeDefined();
 
             if (reader) {

package/src/resources/agent.ts

@@ -75,17 +75,17 @@ export class Agent extends BaseResource {
      * @param params - Stream parameters including prompt
      * @returns Promise containing the streamed response
      */
-    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>) {
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<Response> {
         const processedParams = {
             ...params,
             output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
-            stream: true
         };
 
         return this.request(`/api/agents/${this.agentId}/stream`, {
             method: 'POST',
             body: processedParams,
-        }) as T extends undefined ? Promise<ReadableStream<Uint8Array>> : Promise<Response>;
+            stream: true
+        })
     }
 
     /**

package/src/resources/base.ts

@@ -17,10 +17,8 @@ export class BaseResource {
         let lastError: Error | null = null;
         const { baseUrl, retries = 3, backoffMs = 100, maxBackoffMs = 1000, headers = {} } = this.options;
 
-
         let delay = backoffMs;
 
-
         for (let attempt = 0; attempt <= retries; attempt++) {
             try {
                 const response = await fetch(`${baseUrl}${path}`, {
@@ -37,11 +35,9 @@ export class BaseResource {
                     const errorBody = await response.text();
                     let errorMessage = `HTTP error! status: ${response.status}`;
                     try {
-                        // Try to parse as JSON for structured error messages
                         const errorJson = JSON.parse(errorBody);
                         errorMessage += ` - ${JSON.stringify(errorJson)}`;
                     } catch {
-                        // If not JSON, include the raw error body if present
                         if (errorBody) {
                             errorMessage += ` - ${errorBody}`;
                         }
@@ -49,14 +45,8 @@ export class BaseResource {
                     throw new Error(errorMessage);
                 }
 
-                // For streaming responses, return the stream directly
-                if (response.headers.get('Content-Type')?.includes('text/event-stream')) {
-                    return response.body as T;
-                }
-
-                // For unknown responses, return the response as is
-                if (response.headers.get('Content-Type')?.includes('text/x-unknown')) {
-                    return response as T;
+                if (options.stream) {
+                    return response as unknown as T;
                 }
 
                 const data = await response.json();
@@ -68,7 +58,6 @@ export class BaseResource {
                     break;
                 }
 
-                // Wait with exponential backoff
                 await new Promise(resolve => setTimeout(resolve, delay));
                 delay = Math.min(delay * 2, maxBackoffMs);
             }

package/src/types.ts

@@ -19,6 +19,7 @@ export interface RequestOptions {
     method?: string;
     headers?: Record<string, string>;
     body?: any;
+    stream?: boolean;
 }
 
 export interface GetAgentResponse {