@mastra/client-js 0.0.0-afterToolExecute-20250414225911 → 0.0.0-agui-20250501182100

package/dist/index.d.cts

@@ -1,8 +1,12 @@
+import { AbstractAgent, AgentConfig, RunAgentInput, BaseEvent } from '@ag-ui/client';
+import { Observable } from 'rxjs';
 import { processDataStream } from '@ai-sdk/ui-utils';
-import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, WorkflowRunResult as WorkflowRunResult$1, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
+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 { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
+import { NewWorkflow, WorkflowResult, WatchEvent } from '@mastra/core/workflows/vNext';
+import { RuntimeContext } from '@mastra/core/runtime-context';
 
 interface ClientOptions {
     /** Base URL for API requests */
@@ -32,12 +36,15 @@ interface GetAgentResponse {
 }
 type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
     messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Partial<Omit<AgentGenerateOptions<T>, 'onAfterToolExecute'>>;
+} & Partial<AgentGenerateOptions<T>>;
 type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
     messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry' | 'onAfterToolExecute'>;
+} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
 interface GetEvalsByAgentIdResponse extends GetAgentResponse {
     evals: any[];
+    instructions: string;
+    name: string;
+    id: string;
 }
 interface GetToolResponse {
     id: string;
@@ -53,6 +60,7 @@ interface GetWorkflowResponse {
     stepSubscriberGraph: Record<string, StepGraph>;
     workflowId?: string;
 }
+type GetWorkflowRunsResponse = WorkflowRuns;
 type WorkflowRunResult = {
     activePaths: Record<string, {
         status: string;
@@ -63,6 +71,26 @@ type WorkflowRunResult = {
     timestamp: number;
     runId: string;
 };
+interface GetVNextWorkflowResponse {
+    name: string;
+    steps: {
+        [key: string]: {
+            id: string;
+            description: string;
+            inputSchema: string;
+            outputSchema: string;
+            resumeSchema: string;
+            suspendSchema: string;
+        };
+    };
+    stepGraph: NewWorkflow['serializedStepGraph'];
+    inputSchema: string;
+    outputSchema: string;
+}
+type VNextWorkflowWatchResult = WatchEvent & {
+    runId: string;
+};
+type VNextWorkflowRunResult = WorkflowResult<any, any>;
 interface UpsertVectorParams {
     indexName: string;
     vectors: number[][];
@@ -97,7 +125,7 @@ type SaveMessageToMemoryResponse = MessageType[];
 interface CreateMemoryThreadParams {
     title: string;
     metadata: Record<string, any>;
-    resourceid: string;
+    resourceId: string;
     threadId: string;
     agentId: string;
 }
@@ -110,7 +138,7 @@ type GetMemoryThreadResponse = StorageThreadType[];
 interface UpdateMemoryThreadParams {
     title: string;
     metadata: Record<string, any>;
-    resourceid: string;
+    resourceId: string;
 }
 interface GetMemoryThreadMessagesResponse {
     messages: CoreMessage[];
@@ -269,6 +297,18 @@ declare class Agent extends BaseResource {
     liveEvals(): Promise<GetEvalsByAgentIdResponse>;
 }
 
+interface MastraAgentConfig extends AgentConfig {
+    agent: Agent;
+    agentId: string;
+    resourceId?: string;
+}
+declare class AGUIAdapter extends AbstractAgent {
+    agent: Agent;
+    resourceId?: string;
+    constructor({ agent, agentId, resourceId, ...rest }: MastraAgentConfig);
+    protected run(input: RunAgentInput): Observable<BaseEvent>;
+}
+
 declare class Network extends BaseResource {
     private networkId;
     constructor(options: ClientOptions, networkId: string);
@@ -376,6 +416,11 @@ declare class Workflow extends BaseResource {
      * @returns Promise containing workflow details including steps and graphs
      */
     details(): Promise<GetWorkflowResponse>;
+    /**
+     * Retrieves all runs for a workflow
+     * @returns Promise containing workflow runs array
+     */
+    runs(): Promise<GetWorkflowRunsResponse>;
     /**
      * @deprecated Use `startAsync` instead
      * Executes the workflow with the provided parameters
@@ -469,9 +514,97 @@ declare class Tool extends BaseResource {
      */
     execute(params: {
         data: any;
+        runId?: string;
     }): Promise<any>;
 }
 
+declare class VNextWorkflow extends BaseResource {
+    private workflowId;
+    constructor(options: ClientOptions, workflowId: string);
+    /**
+     * Creates an async generator that processes a readable stream and yields vNext workflow records
+     * separated by the Record Separator character (\x1E)
+     *
+     * @param stream - The readable stream to process
+     * @returns An async generator that yields parsed records
+     */
+    private streamProcessor;
+    /**
+     * Retrieves details about the vNext workflow
+     * @returns Promise containing vNext workflow details including steps and graphs
+     */
+    details(): Promise<GetVNextWorkflowResponse>;
+    /**
+     * Retrieves all runs for a vNext workflow
+     * @returns Promise containing vNext workflow runs array
+     */
+    runs(): Promise<GetWorkflowRunsResponse>;
+    /**
+     * Creates a new vNext workflow run
+     * @param params - Optional object containing the optional runId
+     * @returns Promise containing the runId of the created run
+     */
+    createRun(params?: {
+        runId?: string;
+    }): Promise<{
+        runId: string;
+    }>;
+    /**
+     * Starts a vNext workflow run synchronously without waiting for the workflow to complete
+     * @param params - Object containing the runId, inputData and runtimeContext
+     * @returns Promise containing success message
+     */
+    start(params: {
+        runId: string;
+        inputData: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<{
+        message: string;
+    }>;
+    /**
+     * Resumes a suspended vNext workflow step synchronously without waiting for the vNext workflow to complete
+     * @param params - Object containing the runId, step, resumeData and runtimeContext
+     * @returns Promise containing success message
+     */
+    resume({ step, runId, resumeData, runtimeContext, }: {
+        step: string | string[];
+        runId: string;
+        resumeData?: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<{
+        message: string;
+    }>;
+    /**
+     * Starts a vNext workflow run asynchronously and returns a promise that resolves when the vNext workflow is complete
+     * @param params - Object containing the optional runId, inputData and runtimeContext
+     * @returns Promise containing the vNext workflow execution results
+     */
+    startAsync(params: {
+        runId?: string;
+        inputData: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<VNextWorkflowRunResult>;
+    /**
+     * Resumes a suspended vNext workflow step asynchronously and returns a promise that resolves when the vNext workflow is complete
+     * @param params - Object containing the runId, step, resumeData and runtimeContext
+     * @returns Promise containing the vNext workflow resume results
+     */
+    resumeAsync(params: {
+        runId: string;
+        step: string | string[];
+        resumeData?: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<VNextWorkflowRunResult>;
+    /**
+     * Watches vNext workflow transitions in real-time
+     * @param runId - Optional run ID to filter the watch stream
+     * @returns AsyncGenerator that yields parsed records from the vNext workflow watch stream
+     */
+    watch({ runId }: {
+        runId?: string;
+    }, onRecord: (record: VNextWorkflowWatchResult) => void): Promise<void>;
+}
+
 declare class MastraClient extends BaseResource {
     constructor(options: ClientOptions);
     /**
@@ -479,6 +612,7 @@ declare class MastraClient extends BaseResource {
      * @returns Promise containing map of agent IDs to agent details
      */
     getAgents(): Promise<Record<string, GetAgentResponse>>;
+    getAGUI(): Promise<Record<string, AGUIAdapter>>;
     /**
      * Gets an agent instance by ID
      * @param agentId - ID of the agent to retrieve
@@ -538,6 +672,17 @@ declare class MastraClient extends BaseResource {
      * @returns Workflow instance
      */
     getWorkflow(workflowId: string): Workflow;
+    /**
+     * Retrieves all available vNext workflows
+     * @returns Promise containing map of vNext workflow IDs to vNext workflow details
+     */
+    getVNextWorkflows(): Promise<Record<string, GetVNextWorkflowResponse>>;
+    /**
+     * Gets a vNext workflow instance by ID
+     * @param workflowId - ID of the vNext workflow to retrieve
+     * @returns vNext Workflow instance
+     */
+    getVNextWorkflow(workflowId: string): VNextWorkflow;
     /**
      * Gets a vector instance by name
      * @param vectorName - Name of the vector to retrieve
@@ -582,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 GetVectorIndexResponse, type GetWorkflowResponse, MastraClient, type QueryVectorParams, type QueryVectorResponse, type RequestFunction, type RequestOptions, type SaveMessageToMemoryParams, type SaveMessageToMemoryResponse, type StreamParams, type UpdateMemoryThreadParams, type UpsertVectorParams, 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 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.d.ts

@@ -1,8 +1,12 @@
+import { AbstractAgent, AgentConfig, RunAgentInput, BaseEvent } from '@ag-ui/client';
+import { Observable } from 'rxjs';
 import { processDataStream } from '@ai-sdk/ui-utils';
-import { CoreMessage, AiMessageType, StorageThreadType, MessageType, StepAction, StepGraph, WorkflowRunResult as WorkflowRunResult$1, QueryResult, BaseLogMessage, GenerateReturn } from '@mastra/core';
+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 { AgentGenerateOptions, AgentStreamOptions } from '@mastra/core/agent';
+import { NewWorkflow, WorkflowResult, WatchEvent } from '@mastra/core/workflows/vNext';
+import { RuntimeContext } from '@mastra/core/runtime-context';
 
 interface ClientOptions {
     /** Base URL for API requests */
@@ -32,12 +36,15 @@ interface GetAgentResponse {
 }
 type GenerateParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
     messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Partial<Omit<AgentGenerateOptions<T>, 'onAfterToolExecute'>>;
+} & Partial<AgentGenerateOptions<T>>;
 type StreamParams<T extends JSONSchema7 | ZodSchema | undefined = undefined> = {
     messages: string | string[] | CoreMessage[] | AiMessageType[];
-} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry' | 'onAfterToolExecute'>;
+} & Omit<AgentStreamOptions<T>, 'onFinish' | 'onStepFinish' | 'telemetry'>;
 interface GetEvalsByAgentIdResponse extends GetAgentResponse {
     evals: any[];
+    instructions: string;
+    name: string;
+    id: string;
 }
 interface GetToolResponse {
     id: string;
@@ -53,6 +60,7 @@ interface GetWorkflowResponse {
     stepSubscriberGraph: Record<string, StepGraph>;
     workflowId?: string;
 }
+type GetWorkflowRunsResponse = WorkflowRuns;
 type WorkflowRunResult = {
     activePaths: Record<string, {
         status: string;
@@ -63,6 +71,26 @@ type WorkflowRunResult = {
     timestamp: number;
     runId: string;
 };
+interface GetVNextWorkflowResponse {
+    name: string;
+    steps: {
+        [key: string]: {
+            id: string;
+            description: string;
+            inputSchema: string;
+            outputSchema: string;
+            resumeSchema: string;
+            suspendSchema: string;
+        };
+    };
+    stepGraph: NewWorkflow['serializedStepGraph'];
+    inputSchema: string;
+    outputSchema: string;
+}
+type VNextWorkflowWatchResult = WatchEvent & {
+    runId: string;
+};
+type VNextWorkflowRunResult = WorkflowResult<any, any>;
 interface UpsertVectorParams {
     indexName: string;
     vectors: number[][];
@@ -97,7 +125,7 @@ type SaveMessageToMemoryResponse = MessageType[];
 interface CreateMemoryThreadParams {
     title: string;
     metadata: Record<string, any>;
-    resourceid: string;
+    resourceId: string;
     threadId: string;
     agentId: string;
 }
@@ -110,7 +138,7 @@ type GetMemoryThreadResponse = StorageThreadType[];
 interface UpdateMemoryThreadParams {
     title: string;
     metadata: Record<string, any>;
-    resourceid: string;
+    resourceId: string;
 }
 interface GetMemoryThreadMessagesResponse {
     messages: CoreMessage[];
@@ -269,6 +297,18 @@ declare class Agent extends BaseResource {
     liveEvals(): Promise<GetEvalsByAgentIdResponse>;
 }
 
+interface MastraAgentConfig extends AgentConfig {
+    agent: Agent;
+    agentId: string;
+    resourceId?: string;
+}
+declare class AGUIAdapter extends AbstractAgent {
+    agent: Agent;
+    resourceId?: string;
+    constructor({ agent, agentId, resourceId, ...rest }: MastraAgentConfig);
+    protected run(input: RunAgentInput): Observable<BaseEvent>;
+}
+
 declare class Network extends BaseResource {
     private networkId;
     constructor(options: ClientOptions, networkId: string);
@@ -376,6 +416,11 @@ declare class Workflow extends BaseResource {
      * @returns Promise containing workflow details including steps and graphs
      */
     details(): Promise<GetWorkflowResponse>;
+    /**
+     * Retrieves all runs for a workflow
+     * @returns Promise containing workflow runs array
+     */
+    runs(): Promise<GetWorkflowRunsResponse>;
     /**
      * @deprecated Use `startAsync` instead
      * Executes the workflow with the provided parameters
@@ -469,9 +514,97 @@ declare class Tool extends BaseResource {
      */
     execute(params: {
         data: any;
+        runId?: string;
     }): Promise<any>;
 }
 
+declare class VNextWorkflow extends BaseResource {
+    private workflowId;
+    constructor(options: ClientOptions, workflowId: string);
+    /**
+     * Creates an async generator that processes a readable stream and yields vNext workflow records
+     * separated by the Record Separator character (\x1E)
+     *
+     * @param stream - The readable stream to process
+     * @returns An async generator that yields parsed records
+     */
+    private streamProcessor;
+    /**
+     * Retrieves details about the vNext workflow
+     * @returns Promise containing vNext workflow details including steps and graphs
+     */
+    details(): Promise<GetVNextWorkflowResponse>;
+    /**
+     * Retrieves all runs for a vNext workflow
+     * @returns Promise containing vNext workflow runs array
+     */
+    runs(): Promise<GetWorkflowRunsResponse>;
+    /**
+     * Creates a new vNext workflow run
+     * @param params - Optional object containing the optional runId
+     * @returns Promise containing the runId of the created run
+     */
+    createRun(params?: {
+        runId?: string;
+    }): Promise<{
+        runId: string;
+    }>;
+    /**
+     * Starts a vNext workflow run synchronously without waiting for the workflow to complete
+     * @param params - Object containing the runId, inputData and runtimeContext
+     * @returns Promise containing success message
+     */
+    start(params: {
+        runId: string;
+        inputData: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<{
+        message: string;
+    }>;
+    /**
+     * Resumes a suspended vNext workflow step synchronously without waiting for the vNext workflow to complete
+     * @param params - Object containing the runId, step, resumeData and runtimeContext
+     * @returns Promise containing success message
+     */
+    resume({ step, runId, resumeData, runtimeContext, }: {
+        step: string | string[];
+        runId: string;
+        resumeData?: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<{
+        message: string;
+    }>;
+    /**
+     * Starts a vNext workflow run asynchronously and returns a promise that resolves when the vNext workflow is complete
+     * @param params - Object containing the optional runId, inputData and runtimeContext
+     * @returns Promise containing the vNext workflow execution results
+     */
+    startAsync(params: {
+        runId?: string;
+        inputData: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<VNextWorkflowRunResult>;
+    /**
+     * Resumes a suspended vNext workflow step asynchronously and returns a promise that resolves when the vNext workflow is complete
+     * @param params - Object containing the runId, step, resumeData and runtimeContext
+     * @returns Promise containing the vNext workflow resume results
+     */
+    resumeAsync(params: {
+        runId: string;
+        step: string | string[];
+        resumeData?: Record<string, any>;
+        runtimeContext?: RuntimeContext;
+    }): Promise<VNextWorkflowRunResult>;
+    /**
+     * Watches vNext workflow transitions in real-time
+     * @param runId - Optional run ID to filter the watch stream
+     * @returns AsyncGenerator that yields parsed records from the vNext workflow watch stream
+     */
+    watch({ runId }: {
+        runId?: string;
+    }, onRecord: (record: VNextWorkflowWatchResult) => void): Promise<void>;
+}
+
 declare class MastraClient extends BaseResource {
     constructor(options: ClientOptions);
     /**
@@ -479,6 +612,7 @@ declare class MastraClient extends BaseResource {
      * @returns Promise containing map of agent IDs to agent details
      */
     getAgents(): Promise<Record<string, GetAgentResponse>>;
+    getAGUI(): Promise<Record<string, AGUIAdapter>>;
     /**
      * Gets an agent instance by ID
      * @param agentId - ID of the agent to retrieve
@@ -538,6 +672,17 @@ declare class MastraClient extends BaseResource {
      * @returns Workflow instance
      */
     getWorkflow(workflowId: string): Workflow;
+    /**
+     * Retrieves all available vNext workflows
+     * @returns Promise containing map of vNext workflow IDs to vNext workflow details
+     */
+    getVNextWorkflows(): Promise<Record<string, GetVNextWorkflowResponse>>;
+    /**
+     * Gets a vNext workflow instance by ID
+     * @param workflowId - ID of the vNext workflow to retrieve
+     * @returns vNext Workflow instance
+     */
+    getVNextWorkflow(workflowId: string): VNextWorkflow;
     /**
      * Gets a vector instance by name
      * @param vectorName - Name of the vector to retrieve
@@ -582,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 GetVectorIndexResponse, type GetWorkflowResponse, MastraClient, type QueryVectorParams, type QueryVectorResponse, type RequestFunction, type RequestOptions, type SaveMessageToMemoryParams, type SaveMessageToMemoryResponse, type StreamParams, type UpdateMemoryThreadParams, type UpsertVectorParams, 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 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 };