@mastra/client-js 0.1.20-alpha.5 → 0.1.20

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @mastra/client-js@0.1.20-alpha.5 build /home/runner/work/mastra/mastra/client-sdks/client-js
+> @mastra/client-js@0.1.20-alpha.6 build /home/runner/work/mastra/mastra/client-sdks/client-js
 > tsup src/index.ts --format esm,cjs --dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
@@ -9,11 +9,11 @@
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 29.18 KB
-ESM ⚡️ Build success in 499ms
-CJS dist/index.cjs 29.36 KB
-CJS ⚡️ Build success in 502ms
+CJS dist/index.cjs 34.65 KB
+CJS ⚡️ Build success in 1542ms
+ESM dist/index.js 34.43 KB
+ESM ⚡️ Build success in 1566ms
 DTS Build start
-DTS ⚡️ Build success in 14118ms
-DTS dist/index.d.ts  24.97 KB
-DTS dist/index.d.cts 24.97 KB
+DTS ⚡️ Build success in 14884ms
+DTS dist/index.d.ts  25.12 KB
+DTS dist/index.d.cts 25.12 KB

package/CHANGELOG.md

@@ -1,5 +1,50 @@
 # @mastra/client-js
 
+## 0.1.20
+
+### Patch Changes
+
+- 2cf3b8f: dependencies updates:
+  - Updated dependency [`zod@^3.24.3` ↗︎](https://www.npmjs.com/package/zod/v/3.24.3) (from `^3.24.2`, in `dependencies`)
+  - Updated dependency [`zod-to-json-schema@^3.24.5` ↗︎](https://www.npmjs.com/package/zod-to-json-schema/v/3.24.5) (from `^3.24.3`, in `dependencies`)
+- 4155f47: Add parameters to filter workflow runs
+  Add fromDate and toDate to telemetry parameters
+- 254f5c3: Audit, cleanup MastraClient
+- 0097d50: Add serializedStepGraph to vNext workflow
+  Return serializedStepGraph from vNext workflow
+  Use serializedStepGraph in vNext workflow graph
+- 2429c74: Add get workflow runs api to client-js
+- 2e4f8e9: Agui client integration
+- Updated dependencies [6052aa6]
+- Updated dependencies [967b41c]
+- Updated dependencies [3d2fb5c]
+- Updated dependencies [26738f4]
+- Updated dependencies [4155f47]
+- Updated dependencies [7eeb2bc]
+- Updated dependencies [b804723]
+- Updated dependencies [8607972]
+- Updated dependencies [ccef9f9]
+- Updated dependencies [0097d50]
+- Updated dependencies [7eeb2bc]
+- Updated dependencies [17826a9]
+- Updated dependencies [7d8b7c7]
+- Updated dependencies [fba031f]
+- Updated dependencies [3a5f1e1]
+- Updated dependencies [51e6923]
+- Updated dependencies [8398d89]
+  - @mastra/core@0.9.2
+
+## 0.1.20-alpha.6
+
+### Patch Changes
+
+- 2e4f8e9: Agui client integration
+- Updated dependencies [6052aa6]
+- Updated dependencies [7d8b7c7]
+- Updated dependencies [3a5f1e1]
+- Updated dependencies [8398d89]
+  - @mastra/core@0.9.2-alpha.6
+
 ## 0.1.20-alpha.5
 
 ### Patch Changes

package/dist/index.cjs

@@ -1,10 +1,175 @@
 'use strict';
 
+var client = require('@ag-ui/client');
+var rxjs = require('rxjs');
 var uiUtils = require('@ai-sdk/ui-utils');
 var zod = require('zod');
 var zodToJsonSchema = require('zod-to-json-schema');
 
-// src/resources/agent.ts
+// src/adapters/agui.ts
+var AGUIAdapter = class extends client.AbstractAgent {
+  agent;
+  resourceId;
+  constructor({ agent, agentId, resourceId, ...rest }) {
+    super({
+      agentId,
+      ...rest
+    });
+    this.agent = agent;
+    this.resourceId = resourceId;
+  }
+  run(input) {
+    return new rxjs.Observable((subscriber) => {
+      const convertedMessages = convertMessagesToMastraMessages(input.messages);
+      subscriber.next({
+        type: client.EventType.RUN_STARTED,
+        threadId: input.threadId,
+        runId: input.runId
+      });
+      this.agent.stream({
+        threadId: input.threadId,
+        resourceId: this.resourceId ?? "",
+        runId: input.runId,
+        messages: convertedMessages,
+        clientTools: input.tools.reduce(
+          (acc, tool) => {
+            acc[tool.name] = {
+              id: tool.name,
+              description: tool.description,
+              inputSchema: tool.parameters
+            };
+            return acc;
+          },
+          {}
+        )
+      }).then((response) => {
+        let currentMessageId = void 0;
+        return response.processDataStream({
+          onTextPart: (text) => {
+            if (currentMessageId === void 0) {
+              currentMessageId = generateUUID();
+              const message2 = {
+                type: client.EventType.TEXT_MESSAGE_START,
+                messageId: currentMessageId,
+                role: "assistant"
+              };
+              subscriber.next(message2);
+            }
+            const message = {
+              type: client.EventType.TEXT_MESSAGE_CONTENT,
+              messageId: currentMessageId,
+              delta: text
+            };
+            subscriber.next(message);
+          },
+          onFinishMessagePart: (message) => {
+            console.log("onFinishMessagePart", message);
+            if (currentMessageId !== void 0) {
+              const message2 = {
+                type: client.EventType.TEXT_MESSAGE_END,
+                messageId: currentMessageId
+              };
+              subscriber.next(message2);
+            }
+            subscriber.next({
+              type: client.EventType.RUN_FINISHED,
+              threadId: input.threadId,
+              runId: input.runId
+            });
+            subscriber.complete();
+          },
+          onToolCallPart(streamPart) {
+            const parentMessageId = currentMessageId || generateUUID();
+            subscriber.next({
+              type: client.EventType.TOOL_CALL_START,
+              toolCallId: streamPart.toolCallId,
+              toolCallName: streamPart.toolName,
+              parentMessageId
+            });
+            subscriber.next({
+              type: client.EventType.TOOL_CALL_ARGS,
+              toolCallId: streamPart.toolCallId,
+              delta: JSON.stringify(streamPart.args),
+              parentMessageId
+            });
+            subscriber.next({
+              type: client.EventType.TOOL_CALL_END,
+              toolCallId: streamPart.toolCallId,
+              parentMessageId
+            });
+          }
+        });
+      }).catch((error) => {
+        console.log("error", error);
+        subscriber.error(error);
+      });
+      return () => {
+      };
+    });
+  }
+};
+function generateUUID() {
+  if (typeof crypto !== "undefined") {
+    if (typeof crypto.randomUUID === "function") {
+      return crypto.randomUUID();
+    }
+    if (typeof crypto.getRandomValues === "function") {
+      const buffer = new Uint8Array(16);
+      crypto.getRandomValues(buffer);
+      buffer[6] = buffer[6] & 15 | 64;
+      buffer[8] = buffer[8] & 63 | 128;
+      let hex = "";
+      for (let i = 0; i < 16; i++) {
+        hex += buffer[i].toString(16).padStart(2, "0");
+        if (i === 3 || i === 5 || i === 7 || i === 9) hex += "-";
+      }
+      return hex;
+    }
+  }
+  return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
+    const r = Math.random() * 16 | 0;
+    const v = c === "x" ? r : r & 3 | 8;
+    return v.toString(16);
+  });
+}
+function convertMessagesToMastraMessages(messages) {
+  const result = [];
+  for (const message of messages) {
+    if (message.role === "assistant") {
+      const parts = message.content ? [{ type: "text", text: message.content }] : [];
+      for (const toolCall of message.toolCalls ?? []) {
+        parts.push({
+          type: "tool-call",
+          toolCallId: toolCall.id,
+          toolName: toolCall.function.name,
+          args: JSON.parse(toolCall.function.arguments)
+        });
+      }
+      result.push({
+        role: "assistant",
+        content: parts
+      });
+    } else if (message.role === "user") {
+      result.push({
+        role: "user",
+        content: message.content || ""
+      });
+    } else if (message.role === "tool") {
+      result.push({
+        role: "tool",
+        content: [
+          {
+            type: "tool-result",
+            toolCallId: message.toolCallId,
+            toolName: "unknown",
+            result: message.content
+          }
+        ]
+      });
+    }
+  }
+  return result;
+}
 
 // src/resources/base.ts
 var BaseResource = class {
@@ -778,6 +943,21 @@ var MastraClient = class extends BaseResource {
   getAgents() {
     return this.request("/api/agents");
   }
+  async getAGUI({ resourceId }) {
+    const agents = await this.getAgents();
+    return Object.entries(agents).reduce(
+      (acc, [agentId]) => {
+        const agent = this.getAgent(agentId);
+        acc[agentId] = new AGUIAdapter({
+          agentId,
+          agent,
+          resourceId
+        });
+        return acc;
+      },
+      {}
+    );
+  }
   /**
    * Gets an agent instance by ID
    * @param agentId - ID of the agent to retrieve

package/dist/index.d.cts

@@ -1,3 +1,4 @@
+import { AbstractAgent } from '@ag-ui/client';
 import { processDataStream } from '@ai-sdk/ui-utils';
 import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, WorkflowRuns, WorkflowRunResult as WorkflowRunResult$1, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
 import { JSONSchema7 } from 'json-schema';
@@ -609,6 +610,9 @@ declare class MastraClient extends BaseResource {
      * @returns Promise containing map of agent IDs to agent details
      */
     getAgents(): Promise<Record<string, GetAgentResponse>>;
+    getAGUI({ resourceId }: {
+        resourceId: string;
+    }): Promise<Record<string, AbstractAgent>>;
     /**
      * Gets an agent instance by ID
      * @param agentId - ID of the agent to retrieve

package/dist/index.d.ts

@@ -1,3 +1,4 @@
+import { AbstractAgent } from '@ag-ui/client';
 import { processDataStream } from '@ai-sdk/ui-utils';
 import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, WorkflowRuns, WorkflowRunResult as WorkflowRunResult$1, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
 import { JSONSchema7 } from 'json-schema';
@@ -609,6 +610,9 @@ declare class MastraClient extends BaseResource {
      * @returns Promise containing map of agent IDs to agent details
      */
     getAgents(): Promise<Record<string, GetAgentResponse>>;
+    getAGUI({ resourceId }: {
+        resourceId: string;
+    }): Promise<Record<string, AbstractAgent>>;
     /**
      * Gets an agent instance by ID
      * @param agentId - ID of the agent to retrieve

package/dist/index.js

@@ -1,8 +1,173 @@
+import { AbstractAgent, EventType } from '@ag-ui/client';
+import { Observable } from 'rxjs';
 import { processDataStream } from '@ai-sdk/ui-utils';
 import { ZodSchema } from 'zod';
 import { zodToJsonSchema } from 'zod-to-json-schema';
 
-// src/resources/agent.ts
+// src/adapters/agui.ts
+var AGUIAdapter = class extends AbstractAgent {
+  agent;
+  resourceId;
+  constructor({ agent, agentId, resourceId, ...rest }) {
+    super({
+      agentId,
+      ...rest
+    });
+    this.agent = agent;
+    this.resourceId = resourceId;
+  }
+  run(input) {
+    return new Observable((subscriber) => {
+      const convertedMessages = convertMessagesToMastraMessages(input.messages);
+      subscriber.next({
+        type: EventType.RUN_STARTED,
+        threadId: input.threadId,
+        runId: input.runId
+      });
+      this.agent.stream({
+        threadId: input.threadId,
+        resourceId: this.resourceId ?? "",
+        runId: input.runId,
+        messages: convertedMessages,
+        clientTools: input.tools.reduce(
+          (acc, tool) => {
+            acc[tool.name] = {
+              id: tool.name,
+              description: tool.description,
+              inputSchema: tool.parameters
+            };
+            return acc;
+          },
+          {}
+        )
+      }).then((response) => {
+        let currentMessageId = void 0;
+        return response.processDataStream({
+          onTextPart: (text) => {
+            if (currentMessageId === void 0) {
+              currentMessageId = generateUUID();
+              const message2 = {
+                type: EventType.TEXT_MESSAGE_START,
+                messageId: currentMessageId,
+                role: "assistant"
+              };
+              subscriber.next(message2);
+            }
+            const message = {
+              type: EventType.TEXT_MESSAGE_CONTENT,
+              messageId: currentMessageId,
+              delta: text
+            };
+            subscriber.next(message);
+          },
+          onFinishMessagePart: (message) => {
+            console.log("onFinishMessagePart", message);
+            if (currentMessageId !== void 0) {
+              const message2 = {
+                type: EventType.TEXT_MESSAGE_END,
+                messageId: currentMessageId
+              };
+              subscriber.next(message2);
+            }
+            subscriber.next({
+              type: EventType.RUN_FINISHED,
+              threadId: input.threadId,
+              runId: input.runId
+            });
+            subscriber.complete();
+          },
+          onToolCallPart(streamPart) {
+            const parentMessageId = currentMessageId || generateUUID();
+            subscriber.next({
+              type: EventType.TOOL_CALL_START,
+              toolCallId: streamPart.toolCallId,
+              toolCallName: streamPart.toolName,
+              parentMessageId
+            });
+            subscriber.next({
+              type: EventType.TOOL_CALL_ARGS,
+              toolCallId: streamPart.toolCallId,
+              delta: JSON.stringify(streamPart.args),
+              parentMessageId
+            });
+            subscriber.next({
+              type: EventType.TOOL_CALL_END,
+              toolCallId: streamPart.toolCallId,
+              parentMessageId
+            });
+          }
+        });
+      }).catch((error) => {
+        console.log("error", error);
+        subscriber.error(error);
+      });
+      return () => {
+      };
+    });
+  }
+};
+function generateUUID() {
+  if (typeof crypto !== "undefined") {
+    if (typeof crypto.randomUUID === "function") {
+      return crypto.randomUUID();
+    }
+    if (typeof crypto.getRandomValues === "function") {
+      const buffer = new Uint8Array(16);
+      crypto.getRandomValues(buffer);
+      buffer[6] = buffer[6] & 15 | 64;
+      buffer[8] = buffer[8] & 63 | 128;
+      let hex = "";
+      for (let i = 0; i < 16; i++) {
+        hex += buffer[i].toString(16).padStart(2, "0");
+        if (i === 3 || i === 5 || i === 7 || i === 9) hex += "-";
+      }
+      return hex;
+    }
+  }
+  return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
+    const r = Math.random() * 16 | 0;
+    const v = c === "x" ? r : r & 3 | 8;
+    return v.toString(16);
+  });
+}
+function convertMessagesToMastraMessages(messages) {
+  const result = [];
+  for (const message of messages) {
+    if (message.role === "assistant") {
+      const parts = message.content ? [{ type: "text", text: message.content }] : [];
+      for (const toolCall of message.toolCalls ?? []) {
+        parts.push({
+          type: "tool-call",
+          toolCallId: toolCall.id,
+          toolName: toolCall.function.name,
+          args: JSON.parse(toolCall.function.arguments)
+        });
+      }
+      result.push({
+        role: "assistant",
+        content: parts
+      });
+    } else if (message.role === "user") {
+      result.push({
+        role: "user",
+        content: message.content || ""
+      });
+    } else if (message.role === "tool") {
+      result.push({
+        role: "tool",
+        content: [
+          {
+            type: "tool-result",
+            toolCallId: message.toolCallId,
+            toolName: "unknown",
+            result: message.content
+          }
+        ]
+      });
+    }
+  }
+  return result;
+}
 
 // src/resources/base.ts
 var BaseResource = class {
@@ -776,6 +941,21 @@ var MastraClient = class extends BaseResource {
   getAgents() {
     return this.request("/api/agents");
   }
+  async getAGUI({ resourceId }) {
+    const agents = await this.getAgents();
+    return Object.entries(agents).reduce(
+      (acc, [agentId]) => {
+        const agent = this.getAgent(agentId);
+        acc[agentId] = new AGUIAdapter({
+          agentId,
+          agent,
+          resourceId
+        });
+        return acc;
+      },
+      {}
+    );
+  }
   /**
    * Gets an agent instance by ID
    * @param agentId - ID of the agent to retrieve

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/client-js",
-  "version": "0.1.20-alpha.5",
+  "version": "0.1.20",
   "description": "The official TypeScript library for the Mastra Client API",
   "author": "",
   "type": "module",
@@ -22,11 +22,13 @@
   "repository": "github:mastra-ai/client-js",
   "license": "Elastic-2.0",
   "dependencies": {
+    "@ag-ui/client": "^0.0.27",
     "@ai-sdk/ui-utils": "^1.1.19",
     "json-schema": "^0.4.0",
-    "zod": "^3.24.2",
-    "zod-to-json-schema": "^3.24.3",
-    "@mastra/core": "^0.9.2-alpha.5"
+    "rxjs": "7.8.1",
+    "zod": "^3.24.3",
+    "zod-to-json-schema": "^3.24.5",
+    "@mastra/core": "^0.9.2"
   },
   "peerDependencies": {
     "zod": "^3.24.2"
@@ -40,7 +42,7 @@
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
     "vitest": "^3.1.2",
-    "@internal/lint": "0.0.2"
+    "@internal/lint": "0.0.3"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --dts --clean --treeshake=smallest --splitting",

package/src/adapters/agui.test.ts

@@ -0,0 +1,167 @@
+import type { Message } from '@ag-ui/client';
+import { describe, it, expect } from 'vitest';
+import { generateUUID, convertMessagesToMastraMessages } from './agui';
+
+describe('generateUUID', () => {
+  it('should generate a valid UUID v4 string', () => {
+    const uuid = generateUUID();
+    // Check UUID format (8-4-4-4-12 hex digits)
+    expect(uuid).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i);
+  });
+
+  it('should generate unique UUIDs', () => {
+    const uuids = new Set();
+    for (let i = 0; i < 100; i++) {
+      uuids.add(generateUUID());
+    }
+    // All UUIDs should be unique
+    expect(uuids.size).toBe(100);
+  });
+});
+
+describe('convertMessagesToMastraMessages', () => {
+  it('should convert user messages correctly', () => {
+    const messages: Message[] = [
+      {
+        id: '1',
+        role: 'user',
+        content: 'Hello, world!',
+      },
+    ];
+
+    const result = convertMessagesToMastraMessages(messages);
+
+    expect(result).toEqual([
+      {
+        role: 'user',
+        content: 'Hello, world!',
+      },
+    ]);
+  });
+
+  it('should convert assistant messages correctly', () => {
+    const messages: Message[] = [
+      {
+        id: '1',
+        role: 'assistant',
+        content: 'Hello, I am an assistant',
+      },
+    ];
+
+    const result = convertMessagesToMastraMessages(messages);
+
+    expect(result).toEqual([
+      {
+        role: 'assistant',
+        content: [{ type: 'text', text: 'Hello, I am an assistant' }],
+      },
+    ]);
+  });
+
+  it('should convert assistant messages with tool calls correctly', () => {
+    const messages: Message[] = [
+      {
+        id: '1',
+        role: 'assistant',
+        content: undefined,
+        toolCalls: [
+          {
+            id: 'tool-call-1',
+            type: 'function',
+            function: {
+              name: 'getWeather',
+              arguments: '{"location":"San Francisco"}',
+            },
+          },
+        ],
+      },
+    ];
+
+    const result = convertMessagesToMastraMessages(messages);
+
+    expect(result).toEqual([
+      {
+        role: 'assistant',
+        content: [
+          {
+            type: 'tool-call',
+            toolCallId: 'tool-call-1',
+            toolName: 'getWeather',
+            args: { location: 'San Francisco' },
+          },
+        ],
+      },
+    ]);
+  });
+
+  it('should convert tool messages correctly', () => {
+    const messages: Message[] = [
+      {
+        id: '1',
+        role: 'tool',
+        toolCallId: 'tool-call-1',
+        content: '{"temperature":72,"unit":"F"}',
+      },
+    ];
+
+    const result = convertMessagesToMastraMessages(messages);
+
+    expect(result).toEqual([
+      {
+        role: 'tool',
+        content: [
+          {
+            type: 'tool-result',
+            toolCallId: 'tool-call-1',
+            toolName: 'unknown',
+            result: '{"temperature":72,"unit":"F"}',
+          },
+        ],
+      },
+    ]);
+  });
+
+  it('should convert a complex conversation correctly', () => {
+    const messages: Message[] = [
+      {
+        id: '1',
+        role: 'user',
+        content: "What's the weather in San Francisco?",
+      },
+      {
+        id: '2',
+        role: 'assistant',
+        content: undefined,
+        toolCalls: [
+          {
+            id: 'tool-call-1',
+            type: 'function',
+            function: {
+              name: 'getWeather',
+              arguments: '{"location":"San Francisco"}',
+            },
+          },
+        ],
+      },
+      {
+        id: '3',
+        role: 'tool',
+        toolCallId: 'tool-call-1',
+        content: '{"temperature":72,"unit":"F"}',
+      },
+      {
+        id: '4',
+        role: 'assistant',
+        content: 'The weather in San Francisco is 72°F.',
+      },
+    ];
+
+    const result = convertMessagesToMastraMessages(messages);
+
+    expect(result).toHaveLength(4);
+    expect(result[0].role).toBe('user');
+    expect(result[1].role).toBe('assistant');
+    expect(result[2].role).toBe('tool');
+    expect(result[3].role).toBe('assistant');
+  });
+});

package/src/adapters/agui.ts

@@ -0,0 +1,219 @@
+// Cross-platform UUID generation function
+import { AbstractAgent, EventType } from '@ag-ui/client';
+import type {
+  BaseEvent,
+  RunAgentInput,
+  AgentConfig,
+  RunStartedEvent,
+  RunFinishedEvent,
+  TextMessageStartEvent,
+  TextMessageContentEvent,
+  TextMessageEndEvent,
+  Message,
+  ToolCallStartEvent,
+  ToolCallArgsEvent,
+  ToolCallEndEvent,
+} from '@ag-ui/client';
+import type { CoreMessage } from '@mastra/core';
+import { Observable } from 'rxjs';
+import type { Agent } from '../resources/agent';
+
+interface MastraAgentConfig extends AgentConfig {
+  agent: Agent;
+  agentId: string;
+  resourceId?: string;
+}
+
+export class AGUIAdapter extends AbstractAgent {
+  agent: Agent;
+  resourceId?: string;
+  constructor({ agent, agentId, resourceId, ...rest }: MastraAgentConfig) {
+    super({
+      agentId,
+      ...rest,
+    });
+    this.agent = agent;
+    this.resourceId = resourceId;
+  }
+
+  protected run(input: RunAgentInput): Observable<BaseEvent> {
+    return new Observable<BaseEvent>(subscriber => {
+      const convertedMessages = convertMessagesToMastraMessages(input.messages);
+
+      subscriber.next({
+        type: EventType.RUN_STARTED,
+        threadId: input.threadId,
+        runId: input.runId,
+      } as RunStartedEvent);
+
+      this.agent
+        .stream({
+          threadId: input.threadId,
+          resourceId: this.resourceId ?? '',
+          runId: input.runId,
+          messages: convertedMessages,
+          clientTools: input.tools.reduce(
+            (acc, tool) => {
+              acc[tool.name as string] = {
+                id: tool.name,
+                description: tool.description,
+                inputSchema: tool.parameters,
+              };
+              return acc;
+            },
+            {} as Record<string, any>,
+          ),
+        })
+        .then(response => {
+          let currentMessageId: string | undefined = undefined;
+          return response.processDataStream({
+            onTextPart: text => {
+              if (currentMessageId === undefined) {
+                currentMessageId = generateUUID();
+
+                const message: TextMessageStartEvent = {
+                  type: EventType.TEXT_MESSAGE_START,
+                  messageId: currentMessageId,
+                  role: 'assistant',
+                };
+                subscriber.next(message);
+              }
+
+              const message: TextMessageContentEvent = {
+                type: EventType.TEXT_MESSAGE_CONTENT,
+                messageId: currentMessageId,
+                delta: text,
+              };
+              subscriber.next(message);
+            },
+            onFinishMessagePart: message => {
+              console.log('onFinishMessagePart', message);
+              if (currentMessageId !== undefined) {
+                const message: TextMessageEndEvent = {
+                  type: EventType.TEXT_MESSAGE_END,
+                  messageId: currentMessageId,
+                };
+                subscriber.next(message);
+              }
+              // Emit run finished event
+              subscriber.next({
+                type: EventType.RUN_FINISHED,
+                threadId: input.threadId,
+                runId: input.runId,
+              } as RunFinishedEvent);
+
+              // Complete the observable
+              subscriber.complete();
+            },
+            onToolCallPart(streamPart) {
+              const parentMessageId = currentMessageId || generateUUID();
+              subscriber.next({
+                type: EventType.TOOL_CALL_START,
+                toolCallId: streamPart.toolCallId,
+                toolCallName: streamPart.toolName,
+                parentMessageId,
+              } as ToolCallStartEvent);
+
+              subscriber.next({
+                type: EventType.TOOL_CALL_ARGS,
+                toolCallId: streamPart.toolCallId,
+                delta: JSON.stringify(streamPart.args),
+                parentMessageId,
+              } as ToolCallArgsEvent);
+
+              subscriber.next({
+                type: EventType.TOOL_CALL_END,
+                toolCallId: streamPart.toolCallId,
+                parentMessageId,
+              } as ToolCallEndEvent);
+            },
+          });
+        })
+        .catch(error => {
+          console.log('error', error);
+          // Handle error
+          subscriber.error(error);
+        });
+
+      return () => {};
+    });
+  }
+}
+
+/**
+ * Generates a UUID v4 that works in both browser and Node.js environments
+ */
+export function generateUUID(): string {
+  // Use crypto.randomUUID() if available (Node.js environment or modern browsers)
+  if (typeof crypto !== 'undefined') {
+    // Browser crypto API or Node.js crypto global
+    if (typeof crypto.randomUUID === 'function') {
+      return crypto.randomUUID();
+    }
+    // Fallback for older browsers
+    if (typeof crypto.getRandomValues === 'function') {
+      const buffer = new Uint8Array(16);
+      crypto.getRandomValues(buffer);
+      // Set version (4) and variant (8, 9, A, or B)
+      buffer[6] = (buffer[6]! & 0x0f) | 0x40; // version 4
+      buffer[8] = (buffer[8]! & 0x3f) | 0x80; // variant
+
+      // Convert to hex string in UUID format
+      let hex = '';
+      for (let i = 0; i < 16; i++) {
+        hex += buffer[i]!.toString(16).padStart(2, '0');
+        // Add hyphens at standard positions
+        if (i === 3 || i === 5 || i === 7 || i === 9) hex += '-';
+      }
+      return hex;
+    }
+  }
+
+  // Last resort fallback (less secure but works everywhere)
+  return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, c => {
+    const r = (Math.random() * 16) | 0;
+    const v = c === 'x' ? r : (r & 0x3) | 0x8;
+    return v.toString(16);
+  });
+}
+
+export function convertMessagesToMastraMessages(messages: Message[]): CoreMessage[] {
+  const result: CoreMessage[] = [];
+
+  for (const message of messages) {
+    if (message.role === 'assistant') {
+      const parts: any[] = message.content ? [{ type: 'text', text: message.content }] : [];
+      for (const toolCall of message.toolCalls ?? []) {
+        parts.push({
+          type: 'tool-call',
+          toolCallId: toolCall.id,
+          toolName: toolCall.function.name,
+          args: JSON.parse(toolCall.function.arguments),
+        });
+      }
+      result.push({
+        role: 'assistant',
+        content: parts,
+      });
+    } else if (message.role === 'user') {
+      result.push({
+        role: 'user',
+        content: message.content || '',
+      });
+    } else if (message.role === 'tool') {
+      result.push({
+        role: 'tool',
+        content: [
+          {
+            type: 'tool-result',
+            toolCallId: message.toolCallId,
+            toolName: 'unknown',
+            result: message.content,
+          },
+        ],
+      });
+    }
+  }
+
+  return result;
+}

package/src/client.ts

@@ -1,3 +1,5 @@
+import type { AbstractAgent } from '@ag-ui/client';
+import { AGUIAdapter } from './adapters/agui';
 import { Agent, MemoryThread, Tool, Workflow, Vector, BaseResource, Network, VNextWorkflow } from './resources';
 import type {
   ClientOptions,
@@ -32,6 +34,25 @@ export class MastraClient extends BaseResource {
     return this.request('/api/agents');
   }
 
+  public async getAGUI({ resourceId }: { resourceId: string }): Promise<Record<string, AbstractAgent>> {
+    const agents = await this.getAgents();
+
+    return Object.entries(agents).reduce(
+      (acc, [agentId]) => {
+        const agent = this.getAgent(agentId);
+
+        acc[agentId] = new AGUIAdapter({
+          agentId,
+          agent,
+          resourceId,
+        });
+
+        return acc;
+      },
+      {} as Record<string, AbstractAgent>,
+    );
+  }
+
   /**
    * Gets an agent instance by ID
    * @param agentId - ID of the agent to retrieve