@mastra/client-js 0.0.0-feat-tool-input-validation-20250731232758 → 0.0.0-feat-support-ai-sdk-5-again-20250813225910

package/src/resources/agent.ts

@@ -1,15 +1,5 @@
-import {
-  parsePartialJson,
-  processDataStream,
-  type JSONValue,
-  type ReasoningUIPart,
-  type TextUIPart,
-  type ToolInvocation,
-  type ToolInvocationUIPart,
-  type UIMessage,
-  type UseChatOptions,
-} from '@ai-sdk/ui-utils';
-import { Tool, type CoreMessage } from '@mastra/core';
+import { processDataStream, type JSONValue, type ToolInvocation, type UIMessage } from 'ai';
+import { Tool, type CoreMessage, type MastraLanguageModel } from '@mastra/core';
 import { type GenerateReturn } from '@mastra/core/llm';
 import type { JSONSchema7 } from 'json-schema';
 import { ZodSchema } from 'zod';
@@ -213,7 +203,7 @@ export class Agent extends BaseResource {
   }: {
     stream: ReadableStream<Uint8Array>;
     update: (options: { message: UIMessage; data: JSONValue[] | undefined; replaceLastMessage: boolean }) => void;
-    onToolCall?: UseChatOptions['onToolCall'];
+    onToolCall?: any;
     onFinish?: (options: { message: UIMessage | undefined; finishReason: string; usage: string }) => void;
     generateId?: () => string;
     getCurrentDate?: () => Date;
@@ -238,14 +228,14 @@ export class Agent extends BaseResource {
           parts: [],
         };
 
-    let currentTextPart: TextUIPart | undefined = undefined;
-    let currentReasoningPart: ReasoningUIPart | undefined = undefined;
+    let currentTextPart: any | undefined = undefined;
+    let currentReasoningPart: any | undefined = undefined;
     let currentReasoningTextDetail: { type: 'text'; text: string; signature?: string } | undefined = undefined;
 
     function updateToolInvocationPart(toolCallId: string, invocation: ToolInvocation) {
       const part = message.parts.find(
         part => part.type === 'tool-invocation' && part.toolInvocation.toolCallId === toolCallId,
-      ) as ToolInvocationUIPart | undefined;
+      ) as any | undefined;
 
       if (part != null) {
         part.toolInvocation = invocation;
@@ -415,7 +405,13 @@ export class Agent extends BaseResource {
 
         partialToolCall!.text += value.argsTextDelta;
 
-        const { value: partialArgs } = parsePartialJson(partialToolCall!.text);
+        // In v4, we don't have parsePartialJson, so we'll try to parse it directly
+        let partialArgs;
+        try {
+          partialArgs = JSON.parse(partialToolCall!.text);
+        } catch {
+          partialArgs = undefined;
+        }
 
         const invocation = {
           state: 'partial-call',
@@ -671,7 +667,7 @@ export class Agent extends BaseResource {
 
                 const toolInvocationPart = lastMessage?.parts?.find(
                   part => part.type === 'tool-invocation' && part.toolInvocation?.toolCallId === toolCall.toolCallId,
-                ) as ToolInvocationUIPart | undefined;
+                ) as any | undefined;
 
                 if (toolInvocationPart) {
                   toolInvocationPart.toolInvocation = {
@@ -782,4 +778,16 @@ export class Agent extends BaseResource {
   liveEvals(): Promise<GetEvalsByAgentIdResponse> {
     return this.request(`/api/agents/${this.agentId}/evals/live`);
   }
+
+  /**
+   * Updates the model for the agent
+   * @param params - Parameters for updating the model
+   * @returns Promise containing the updated model
+   */
+  updateModel(params: { model: MastraLanguageModel }): Promise<{ message: string }> {
+    return this.request(`/api/agents/${this.agentId}/model`, {
+      method: 'POST',
+      body: params,
+    });
+  }
 }

package/src/resources/base.ts

@@ -33,6 +33,7 @@ export class BaseResource {
             ...options.headers,
             // TODO: Bring this back once we figure out what we/users need to do to make this work with cross-origin requests
             // 'x-mastra-client-type': 'js',
+            'x-ai-sdk-compat': 'v4',
           },
           signal: this.options.abortSignal,
           body:

package/src/resources/memory-thread.ts

@@ -59,6 +59,7 @@ export class MemoryThread extends BaseResource {
     const query = new URLSearchParams({
       agentId: this.agentId,
       ...(params?.limit ? { limit: params.limit.toString() } : {}),
+      ...(params?.format ? { format: params.format } : {}),
     });
     return this.request(`/api/memory/threads/${this.threadId}/messages?${query.toString()}`);
   }

package/src/resources/network.ts

@@ -1,4 +1,4 @@
-import { processDataStream } from '@ai-sdk/ui-utils';
+import { processDataStream } from 'ai';
 import type { GenerateReturn } from '@mastra/core';
 import type { JSONSchema7 } from 'json-schema';
 import type { ZodSchema } from 'zod';

package/src/types.ts

@@ -1,6 +1,6 @@
+import type { UIMessage } from 'ai';
 import type {
   MastraMessageV1,
-  AiMessageType,
   CoreMessage,
   QueryResult,
   StorageThreadType,
@@ -11,7 +11,7 @@ import type {
   PaginationInfo,
   MastraMessageV2,
 } from '@mastra/core';
-import type { AgentGenerateOptions, AgentStreamOptions, ToolsInput, UIMessageWithMetadata } from '@mastra/core/agent';
+import type { AgentGenerateOptions, AgentStreamOptions, ToolsInput } from '@mastra/core/agent';
 import type { BaseLogMessage, LogLevel } from '@mastra/core/logger';
 
 import type { MCPToolType, ServerInfo } from '@mastra/core/mcp';
@@ -70,7 +70,7 @@ export interface GetAgentResponse {
 }
 
 export type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
-  messages: string | string[] | CoreMessage[] | AiMessageType[] | UIMessageWithMetadata[];
+  messages: string | string[] | CoreMessage[] | UIMessage[];
   output?: T;
   experimental_output?: T;
   runtimeContext?: RuntimeContext | Record<string, any>;
@@ -80,7 +80,7 @@ export type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undef
 >;
 
 export type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
-  messages: string | string[] | CoreMessage[] | AiMessageType[] | UIMessageWithMetadata[];
+  messages: string | string[] | CoreMessage[] | UIMessage[];
   output?: T;
   experimental_output?: T;
   runtimeContext?: RuntimeContext | Record<string, any>;
@@ -250,13 +250,17 @@ export interface GetMemoryThreadMessagesParams {
    * Limit the number of messages to retrieve (default: 40)
    */
   limit?: number;
+  /**
+   * Format of UI messages to return (default: aiv5)
+   */
+  format?: 'aiv4' | 'aiv5';
 }
 
 export type GetMemoryThreadMessagesPaginatedParams = Omit<StorageGetMessagesArg, 'threadConfig' | 'threadId'>;
 
 export interface GetMemoryThreadMessagesResponse {
   messages: CoreMessage[];
-  uiMessages: AiMessageType[];
+  uiMessages: UIMessage[];
 }
 
 export type GetMemoryThreadMessagesPaginatedResponse = PaginationInfo & {

package/src/utils/process-client-tools.ts

@@ -14,7 +14,7 @@ export function processClientTools(clientTools: ToolsInput | undefined): ToolsIn
           key,
           {
             ...value,
-            parameters: value.parameters ? zodToJsonSchema(value.parameters) : undefined,
+            inputSchema: value.inputSchema ? zodToJsonSchema(value.inputSchema) : undefined,
           },
         ];
       } else {

package/tsconfig.build.json

@@ -0,0 +1,9 @@
+{
+  "extends": ["./tsconfig.json", "../../tsconfig.build.json"],
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "**/*.test.ts", "src/**/*.mock.ts"]
+}

package/tsconfig.json

@@ -1,5 +1,5 @@
 {
   "extends": "../../tsconfig.node.json",
-  "include": ["src/**/*"],
+  "include": ["src/**/*", "tsup.config.ts"],
   "exclude": ["node_modules", "**/*.test.ts"]
 }

package/tsup.config.ts

@@ -0,0 +1,17 @@
+import { defineConfig } from 'tsup';
+import { generateTypes } from '@internal/types-builder';
+
+export default defineConfig({
+  entry: ['src/index.ts'],
+  format: ['esm', 'cjs'],
+  clean: true,
+  dts: false,
+  splitting: true,
+  treeshake: {
+    preset: 'smallest',
+  },
+  sourcemap: true,
+  onSuccess: async () => {
+    await generateTypes(process.cwd());
+  },
+});