@mastra/client-js 0.0.0-pg-pool-options-20250428183821 → 0.0.0-redis-cloud-transporter-20250508194049

package/dist/index.d.ts

@@ -1,7 +1,8 @@
-import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, WorkflowRunResult as WorkflowRunResult$1, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
+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';
 import { ZodSchema } from 'zod';
-import { processDataStream } from '@ai-sdk/ui-utils';
 import { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
 import { NewWorkflow, WorkflowResult, WatchEvent } from '@mastra/core/workflows/vNext';
 import { RuntimeContext } from '@mastra/core/runtime-context';
@@ -40,6 +41,9 @@ type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
 } & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
 interface GetEvalsByAgentIdResponse extends GetAgentResponse {
     evals: any[];
+    instructions: string;
+    name: string;
+    id: string;
 }
 interface GetToolResponse {
     id: string;
@@ -55,6 +59,14 @@ interface GetWorkflowResponse {
     stepSubscriberGraph: Record<string, StepGraph>;
     workflowId?: string;
 }
+interface GetWorkflowRunsParams {
+    fromDate?: Date;
+    toDate?: Date;
+    limit?: number;
+    offset?: number;
+    resourceId?: string;
+}
+type GetWorkflowRunsResponse = WorkflowRuns;
 type WorkflowRunResult = {
     activePaths: Record<string, {
         status: string;
@@ -67,8 +79,17 @@ type WorkflowRunResult = {
 };
 interface GetVNextWorkflowResponse {
     name: string;
-    steps: NewWorkflow['steps'];
-    stepGraph: NewWorkflow['stepGraph'];
+    steps: {
+        [key: string]: {
+            id: string;
+            description: string;
+            inputSchema: string;
+            outputSchema: string;
+            resumeSchema: string;
+            suspendSchema: string;
+        };
+    };
+    stepGraph: NewWorkflow['serializedStepGraph'];
     inputSchema: string;
     outputSchema: string;
 }
@@ -184,6 +205,8 @@ interface GetTelemetryParams {
     page?: number;
     perPage?: number;
     attribute?: Record<string, string>;
+    fromDate?: Date;
+    toDate?: Date;
 }
 interface GetNetworkResponse {
     name: string;
@@ -389,6 +412,12 @@ declare class Workflow extends BaseResource {
      * @returns Promise containing workflow details including steps and graphs
      */
     details(): Promise<GetWorkflowResponse>;
+    /**
+     * Retrieves all runs for a workflow
+     * @param params - Parameters for filtering runs
+     * @returns Promise containing workflow runs array
+     */
+    runs(params?: GetWorkflowRunsParams): Promise<GetWorkflowRunsResponse>;
     /**
      * @deprecated Use `startAsync` instead
      * Executes the workflow with the provided parameters
@@ -482,6 +511,7 @@ declare class Tool extends BaseResource {
      */
     execute(params: {
         data: any;
+        runId?: string;
     }): Promise<any>;
 }
 
@@ -501,6 +531,12 @@ declare class VNextWorkflow extends BaseResource {
      * @returns Promise containing vNext workflow details including steps and graphs
      */
     details(): Promise<GetVNextWorkflowResponse>;
+    /**
+     * Retrieves all runs for a vNext workflow
+     * @param params - Parameters for filtering runs
+     * @returns Promise containing vNext workflow runs array
+     */
+    runs(params?: GetWorkflowRunsParams): Promise<GetWorkflowRunsResponse>;
     /**
      * Creates a new vNext workflow run
      * @param params - Optional object containing the optional runId
@@ -574,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
@@ -688,4 +727,4 @@ declare class MastraClient extends BaseResource {
     getNetwork(networkId: string): Network;
 }
 
-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 GetNetworkResponse, type GetTelemetryParams, type GetTelemetryResponse, type GetToolResponse, type GetVNextWorkflowResponse, type GetVectorIndexResponse, type GetWorkflowResponse, MastraClient, type QueryVectorParams, type QueryVectorResponse, type RequestFunction, type RequestOptions, type SaveMessageToMemoryParams, type SaveMessageToMemoryResponse, type StreamParams, type UpdateMemoryThreadParams, type UpsertVectorParams, type VNextWorkflowRunResult, type VNextWorkflowWatchResult, type WorkflowRunResult };
+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 GetNetworkResponse, type GetTelemetryParams, type GetTelemetryResponse, type GetToolResponse, type GetVNextWorkflowResponse, type GetVectorIndexResponse, type GetWorkflowResponse, type GetWorkflowRunsParams, type GetWorkflowRunsResponse, MastraClient, type QueryVectorParams, type QueryVectorResponse, type RequestFunction, type RequestOptions, type SaveMessageToMemoryParams, type SaveMessageToMemoryResponse, type StreamParams, type UpdateMemoryThreadParams, type UpsertVectorParams, type VNextWorkflowRunResult, type VNextWorkflowWatchResult, type WorkflowRunResult };

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';
-import { processDataStream } from '@ai-sdk/ui-utils';
 
-// 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 {
@@ -134,7 +299,8 @@ var Agent = class extends BaseResource {
     const processedParams = {
       ...params,
       output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
-      experimental_output: params.experimental_output instanceof ZodSchema ? zodToJsonSchema(params.experimental_output) : params.experimental_output
+      experimental_output: params.experimental_output instanceof ZodSchema ? zodToJsonSchema(params.experimental_output) : params.experimental_output,
+      runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0
     };
     return this.request(`/api/agents/${this.agentId}/generate`, {
       method: "POST",
@@ -150,7 +316,8 @@ var Agent = class extends BaseResource {
     const processedParams = {
       ...params,
       output: params.output instanceof ZodSchema ? zodToJsonSchema(params.output) : params.output,
-      experimental_output: params.experimental_output instanceof ZodSchema ? zodToJsonSchema(params.experimental_output) : params.experimental_output
+      experimental_output: params.experimental_output instanceof ZodSchema ? zodToJsonSchema(params.experimental_output) : params.experimental_output,
+      runtimeContext: params.runtimeContext ? Object.fromEntries(params.runtimeContext.entries()) : void 0
     };
     const response = await this.request(`/api/agents/${this.agentId}/stream`, {
       method: "POST",
@@ -371,6 +538,34 @@ var Workflow = class extends BaseResource {
   details() {
     return this.request(`/api/workflows/${this.workflowId}`);
   }
+  /**
+   * Retrieves all runs for a workflow
+   * @param params - Parameters for filtering runs
+   * @returns Promise containing workflow runs array
+   */
+  runs(params) {
+    const searchParams = new URLSearchParams();
+    if (params?.fromDate) {
+      searchParams.set("fromDate", params.fromDate.toISOString());
+    }
+    if (params?.toDate) {
+      searchParams.set("toDate", params.toDate.toISOString());
+    }
+    if (params?.limit) {
+      searchParams.set("limit", String(params.limit));
+    }
+    if (params?.offset) {
+      searchParams.set("offset", String(params.offset));
+    }
+    if (params?.resourceId) {
+      searchParams.set("resourceId", params.resourceId);
+    }
+    if (searchParams.size) {
+      return this.request(`/api/workflows/${this.workflowId}/runs?${searchParams}`);
+    } else {
+      return this.request(`/api/workflows/${this.workflowId}/runs`);
+    }
+  }
   /**
    * @deprecated Use `startAsync` instead
    * Executes the workflow with the provided parameters
@@ -487,7 +682,7 @@ var Workflow = class extends BaseResource {
               }
             }
           }
-        } catch (error) {
+        } catch {
         }
       }
       if (buffer) {
@@ -541,9 +736,13 @@ var Tool = class extends BaseResource {
    * @returns Promise containing the tool execution results
    */
   execute(params) {
-    return this.request(`/api/tools/${this.toolId}/execute`, {
+    const url = new URLSearchParams();
+    if (params.runId) {
+      url.set("runId", params.runId);
+    }
+    return this.request(`/api/tools/${this.toolId}/execute?${url.toString()}`, {
       method: "POST",
-      body: params
+      body: params.data
     });
   }
 };
@@ -586,7 +785,7 @@ var VNextWorkflow = class extends BaseResource {
               }
             }
           }
-        } catch (error) {
+        } catch {
         }
       }
       if (buffer) {
@@ -607,6 +806,34 @@ var VNextWorkflow = class extends BaseResource {
   details() {
     return this.request(`/api/workflows/v-next/${this.workflowId}`);
   }
+  /**
+   * Retrieves all runs for a vNext workflow
+   * @param params - Parameters for filtering runs
+   * @returns Promise containing vNext workflow runs array
+   */
+  runs(params) {
+    const searchParams = new URLSearchParams();
+    if (params?.fromDate) {
+      searchParams.set("fromDate", params.fromDate.toISOString());
+    }
+    if (params?.toDate) {
+      searchParams.set("toDate", params.toDate.toISOString());
+    }
+    if (params?.limit) {
+      searchParams.set("limit", String(params.limit));
+    }
+    if (params?.offset) {
+      searchParams.set("offset", String(params.offset));
+    }
+    if (params?.resourceId) {
+      searchParams.set("resourceId", params.resourceId);
+    }
+    if (searchParams.size) {
+      return this.request(`/api/workflows/v-next/${this.workflowId}/runs?${searchParams}`);
+    } else {
+      return this.request(`/api/workflows/v-next/${this.workflowId}/runs`);
+    }
+  }
   /**
    * Creates a new vNext workflow run
    * @param params - Optional object containing the optional runId
@@ -716,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
@@ -848,7 +1090,7 @@ var MastraClient = class extends BaseResource {
    * @returns Promise containing telemetry data
    */
   getTelemetry(params) {
-    const { name, scope, page, perPage, attribute } = params || {};
+    const { name, scope, page, perPage, attribute, fromDate, toDate } = params || {};
     const _attribute = attribute ? Object.entries(attribute).map(([key, value]) => `${key}:${value}`) : [];
     const searchParams = new URLSearchParams();
     if (name) {
@@ -872,6 +1114,12 @@ var MastraClient = class extends BaseResource {
         searchParams.set("attribute", _attribute);
       }
     }
+    if (fromDate) {
+      searchParams.set("fromDate", fromDate.toISOString());
+    }
+    if (toDate) {
+      searchParams.set("toDate", toDate.toISOString());
+    }
     if (searchParams.size) {
       return this.request(`/api/telemetry?${searchParams}`);
     } else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/client-js",
-  "version": "0.0.0-pg-pool-options-20250428183821",
+  "version": "0.0.0-redis-cloud-transporter-20250508194049",
   "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.0.0-pg-pool-options-20250428183821"
+    "rxjs": "7.8.1",
+    "zod": "^3.24.3",
+    "zod-to-json-schema": "^3.24.5",
+    "@mastra/core": "0.0.0-redis-cloud-transporter-20250508194049"
   },
   "peerDependencies": {
     "zod": "^3.24.2"
@@ -39,8 +41,8 @@
     "@types/node": "^20.17.27",
     "tsup": "^8.4.0",
     "typescript": "^5.8.2",
-    "vitest": "^3.0.9",
-    "@internal/lint": "0.0.2"
+    "vitest": "^3.1.2",
+    "@internal/lint": "0.0.0-redis-cloud-transporter-20250508194049"
   },
   "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');
+  });
+});