@mastra/react 0.0.2 → 0.0.3

package/dist/src/agent/hooks.d.ts

@@ -1,4 +1,5 @@
 import { ModelSettings } from './types';
+import { UIMessage } from '@ai-sdk/react';
 import { CoreUserMessage } from '@mastra/core/llm';
 import { RuntimeContext } from '@mastra/core/runtime-context';
 import { ChunkType, NetworkChunkType } from '@mastra/core/stream';
@@ -6,27 +7,29 @@ export interface MastraChatProps<TMessage> {
     agentId: string;
     initializeMessages?: () => TMessage[];
 }
-export interface StreamVNextArgs<TMessage> {
+interface SharedArgs {
     coreUserMessages: CoreUserMessage[];
     runtimeContext?: RuntimeContext;
     threadId?: string;
-    onChunk: (chunk: ChunkType, conversation: TMessage[]) => TMessage[];
     modelSettings?: ModelSettings;
     signal?: AbortSignal;
 }
-export interface NetworkArgs<TMessage> {
-    coreUserMessages: CoreUserMessage[];
-    runtimeContext?: RuntimeContext;
-    threadId?: string;
+export type GenerateArgs<TMessage> = SharedArgs & {
+    onFinish: (messages: UIMessage[]) => TMessage[];
+};
+export type StreamArgs<TMessage> = SharedArgs & {
+    onChunk: (chunk: ChunkType, conversation: TMessage[]) => TMessage[];
+};
+export type NetworkArgs<TMessage> = SharedArgs & {
     onNetworkChunk: (chunk: NetworkChunkType, conversation: TMessage[]) => TMessage[];
-    modelSettings?: ModelSettings;
-    signal?: AbortSignal;
-}
+};
 export declare const useChat: <TMessage>({ agentId, initializeMessages }: MastraChatProps<TMessage>) => {
     network: ({ coreUserMessages, runtimeContext, threadId, onNetworkChunk, modelSettings, signal, }: NetworkArgs<TMessage>) => Promise<void>;
-    streamVNext: ({ coreUserMessages, runtimeContext, threadId, onChunk, modelSettings, signal, }: StreamVNextArgs<TMessage>) => Promise<void>;
+    stream: ({ coreUserMessages, runtimeContext, threadId, onChunk, modelSettings, signal, }: StreamArgs<TMessage>) => Promise<void>;
+    generate: ({ coreUserMessages, runtimeContext, threadId, modelSettings, signal, onFinish, }: GenerateArgs<TMessage>) => Promise<void>;
     isRunning: boolean;
     messages: TMessage[];
     setMessages: import('react').Dispatch<import('react').SetStateAction<TMessage[]>>;
     cancelRun: () => void;
 };
+export {};

package/dist/src/agent/types.d.ts

@@ -10,7 +10,6 @@ export interface ModelSettings {
     instructions?: string;
     providerOptions?: Record<string, unknown>;
     chatWithGenerate?: boolean;
-    chatWithGenerateVNext?: boolean;
-    chatWithStreamVNext?: boolean;
+    chatWithStream?: boolean;
     chatWithNetwork?: boolean;
 }

package/dist/src/lib/ai-sdk/index.d.ts

@@ -1,2 +1,3 @@
 export * from './toUIMessage';
 export * from './toNetworkUIMessage';
+export * from './toAssistantUIMessage';

package/dist/src/lib/ai-sdk/toAssistantUIMessage.d.ts

@@ -0,0 +1,14 @@
+import { ThreadMessageLike } from '@assistant-ui/react';
+import { MastraUIMessage } from './toUIMessage';
+/**
+ * Converts a Mastra UIMessage (from AI SDK) to a ThreadMessageLike format compatible with @assistant-ui/react.
+ *
+ * This function handles UIMessages from three sources:
+ * - agent.generate: Full output with all message parts
+ * - toUIMessage: Streaming chunks accumulated into UIMessages
+ * - toNetworkUIMessage: Network execution events accumulated into UIMessages
+ *
+ * @param message - The MastraUIMessage to convert
+ * @returns A ThreadMessageLike compatible with @assistant-ui/react
+ */
+export declare const toAssistantUIMessage: (message: MastraUIMessage) => ThreadMessageLike;

package/dist/src/lib/ai-sdk/toAssistantUIMessage.test.d.ts

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/react",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/mastra-ai/mastra.git",
@@ -22,7 +22,7 @@
   "dependencies": {
     "react": "^19.1.1",
     "react-dom": "^19.1.1",
-    "@mastra/client-js": "0.14.1"
+    "@mastra/client-js": "0.15.0"
   },
   "devDependencies": {
     "@ai-sdk/react": "^2.0.57",
@@ -37,7 +37,7 @@
     "vite": "^7.1.7",
     "vite-plugin-dts": "^4.5.4",
     "vitest": "^3.2.4",
-    "@mastra/core": "0.19.1"
+    "@mastra/core": "0.20.0"
   },
   "scripts": {
     "build": "vite build",