@mastra/client-js 0.1.0-alpha.5 → 0.1.0-alpha.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/client-js",
-  "version": "0.1.0-alpha.5",
+  "version": "0.1.0-alpha.6",
   "description": "The official TypeScript library for the Mastra Client API",
   "author": "",
   "types": "dist/index.d.mts",
@@ -18,7 +18,9 @@
   "license": "ISC",
   "dependencies": {
     "@mastra/core": "^0.2.0-alpha.83",
-    "json-schema": "^0.4.0"
+    "json-schema": "^0.4.0",
+    "zod": "^3.24.1",
+    "zod-to-json-schema": "^3.24.1"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.26.0",

package/src/resources/agent.ts

@@ -11,6 +11,8 @@ import type {
     StreamReturn,
 } from '@mastra/core';
 import type { JSONSchema7 } from 'json-schema';
+import { ZodSchema } from "zod";
+import { zodToJsonSchema } from 'zod-to-json-schema';
 
 export class AgentTool {
     constructor(
@@ -51,10 +53,15 @@ export class Agent {
      * @param params - Generation parameters including prompt
      * @returns Promise containing the generated response
      */
-    generate<T extends JSONSchema7 | undefined = undefined>(params: GenerateParams<T>): Promise<GenerateReturn<T>> {
+    generate<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: GenerateParams<T>): Promise<GenerateReturn<T>> {
+        const processedParams = {
+            ...params,
+            output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output
+        };
+
         return this.request(`/api/agents/${this.agentId}/generate`, {
             method: 'POST',
-            body: params,
+            body: processedParams,
         });
     }
 
@@ -63,10 +70,16 @@ export class Agent {
      * @param params - Stream parameters including prompt
      * @returns Promise containing the streamed response
      */
-    stream<T extends JSONSchema7 | undefined = undefined>(params: StreamParams<T>): Promise<StreamReturn<T>> {
+    stream<T extends JSONSchema7 | ZodSchema | undefined = undefined>(params: StreamParams<T>): Promise<StreamReturn<T>> {
+        const processedParams = {
+            ...params,
+            output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
+            stream: true
+        };
+
         return this.request(`/api/agents/${this.agentId}/generate`, {
             method: 'POST',
-            body: { ...params, stream: true },
+            body: processedParams,
         });
     }
 

package/src/types.ts

@@ -1,5 +1,6 @@
 import type { MessageType, AiMessageType, CoreMessage, QueryResult, StepAction, StepGraph, StorageThreadType, BaseLogMessage, OutputType } from "@mastra/core";
 import type { JSONSchema7 } from 'json-schema';
+import { ZodSchema } from "zod";
 
 export interface ClientOptions {
     baseUrl: string;
@@ -22,14 +23,14 @@ export interface GetAgentResponse {
     tools: Record<string, GetToolResponse>;
 }
 
-export interface GenerateParams<T extends JSONSchema7 | undefined = undefined> {
+export interface GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> {
     messages: MessageType[];
     threadId?: string;
     resourceid?: string;
     output?: OutputType | T
 }
 
-export interface StreamParams<T extends JSONSchema7 | undefined = undefined> {
+export interface StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> {
     messages: MessageType[];
     threadId?: string;
     resourceid?: string;