@mastra/client-js 0.1.0-alpha.3 → 0.1.0-alpha.4

package/README.md

@@ -25,7 +25,7 @@ async function main() {
     
     // Generate a response
     const response = await agent.generate({
-        prompt: "What's the weather like today?"
+        messages: [{ role: 'user', content: "What's the weather like today?" }],
     });
     
     console.log(response);

package/dist/index.d.mts

@@ -1,4 +1,5 @@
-import { StorageThreadType, CoreMessage, AiMessageType, MessageType, StepAction, StepGraph, QueryResult, BaseLogMessage, GenerateReturn, StreamReturn } from '@mastra/core';
+import { MessageType, OutputType, StorageThreadType, CoreMessage, AiMessageType, StepAction, StepGraph, QueryResult, BaseLogMessage, GenerateReturn, StreamReturn } from '@mastra/core';
+import { JSONSchema7 } from 'json-schema';
 import { StorageThreadType as StorageThreadType$1 } from '@mastra/core/memory';
 
 interface ClientOptions {
@@ -19,6 +20,18 @@ interface GetAgentResponse {
     instructions: string;
     tools: Record<string, GetToolResponse>;
 }
+interface GenerateParams<T extends JSONSchema7 | undefined = undefined> {
+    messages: MessageType[];
+    threadId?: string;
+    resourceid?: string;
+    output?: OutputType | T;
+}
+interface StreamParams<T extends JSONSchema7 | undefined = undefined> {
+    messages: MessageType[];
+    threadId?: string;
+    resourceid?: string;
+    output?: OutputType | T;
+}
 interface GetEvalsByAgentIdResponse extends GetAgentResponse {
     evals: any[];
 }
@@ -109,13 +122,13 @@ declare class Agent {
      * @param params - Generation parameters including prompt
      * @returns Promise containing the generated response
      */
-    generate<T>(params: any): Promise<GenerateReturn<T>>;
+    generate<T extends JSONSchema7 | undefined = undefined>(params: GenerateParams<T>): Promise<GenerateReturn<T>>;
     /**
      * Streams a response from the agent
      * @param params - Stream parameters including prompt
      * @returns Promise containing the streamed response
      */
-    stream<T>(params: any): Promise<StreamReturn<T>>;
+    stream<T extends JSONSchema7 | undefined = undefined>(params: StreamParams<T>): Promise<StreamReturn<T>>;
     /**
      * Gets details about a specific tool available to the agent
      * @param toolId - ID of the tool to retrieve
@@ -343,4 +356,4 @@ declare class MastraClient {
     getLogForRun(params: GetLogParams): Promise<GetLogsResponse>;
 }
 
-export { type ClientOptions, type CreateIndexParams, type CreateMemoryThreadParams, type CreateMemoryThreadResponse, type GetAgentResponse, type GetEvalsByAgentIdResponse, type GetLogParams, type GetLogsParams, type GetLogsResponse, type GetMemoryThreadMessagesResponse, type GetMemoryThreadParams, type GetMemoryThreadResponse, type GetToolResponse, type GetVectorIndexResponse, type GetWorkflowResponse, MastraClient, type QueryVectorParams, type QueryVectorResponse, type RequestFunction, type RequestOptions, type SaveMessageToMemoryParams, type SaveMessageToMemoryResponse, type UpdateMemoryThreadParams, type UpsertVectorParams };
+export { type ClientOptions, type CreateIndexParams, type CreateMemoryThreadParams, type CreateMemoryThreadResponse, type GenerateParams, type GetAgentResponse, type GetEvalsByAgentIdResponse, type GetLogParams, type GetLogsParams, type GetLogsResponse, type GetMemoryThreadMessagesResponse, type GetMemoryThreadParams, type GetMemoryThreadResponse, type GetToolResponse, type GetVectorIndexResponse, type GetWorkflowResponse, MastraClient, type QueryVectorParams, type QueryVectorResponse, type RequestFunction, type RequestOptions, type SaveMessageToMemoryParams, type SaveMessageToMemoryResponse, type StreamParams, type UpdateMemoryThreadParams, type UpsertVectorParams };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/client-js",
-  "version": "0.1.0-alpha.3",
+  "version": "0.1.0-alpha.4",
   "description": "The official TypeScript library for the Mastra Client API",
   "author": "",
   "types": "dist/index.d.ts",
@@ -17,12 +17,14 @@
   "repository": "github:mastra-ai/client-js",
   "license": "ISC",
   "dependencies": {
-    "@mastra/core": "^0.2.0-alpha.83"
+    "@mastra/core": "^0.2.0-alpha.83",
+    "json-schema": "^0.4.0"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.26.0",
     "@babel/preset-typescript": "^7.26.0",
     "@tsconfig/recommended": "^1.0.7",
+    "@types/json-schema": "^7.0.15",
     "@types/node": "^22.9.0",
     "tsup": "^8.0.1",
     "typescript": "^5.7.3",

package/src/index.test.ts

@@ -198,13 +198,23 @@ describe('MastraClient Resources', () => {
             };
             mockFetchResponse(mockResponse);
 
-            const result = await agent.generate({ prompt: 'Test prompt' });
+            const result = await agent.generate({
+                messages: [],
+                threadId: 'test-thread',
+                resourceid: 'test-resource',
+                output: {}
+            });
             expect(result).toEqual(mockResponse);
             expect(global.fetch).toHaveBeenCalledWith(
                 `${client.baseUrl}/api/agents/test-agent/generate`,
                 expect.objectContaining({
                     method: 'POST',
-                    body: JSON.stringify({ prompt: 'Test prompt' })
+                    body: JSON.stringify({
+                        messages: [],
+                        threadId: 'test-thread',
+                        resourceid: 'test-resource',
+                        output: {}
+                    })
                 })
             );
         });
@@ -215,13 +225,24 @@ describe('MastraClient Resources', () => {
                 chunks: ['chunk1', 'chunk2']
             };
             mockFetchResponse(mockResponse);
-            const result = await agent.stream({ prompt: 'Test prompt' });
+            const result = await agent.stream({
+                messages: [],
+                threadId: 'test-thread',
+                resourceid: 'test-resource',
+                output: undefined
+            });
             expect(result).toEqual(mockResponse);
             expect(global.fetch).toHaveBeenCalledWith(
                 `${client.baseUrl}/api/agents/test-agent/generate`,
                 expect.objectContaining({
                     method: 'POST',
-                    body: JSON.stringify({ prompt: 'Test prompt', stream: true })
+                    body: JSON.stringify({
+                        messages: [],
+                        threadId: 'test-thread',
+                        resourceid: 'test-resource',
+                        output: undefined,
+                        stream: true
+                    })
                 })
             );
         });

package/src/resources/agent.ts

@@ -1,13 +1,16 @@
 import type {
+    GenerateParams,
     GetAgentResponse,
     GetEvalsByAgentIdResponse,
     GetToolResponse,
     RequestFunction,
+    StreamParams,
 } from '../types';
 import type {
     GenerateReturn,
     StreamReturn,
 } from '@mastra/core';
+import type { JSONSchema7 } from 'json-schema';
 
 export class AgentTool {
     constructor(
@@ -48,7 +51,7 @@ export class Agent {
      * @param params - Generation parameters including prompt
      * @returns Promise containing the generated response
      */
-    generate<T>(params: any): Promise<GenerateReturn<T>> {
+    generate<T extends JSONSchema7 | undefined = undefined>(params: GenerateParams<T>): Promise<GenerateReturn<T>> {
         return this.request(`/api/agents/${this.agentId}/generate`, {
             method: 'POST',
             body: params,
@@ -60,7 +63,7 @@ export class Agent {
      * @param params - Stream parameters including prompt
      * @returns Promise containing the streamed response
      */
-    stream<T>(params: any): Promise<StreamReturn<T>> {
+    stream<T extends JSONSchema7 | undefined = undefined>(params: StreamParams<T>): Promise<StreamReturn<T>> {
         return this.request(`/api/agents/${this.agentId}/generate`, {
             method: 'POST',
             body: { ...params, stream: true },

package/src/types.ts

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