@mastra/core 0.2.0-alpha.99 → 0.2.1-alpha.0

package/dist/{index-Cwb-5AzX.d.ts → base-C7R9FwZ6.d.ts}

@@ -1,14 +1,138 @@
-import { Message, UserContent, AssistantContent, ToolContent, CoreMessage as CoreMessage$1, EmbeddingModel, LanguageModelV1, CoreSystemMessage as CoreSystemMessage$1, CoreAssistantMessage as CoreAssistantMessage$1, CoreUserMessage as CoreUserMessage$1, CoreToolMessage as CoreToolMessage$1, EmbedResult as EmbedResult$1, EmbedManyResult as EmbedManyResult$1, GenerateTextResult, GenerateObjectResult, StreamTextResult, StreamObjectResult, LanguageModel as LanguageModel$1, DeepPartial } from 'ai';
-import { JSONSchema7 } from 'json-schema';
-import { z, ZodSchema } from 'zod';
-import { MastraBase } from './base.js';
-import { a as Metric } from './metric-BWeQNZt6.js';
-import { T as Telemetry } from './telemetry-oCUM52DG.js';
+import * as ai from 'ai';
+import { EmbeddingModel, CoreMessage as CoreMessage$1, Message, UserContent, AssistantContent, LanguageModelV1, GenerateTextResult, GenerateObjectResult, StreamTextResult, StreamObjectResult, CoreSystemMessage as CoreSystemMessage$1, CoreAssistantMessage as CoreAssistantMessage$1, CoreUserMessage as CoreUserMessage$1, CoreToolMessage as CoreToolMessage$1, EmbedResult as EmbedResult$1, EmbedManyResult as EmbedManyResult$1, LanguageModel as LanguageModel$1, DeepPartial, ToolContent } from 'ai';
+import { M as MastraBase, T as Telemetry } from './base-BbtPAA6f.js';
+import { a as Metric, M as MetricResult, T as TestInfo } from './types-m9RryK9a.js';
 import { Query } from 'sift';
-import { B as BaseLogMessage, R as RegisteredLogger, L as Logger, b as Run } from './index-CBZ2mk2H.js';
+import { z, ZodSchema } from 'zod';
+import { JSONSchema7 } from 'json-schema';
 import { MastraVector } from './vector/index.js';
+import { B as BaseLogMessage, R as RegisteredLogger, L as Logger, d as Run } from './index-B2JCcAQt.js';
 import { MastraTTS } from './tts/index.js';
 
+/**
+ * Abstract Memory class that defines the interface for storing and retrieving
+ * conversation threads and messages.
+ */
+declare abstract class MastraMemory extends MastraBase {
+    MAX_CONTEXT_TOKENS?: number;
+    storage: MastraStorage;
+    vector: MastraVector;
+    embedder: EmbeddingModel<string>;
+    protected threadConfig: MemoryConfig;
+    constructor(config: {
+        name: string;
+    } & SharedMemoryConfig);
+    setStorage(storage: MastraStorage): void;
+    setVector(vector: MastraVector): void;
+    setEmbedder(embedder: EmbeddingModel<string>): void;
+    /**
+     * Get a system message to inject into the conversation.
+     * This will be called before each conversation turn.
+     * Implementations can override this to inject custom system messages.
+     */
+    getSystemMessage(_input: {
+        threadId: string;
+        memoryConfig?: MemoryConfig;
+    }): Promise<string | null>;
+    protected createEmbeddingIndex(): Promise<{
+        indexName: string;
+    }>;
+    protected getMergedThreadConfig(config?: MemoryConfig): MemoryConfig;
+    abstract rememberMessages({ threadId, vectorMessageSearch, config, }: {
+        threadId: string;
+        vectorMessageSearch?: string;
+        config?: MemoryConfig;
+    }): Promise<{
+        messages: CoreMessage$1[];
+        uiMessages: Message[];
+    }>;
+    estimateTokens(text: string): number;
+    protected parseMessages(messages: MessageType[]): CoreMessage$1[];
+    protected convertToUIMessages(messages: MessageType[]): Message[];
+    /**
+     * Retrieves a specific thread by its ID
+     * @param threadId - The unique identifier of the thread
+     * @returns Promise resolving to the thread or null if not found
+     */
+    abstract getThreadById({ threadId }: {
+        threadId: string;
+    }): Promise<StorageThreadType | null>;
+    abstract getThreadsByResourceId({ resourceId }: {
+        resourceId: string;
+    }): Promise<StorageThreadType[]>;
+    /**
+     * Saves or updates a thread
+     * @param thread - The thread data to save
+     * @returns Promise resolving to the saved thread
+     */
+    abstract saveThread({ thread, memoryConfig, }: {
+        thread: StorageThreadType;
+        memoryConfig?: MemoryConfig;
+    }): Promise<StorageThreadType>;
+    /**
+     * Saves messages to a thread
+     * @param messages - Array of messages to save
+     * @returns Promise resolving to the saved messages
+     */
+    abstract saveMessages({ messages, memoryConfig, }: {
+        messages: MessageType[];
+        memoryConfig: MemoryConfig | undefined;
+    }): Promise<MessageType[]>;
+    /**
+     * Retrieves all messages for a specific thread
+     * @param threadId - The unique identifier of the thread
+     * @returns Promise resolving to array of messages and uiMessages
+     */
+    abstract query({ threadId, selectBy, }: StorageGetMessagesArg): Promise<{
+        messages: CoreMessage$1[];
+        uiMessages: Message[];
+    }>;
+    /**
+     * Helper method to create a new thread
+     * @param title - Optional title for the thread
+     * @param metadata - Optional metadata for the thread
+     * @returns Promise resolving to the created thread
+     */
+    createThread({ threadId, resourceId, title, metadata, memoryConfig, }: {
+        resourceId: string;
+        threadId?: string;
+        title?: string;
+        metadata?: Record<string, unknown>;
+        memoryConfig?: MemoryConfig;
+    }): Promise<StorageThreadType>;
+    /**
+     * Helper method to delete a thread
+     * @param threadId - the id of the thread to delete
+     */
+    abstract deleteThread(threadId: string): Promise<void>;
+    /**
+     * Helper method to add a single message to a thread
+     * @param threadId - The thread to add the message to
+     * @param content - The message content
+     * @param role - The role of the message sender
+     * @param type - The type of the message
+     * @param toolNames - Optional array of tool names that were called
+     * @param toolCallArgs - Optional array of tool call arguments
+     * @param toolCallIds - Optional array of tool call ids
+     * @returns Promise resolving to the saved message
+     */
+    addMessage({ threadId, config, content, role, type, toolNames, toolCallArgs, toolCallIds, }: {
+        threadId: string;
+        config?: MemoryConfig;
+        content: UserContent | AssistantContent;
+        role: 'user' | 'assistant';
+        type: 'text' | 'tool-call' | 'tool-result';
+        toolNames?: string[];
+        toolCallArgs?: Record<string, unknown>[];
+        toolCallIds?: string[];
+    }): Promise<MessageType>;
+    /**
+     * Generates a unique identifier
+     * @returns A unique string ID
+     */
+    generateId(): string;
+}
+
 interface WorkflowOptions<TTriggerSchema extends z.ZodType<any> = any> {
     name: string;
     triggerSchema?: TTriggerSchema;
@@ -288,402 +412,122 @@ interface WorkflowRunState {
     timestamp: number;
 }
 
-interface StorageColumn {
-    type: 'text' | 'timestamp' | 'uuid' | 'jsonb' | 'integer';
-    primaryKey?: boolean;
-    nullable?: boolean;
-    references?: {
-        table: string;
-        column: string;
-    };
+type MastraPrimitives = {
+    logger?: Logger;
+    telemetry?: Telemetry;
+    storage?: MastraStorage;
+    agents?: Record<string, Agent>;
+    tts?: Record<string, MastraTTS>;
+    vectors?: Record<string, MastraVector>;
+    memory?: MastraMemory;
+};
+interface IExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined, TContext extends WorkflowContext = WorkflowContext> {
+    context: TSchemaIn extends z.ZodSchema ? z.infer<TSchemaIn> & TContext : TContext;
+    runId?: string;
+    mastra?: MastraPrimitives;
+    suspend: () => Promise<void>;
 }
-interface WorkflowRow {
-    workflow_name: string;
-    run_id: string;
-    snapshot: WorkflowRunState;
-    created_at: Date;
-    updated_at: Date;
+interface IAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined, TSchemaOut extends z.ZodSchema | undefined, TContext extends IExecutionContext<TSchemaIn>> {
+    id: TId;
+    description?: string;
+    inputSchema?: TSchemaIn;
+    outputSchema?: TSchemaOut;
+    mastra?: MastraPrimitives;
+    payload?: TSchemaIn extends z.ZodSchema ? Partial<z.infer<TSchemaIn>> : unknown;
+    execute: (context: TContext) => Promise<TSchemaOut extends z.ZodSchema ? z.infer<TSchemaOut> : unknown>;
+    [key: string]: any;
 }
-type StorageGetMessagesArg = {
-    threadId: string;
-    selectBy?: {
-        vectorSearchString?: string;
-        last?: number | false;
-        include?: {
-            id: string;
-            withPreviousMessages?: number;
-            withNextMessages?: number;
-        }[];
+
+type CoreTool = {
+    description?: string;
+    parameters: ZodSchema;
+    execute: (params: any) => Promise<any>;
+};
+interface ToolExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined, TContext extends WorkflowContext = WorkflowContext> extends IExecutionContext<TSchemaIn, TContext> {
+}
+interface ToolAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined = undefined, TSchemaOut extends z.ZodSchema | undefined = undefined, TContext extends ToolExecutionContext<TSchemaIn> = ToolExecutionContext<TSchemaIn>> extends IAction<TId, TSchemaIn, TSchemaOut, TContext> {
+}
+
+type ToolsetsInput = Record<string, Record<string, ToolAction<any, any, any, any>>>;
+type ToolsInput = Record<string, ToolAction<any, any, any, any>>;
+interface AgentConfig<TTools extends Record<string, ToolAction<any, any, any, any>> = Record<string, ToolAction<any, any, any, any>>, TMetrics extends Record<string, Metric> = Record<string, Metric>> {
+    name: string;
+    instructions: string;
+    model: LanguageModelV1;
+    tools?: TTools;
+    mastra?: MastraPrimitives;
+    /** @deprecated This property is deprecated. Use evals instead to add evaluation metrics. */
+    metrics?: TMetrics;
+    evals?: TMetrics;
+    memory?: MastraMemory;
+}
+interface AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> {
+    toolsets?: ToolsetsInput;
+    resourceId?: string;
+    context?: CoreMessage[];
+    threadId?: string;
+    memoryOptions?: MemoryConfig;
+    runId?: string;
+    onStepFinish?: (step: string) => void;
+    maxSteps?: number;
+    output?: OutputType | Z;
+    temperature?: number;
+    toolChoice?: 'auto' | 'required';
+}
+interface AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> {
+    toolsets?: ToolsetsInput;
+    resourceId?: string;
+    context?: CoreMessage[];
+    threadId?: string;
+    memoryOptions?: MemoryConfig;
+    runId?: string;
+    onFinish?: (result: string) => unknown;
+    onStepFinish?: (step: string) => unknown;
+    maxSteps?: number;
+    output?: OutputType | Z;
+    temperature?: number;
+    toolChoice?: 'auto' | 'required';
+}
+
+type LanguageModel = LanguageModelV1;
+type CoreMessage = CoreMessage$1;
+type CoreSystemMessage = CoreSystemMessage$1;
+type CoreAssistantMessage = CoreAssistantMessage$1;
+type CoreUserMessage = CoreUserMessage$1;
+type CoreToolMessage = CoreToolMessage$1;
+type EmbedResult<T> = EmbedResult$1<T>;
+type EmbedManyResult<T> = EmbedManyResult$1<T>;
+type BaseStructuredOutputType = 'string' | 'number' | 'boolean' | 'date';
+type StructuredOutputType = 'array' | 'string' | 'number' | 'object' | 'boolean' | 'date';
+type StructuredOutputArrayItem = {
+    type: BaseStructuredOutputType;
+} | {
+    type: 'object';
+    items: StructuredOutput;
+};
+type StructuredOutput = {
+    [key: string]: {
+        type: BaseStructuredOutputType;
+    } | {
+        type: 'object';
+        items: StructuredOutput;
+    } | {
+        type: 'array';
+        items: StructuredOutputArrayItem;
     };
-    threadConfig?: MemoryConfig;
 };
-type EvalRow = {
-    result: string;
-    meta: string;
-    input: string;
-    output: string;
-    createdAt: string;
-};
-
-type TABLE_NAMES = typeof MastraStorage.TABLE_WORKFLOW_SNAPSHOT | typeof MastraStorage.TABLE_EVALS | typeof MastraStorage.TABLE_MESSAGES | typeof MastraStorage.TABLE_THREADS;
-declare abstract class MastraStorage extends MastraBase {
-    static readonly TABLE_WORKFLOW_SNAPSHOT = "mastra_workflow_snapshot";
-    static readonly TABLE_EVALS = "mastra_evals";
-    static readonly TABLE_MESSAGES = "mastra_messages";
-    static readonly TABLE_THREADS = "mastra_threads";
-    hasInit: boolean;
-    constructor({ name }: {
-        name: string;
-    });
-    abstract createTable({ tableName }: {
-        tableName: TABLE_NAMES;
-        schema: Record<string, StorageColumn>;
-    }): Promise<void>;
-    abstract clearTable({ tableName }: {
-        tableName: TABLE_NAMES;
-    }): Promise<void>;
-    abstract insert({ tableName, record }: {
-        tableName: TABLE_NAMES;
-        record: Record<string, any>;
-    }): Promise<void>;
-    abstract load<R>({ tableName, keys }: {
-        tableName: TABLE_NAMES;
-        keys: Record<string, string>;
-    }): Promise<R | null>;
-    abstract getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    __getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    abstract getThreadsByResourceId({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageThreadType[]>;
-    __getThreadsByResourceId({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageThreadType[]>;
-    abstract saveThread({ thread }: {
-        thread: StorageThreadType;
-    }): Promise<StorageThreadType>;
-    __saveThread({ thread }: {
-        thread: StorageThreadType;
-    }): Promise<StorageThreadType>;
-    abstract updateThread({ id, title, metadata, }: {
-        id: string;
-        title: string;
-        metadata: Record<string, unknown>;
-    }): Promise<StorageThreadType>;
-    __updateThread({ id, title, metadata, }: {
-        id: string;
-        title: string;
-        metadata: Record<string, unknown>;
-    }): Promise<StorageThreadType>;
-    abstract deleteThread({ threadId }: {
-        threadId: string;
-    }): Promise<void>;
-    __deleteThread({ threadId }: {
-        threadId: string;
-    }): Promise<void>;
-    abstract getMessages({ threadId, selectBy, threadConfig }: StorageGetMessagesArg): Promise<MessageType[]>;
-    __getMessages({ threadId, selectBy, threadConfig }: StorageGetMessagesArg): Promise<MessageType[]>;
-    abstract saveMessages({ messages }: {
-        messages: MessageType[];
-    }): Promise<MessageType[]>;
-    __saveMessages({ messages }: {
-        messages: MessageType[];
-    }): Promise<MessageType[]>;
-    init(): Promise<void>;
-    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
-        workflowName: string;
-        runId: string;
-        snapshot: WorkflowRunState;
-    }): Promise<void>;
-    loadWorkflowSnapshot({ workflowName, runId, }: {
-        workflowName: string;
-        runId: string;
-    }): Promise<WorkflowRunState | null>;
-}
-
-type AiMessageType = Message;
-type MessageType = {
-    id: string;
-    content: UserContent | AssistantContent | ToolContent;
-    role: 'system' | 'user' | 'assistant' | 'tool';
-    createdAt: Date;
-    threadId: string;
-    toolCallIds?: string[];
-    toolCallArgs?: Record<string, unknown>[];
-    toolNames?: string[];
-    type: 'text' | 'tool-call' | 'tool-result';
-};
-type StorageThreadType = {
-    id: string;
-    title?: string;
-    resourceId: string;
-    createdAt: Date;
-    updatedAt: Date;
-    metadata?: Record<string, unknown>;
-};
-type MessageResponse<T extends 'raw' | 'core_message'> = {
-    raw: MessageType[];
-    core_message: CoreMessage$1[];
-}[T];
-type MemoryConfig = {
-    lastMessages?: number | false;
-    semanticRecall?: boolean | {
-        topK: number;
-        messageRange: number | {
-            before: number;
-            after: number;
-        };
-    };
-    workingMemory?: {
-        enabled: boolean;
-        template?: string;
-    };
-};
-type SharedMemoryConfig = {
-    storage?: MastraStorage;
-    options?: MemoryConfig;
-    vector?: MastraVector;
-    embedder?: EmbeddingModel<string>;
-};
-/**
- * Abstract Memory class that defines the interface for storing and retrieving
- * conversation threads and messages.
- */
-declare abstract class MastraMemory extends MastraBase {
-    MAX_CONTEXT_TOKENS?: number;
-    storage: MastraStorage;
-    vector?: MastraVector;
-    embedder?: EmbeddingModel<string>;
-    protected threadConfig: MemoryConfig;
-    constructor(config: {
-        name: string;
-    } & SharedMemoryConfig);
-    /**
-     * Get a system message to inject into the conversation.
-     * This will be called before each conversation turn.
-     * Implementations can override this to inject custom system messages.
-     */
-    getSystemMessage(_input: {
-        threadId: string;
-        memoryConfig?: MemoryConfig;
-    }): Promise<string | null>;
-    protected createEmbeddingIndex(): Promise<{
-        indexName: string;
-    }>;
-    protected getEmbedder(): EmbeddingModel<string>;
-    protected getMergedThreadConfig(config?: MemoryConfig): MemoryConfig;
-    abstract rememberMessages({ threadId, vectorMessageSearch, config, }: {
-        threadId: string;
-        vectorMessageSearch?: string;
-        config?: MemoryConfig;
-    }): Promise<{
-        messages: CoreMessage$1[];
-        uiMessages: AiMessageType[];
-    }>;
-    estimateTokens(text: string): number;
-    protected parseMessages(messages: MessageType[]): CoreMessage$1[];
-    protected convertToUIMessages(messages: MessageType[]): AiMessageType[];
-    /**
-     * Retrieves a specific thread by its ID
-     * @param threadId - The unique identifier of the thread
-     * @returns Promise resolving to the thread or null if not found
-     */
-    abstract getThreadById({ threadId }: {
-        threadId: string;
-    }): Promise<StorageThreadType | null>;
-    abstract getThreadsByResourceId({ resourceId }: {
-        resourceId: string;
-    }): Promise<StorageThreadType[]>;
-    /**
-     * Saves or updates a thread
-     * @param thread - The thread data to save
-     * @returns Promise resolving to the saved thread
-     */
-    abstract saveThread({ thread, memoryConfig, }: {
-        thread: StorageThreadType;
-        memoryConfig?: MemoryConfig;
-    }): Promise<StorageThreadType>;
-    /**
-     * Saves messages to a thread
-     * @param messages - Array of messages to save
-     * @returns Promise resolving to the saved messages
-     */
-    abstract saveMessages({ messages, memoryConfig, }: {
-        messages: MessageType[];
-        memoryConfig: MemoryConfig | undefined;
-    }): Promise<MessageType[]>;
-    /**
-     * Retrieves all messages for a specific thread
-     * @param threadId - The unique identifier of the thread
-     * @returns Promise resolving to array of messages and uiMessages
-     */
-    abstract query({ threadId, selectBy, }: StorageGetMessagesArg): Promise<{
-        messages: CoreMessage$1[];
-        uiMessages: AiMessageType[];
-    }>;
-    /**
-     * Helper method to create a new thread
-     * @param title - Optional title for the thread
-     * @param metadata - Optional metadata for the thread
-     * @returns Promise resolving to the created thread
-     */
-    createThread({ threadId, resourceId, title, metadata, memoryConfig, }: {
-        resourceId: string;
-        threadId?: string;
-        title?: string;
-        metadata?: Record<string, unknown>;
-        memoryConfig?: MemoryConfig;
-    }): Promise<StorageThreadType>;
-    /**
-     * Helper method to delete a thread
-     * @param threadId - the id of the thread to delete
-     */
-    abstract deleteThread(threadId: string): Promise<void>;
-    /**
-     * Helper method to add a single message to a thread
-     * @param threadId - The thread to add the message to
-     * @param content - The message content
-     * @param role - The role of the message sender
-     * @param type - The type of the message
-     * @param toolNames - Optional array of tool names that were called
-     * @param toolCallArgs - Optional array of tool call arguments
-     * @param toolCallIds - Optional array of tool call ids
-     * @returns Promise resolving to the saved message
-     */
-    addMessage({ threadId, config, content, role, type, toolNames, toolCallArgs, toolCallIds, }: {
-        threadId: string;
-        config?: MemoryConfig;
-        content: UserContent | AssistantContent;
-        role: 'user' | 'assistant';
-        type: 'text' | 'tool-call' | 'tool-result';
-        toolNames?: string[];
-        toolCallArgs?: Record<string, unknown>[];
-        toolCallIds?: string[];
-    }): Promise<MessageType>;
-    /**
-     * Generates a unique identifier
-     * @returns A unique string ID
-     */
-    generateId(): string;
-}
-
-type MastraPrimitives = {
-    logger?: Logger;
-    telemetry?: Telemetry;
-    storage?: MastraStorage;
-    agents?: Record<string, Agent>;
-    tts?: Record<string, MastraTTS>;
-    vectors?: Record<string, MastraVector>;
-    memory?: MastraMemory;
-};
-interface IExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined, TContext extends WorkflowContext = WorkflowContext> {
-    context: TSchemaIn extends z.ZodSchema ? z.infer<TSchemaIn> & TContext : TContext;
-    runId?: string;
-    mastra?: MastraPrimitives;
-    suspend: () => Promise<void>;
-}
-interface IAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined, TSchemaOut extends z.ZodSchema | undefined, TContext extends IExecutionContext<TSchemaIn>> {
-    id: TId;
-    description?: string;
-    inputSchema?: TSchemaIn;
-    outputSchema?: TSchemaOut;
-    mastra?: MastraPrimitives;
-    payload?: TSchemaIn extends z.ZodSchema ? Partial<z.infer<TSchemaIn>> : unknown;
-    execute: (context: TContext) => Promise<TSchemaOut extends z.ZodSchema ? z.infer<TSchemaOut> : unknown>;
-    [key: string]: any;
-}
-
-type CoreTool = {
-    description?: string;
-    parameters: ZodSchema;
-    execute: (params: any) => Promise<any>;
-};
-interface ToolExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined, TContext extends WorkflowContext = WorkflowContext> extends IExecutionContext<TSchemaIn, TContext> {
-}
-interface ToolAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined = undefined, TSchemaOut extends z.ZodSchema | undefined = undefined, TContext extends ToolExecutionContext<TSchemaIn> = ToolExecutionContext<TSchemaIn>> extends IAction<TId, TSchemaIn, TSchemaOut, TContext> {
-}
-
-type ToolsetsInput = Record<string, Record<string, ToolAction<any, any, any, any>>>;
-type ToolsInput = Record<string, ToolAction<any, any, any, any>>;
-interface AgentConfig<TTools extends Record<string, ToolAction<any, any, any, any>> = Record<string, ToolAction<any, any, any, any>>, TMetrics extends Record<string, Metric> = Record<string, Metric>> {
-    name: string;
-    instructions: string;
-    model: LanguageModelV1;
-    tools?: TTools;
-    mastra?: MastraPrimitives;
-    metrics?: TMetrics;
-    memory?: MastraMemory;
-}
-interface AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> {
-    toolsets?: ToolsetsInput;
-    resourceId?: string;
-    context?: CoreMessage[];
-    threadId?: string;
-    memoryOptions?: MemoryConfig;
-    runId?: string;
-    onStepFinish?: (step: string) => void;
-    maxSteps?: number;
-    output?: OutputType | Z;
-    temperature?: number;
-    toolChoice?: 'auto' | 'required';
-}
-interface AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> {
-    toolsets?: ToolsetsInput;
-    resourceId?: string;
-    context?: CoreMessage[];
-    threadId?: string;
-    memoryOptions?: MemoryConfig;
-    runId?: string;
-    onFinish?: (result: string) => Promise<void> | void;
-    onStepFinish?: (step: string) => void;
-    maxSteps?: number;
-    output?: OutputType | Z;
-    temperature?: number;
-    toolChoice?: 'auto' | 'required';
-}
-
-type LanguageModel = LanguageModelV1;
-type CoreMessage = CoreMessage$1;
-type CoreSystemMessage = CoreSystemMessage$1;
-type CoreAssistantMessage = CoreAssistantMessage$1;
-type CoreUserMessage = CoreUserMessage$1;
-type CoreToolMessage = CoreToolMessage$1;
-type EmbedResult<T> = EmbedResult$1<T>;
-type EmbedManyResult<T> = EmbedManyResult$1<T>;
-type BaseStructuredOutputType = 'string' | 'number' | 'boolean' | 'date';
-type StructuredOutputType = 'array' | 'string' | 'number' | 'object' | 'boolean' | 'date';
-type StructuredOutputArrayItem = {
-    type: BaseStructuredOutputType;
-} | {
-    type: 'object';
-    items: StructuredOutput;
-};
-type StructuredOutput = {
-    [key: string]: {
-        type: BaseStructuredOutputType;
-    } | {
-        type: 'object';
-        items: StructuredOutput;
-    } | {
-        type: 'array';
-        items: StructuredOutputArrayItem;
-    };
-};
-type GenerateReturn<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = Z extends undefined ? GenerateTextResult<any, any> : GenerateObjectResult<any>;
-type StreamReturn<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = Z extends undefined ? StreamTextResult<any, any> : StreamObjectResult<any, any, any>;
-type OutputType = 'text' | StructuredOutput;
-type LLMStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
-    runId?: string;
-    onFinish?: (result: string) => Promise<void> | void;
-    onStepFinish?: (step: string) => void;
-    maxSteps?: number;
-    tools?: ToolsInput;
-    convertedTools?: Record<string, CoreTool>;
-    output?: OutputType | Z;
-    temperature?: number;
+type GenerateReturn<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = Z extends undefined ? GenerateTextResult<any, any> : GenerateObjectResult<any>;
+type StreamReturn<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = Z extends undefined ? StreamTextResult<any, any> : StreamObjectResult<any, any, any>;
+type OutputType = 'text' | StructuredOutput;
+type LLMStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
+    runId?: string;
+    onFinish?: (result: string) => Promise<void> | void;
+    onStepFinish?: (step: string) => void;
+    maxSteps?: number;
+    tools?: ToolsInput;
+    convertedTools?: Record<string, CoreTool>;
+    output?: OutputType | Z;
+    temperature?: number;
 };
 type LLMTextOptions = {
     tools?: ToolsInput;
@@ -701,8 +545,8 @@ type LLMInnerStreamOptions = {
     tools?: ToolsInput;
     convertedTools?: Record<string, CoreTool>;
     messages: CoreMessage[];
-    onStepFinish?: (step: string) => void;
-    onFinish?: (result: string) => Promise<void> | void;
+    onStepFinish?: (step: string) => unknown;
+    onFinish?: (result: string) => unknown;
     maxSteps?: number;
     temperature?: number;
     toolChoice?: 'auto' | 'required';
@@ -719,6 +563,7 @@ declare class MastraLLMBase extends MastraBase {
     });
     getProvider(): string;
     getModelId(): string;
+    getModel(): ai.LanguageModelV1;
     convertToMessages(messages: string | string[] | CoreMessage$1[]): CoreMessage$1[];
     __registerPrimitives(p: MastraPrimitives): void;
     __text(input: LLMTextOptions): Promise<GenerateTextResult<any, any>>;
@@ -736,7 +581,9 @@ declare class Agent<TTools extends Record<string, ToolAction<any, any, any, any>
     instructions: string;
     readonly model?: LanguageModelV1;
     tools: TTools;
+    /** @deprecated This property is deprecated. Use evals instead. */
     metrics: TMetrics;
+    evals: TMetrics;
     constructor(config: AgentConfig<TTools, TMetrics>);
     hasOwnMemory(): boolean;
     getMemory(): MastraMemory | undefined;
@@ -771,7 +618,7 @@ declare class Agent<TTools extends Record<string, ToolAction<any, any, any, any>
         threadId: string;
         memoryConfig: MemoryConfig | undefined;
     }): Promise<void>;
-    sanitizeResponseMessages(messages: Array<CoreToolMessage$1 | CoreAssistantMessage$1>): Array<CoreToolMessage$1 | CoreAssistantMessage$1>;
+    sanitizeResponseMessages(messages: Array<CoreMessage$1>): Array<CoreMessage$1>;
     convertTools({ toolsets, runId, }: {
         toolsets?: ToolsetsInput;
         threadId?: string;
@@ -813,4 +660,204 @@ declare class Agent<TTools extends Record<string, ToolAction<any, any, any, any>
     stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], { context, threadId: threadIdInFn, memoryOptions, resourceId, maxSteps, onFinish, onStepFinish, runId, toolsets, output, temperature, toolChoice, }?: AgentStreamOptions<Z>): Promise<StreamReturn<Z>>;
 }
 
-export { type StepDef as $, Agent as A, type BaseStructuredOutputType as B, type CoreMessage as C, type LLMTextObjectOptions as D, type EvalRow as E, type LLMInnerStreamOptions as F, type GenerateReturn as G, type LLMStreamObjectOptions as H, type IAction as I, type AiMessageType as J, type MessageType as K, type LanguageModel as L, MastraStorage as M, type StorageThreadType as N, type OutputType as O, type MessageResponse as P, type MemoryConfig as Q, type RetryConfig as R, type StepExecutionContext as S, type ToolAction as T, type SharedMemoryConfig as U, type CoreTool as V, type WorkflowOptions as W, type StepNode as X, type VariableReference as Y, type BaseCondition as Z, type ActionContext as _, MastraMemory as a, type StepCondition as a0, type WorkflowContext as a1, type WorkflowLogMessage as a2, type WorkflowEvent as a3, type ResolverFunctionInput as a4, type ResolverFunctionOutput as a5, type SubscriberFunctionOutput as a6, type DependencyCheckOutput as a7, type WorkflowActors as a8, type WorkflowActionParams as a9, type WorkflowActions as aa, type WorkflowState as ab, type StepId as ac, type ExtractSchemaFromStep as ad, type ExtractStepResult as ae, type StepInputType as af, type ExtractSchemaType as ag, type PathsToStringProps as ah, type StepAction as b, type MastraPrimitives as c, type StepVariableType as d, type StepConfig as e, type StepResult as f, type WorkflowRunState as g, type StepGraph as h, type AgentConfig as i, type ToolExecutionContext as j, type TABLE_NAMES as k, type StorageColumn as l, type WorkflowRow as m, type StorageGetMessagesArg as n, type CoreSystemMessage as o, type CoreAssistantMessage as p, type CoreUserMessage as q, type CoreToolMessage as r, type EmbedResult as s, type EmbedManyResult as t, type StructuredOutputType as u, type StructuredOutputArrayItem as v, type StructuredOutput as w, type StreamReturn as x, type LLMStreamOptions as y, type LLMTextOptions as z };
+interface StorageColumn {
+    type: 'text' | 'timestamp' | 'uuid' | 'jsonb' | 'integer' | 'bigint';
+    primaryKey?: boolean;
+    nullable?: boolean;
+    references?: {
+        table: string;
+        column: string;
+    };
+}
+interface WorkflowRow {
+    workflow_name: string;
+    run_id: string;
+    snapshot: WorkflowRunState;
+    created_at: Date;
+    updated_at: Date;
+}
+type StorageGetMessagesArg = {
+    threadId: string;
+    selectBy?: {
+        vectorSearchString?: string;
+        last?: number | false;
+        include?: {
+            id: string;
+            withPreviousMessages?: number;
+            withNextMessages?: number;
+        }[];
+    };
+    threadConfig?: MemoryConfig;
+};
+type EvalRow = {
+    input: string;
+    output: string;
+    result: MetricResult;
+    agentName: string;
+    createdAt: string;
+    metricName: string;
+    instructions: string;
+    runId: string;
+    globalRunId: string;
+    testInfo?: TestInfo;
+};
+
+declare const TABLE_WORKFLOW_SNAPSHOT = "mastra_workflow_snapshot";
+declare const TABLE_EVALS = "mastra_evals";
+declare const TABLE_MESSAGES = "mastra_messages";
+declare const TABLE_THREADS = "mastra_threads";
+declare const TABLE_TRACES = "mastra_traces";
+type TABLE_NAMES = typeof TABLE_WORKFLOW_SNAPSHOT | typeof TABLE_EVALS | typeof TABLE_MESSAGES | typeof TABLE_THREADS | typeof TABLE_TRACES;
+
+type MessageType = {
+    id: string;
+    content: UserContent | AssistantContent | ToolContent;
+    role: 'system' | 'user' | 'assistant' | 'tool';
+    createdAt: Date;
+    threadId: string;
+    toolCallIds?: string[];
+    toolCallArgs?: Record<string, unknown>[];
+    toolNames?: string[];
+    type: 'text' | 'tool-call' | 'tool-result';
+};
+type StorageThreadType = {
+    id: string;
+    title?: string;
+    resourceId: string;
+    createdAt: Date;
+    updatedAt: Date;
+    metadata?: Record<string, unknown>;
+};
+type MessageResponse<T extends 'raw' | 'core_message'> = {
+    raw: MessageType[];
+    core_message: CoreMessage$1[];
+}[T];
+type MemoryConfig = {
+    lastMessages?: number | false;
+    semanticRecall?: boolean | {
+        topK: number;
+        messageRange: number | {
+            before: number;
+            after: number;
+        };
+    };
+    workingMemory?: {
+        enabled: boolean;
+        template?: string;
+    };
+};
+type SharedMemoryConfig = {
+    storage?: MastraStorage;
+    options?: MemoryConfig;
+    vector?: MastraVector;
+    embedder?: EmbeddingModel<string>;
+};
+
+declare abstract class MastraStorage extends MastraBase {
+    /** @deprecated import from { TABLE_WORKFLOW_SNAPSHOT } '@mastra/core/storage' instead */
+    static readonly TABLE_WORKFLOW_SNAPSHOT = "mastra_workflow_snapshot";
+    /** @deprecated import from { TABLE_EVALS } '@mastra/core/storage' instead */
+    static readonly TABLE_EVALS = "mastra_evals";
+    /** @deprecated import from { TABLE_MESSAGES } '@mastra/core/storage' instead */
+    static readonly TABLE_MESSAGES = "mastra_messages";
+    /** @deprecated import from { TABLE_THREADS } '@mastra/core/storage' instead */
+    static readonly TABLE_THREADS = "mastra_threads";
+    /** @deprecated import { TABLE_TRACES } from '@mastra/core/storage' instead */
+    static readonly TABLE_TRACES = "mastra_traces";
+    protected hasInitialized: null | Promise<boolean>;
+    constructor({ name }: {
+        name: string;
+    });
+    abstract createTable({ tableName }: {
+        tableName: TABLE_NAMES;
+        schema: Record<string, StorageColumn>;
+    }): Promise<void>;
+    abstract clearTable({ tableName }: {
+        tableName: TABLE_NAMES;
+    }): Promise<void>;
+    abstract insert({ tableName, record }: {
+        tableName: TABLE_NAMES;
+        record: Record<string, any>;
+    }): Promise<void>;
+    abstract batchInsert({ tableName, records, }: {
+        tableName: TABLE_NAMES;
+        records: Record<string, any>[];
+    }): Promise<void>;
+    __batchInsert({ tableName, records, }: {
+        tableName: TABLE_NAMES;
+        records: Record<string, any>[];
+    }): Promise<void>;
+    abstract load<R>({ tableName, keys }: {
+        tableName: TABLE_NAMES;
+        keys: Record<string, string>;
+    }): Promise<R | null>;
+    abstract getThreadById({ threadId }: {
+        threadId: string;
+    }): Promise<StorageThreadType | null>;
+    __getThreadById({ threadId }: {
+        threadId: string;
+    }): Promise<StorageThreadType | null>;
+    abstract getThreadsByResourceId({ resourceId }: {
+        resourceId: string;
+    }): Promise<StorageThreadType[]>;
+    __getThreadsByResourceId({ resourceId }: {
+        resourceId: string;
+    }): Promise<StorageThreadType[]>;
+    abstract saveThread({ thread }: {
+        thread: StorageThreadType;
+    }): Promise<StorageThreadType>;
+    __saveThread({ thread }: {
+        thread: StorageThreadType;
+    }): Promise<StorageThreadType>;
+    abstract updateThread({ id, title, metadata, }: {
+        id: string;
+        title: string;
+        metadata: Record<string, unknown>;
+    }): Promise<StorageThreadType>;
+    __updateThread({ id, title, metadata, }: {
+        id: string;
+        title: string;
+        metadata: Record<string, unknown>;
+    }): Promise<StorageThreadType>;
+    abstract deleteThread({ threadId }: {
+        threadId: string;
+    }): Promise<void>;
+    __deleteThread({ threadId }: {
+        threadId: string;
+    }): Promise<void>;
+    abstract getMessages({ threadId, selectBy, threadConfig }: StorageGetMessagesArg): Promise<MessageType[]>;
+    __getMessages({ threadId, selectBy, threadConfig }: StorageGetMessagesArg): Promise<MessageType[]>;
+    abstract saveMessages({ messages }: {
+        messages: MessageType[];
+    }): Promise<MessageType[]>;
+    __saveMessages({ messages }: {
+        messages: MessageType[];
+    }): Promise<MessageType[]>;
+    abstract getTraces({ name, scope, page, perPage, attributes, }: {
+        name?: string;
+        scope?: string;
+        page: number;
+        perPage: number;
+        attributes?: Record<string, string>;
+    }): Promise<any[]>;
+    __getTraces({ scope, page, perPage, attributes, }: {
+        scope?: string;
+        page: number;
+        perPage: number;
+        attributes?: Record<string, string>;
+    }): Promise<any[]>;
+    init(): Promise<void>;
+    persistWorkflowSnapshot({ workflowName, runId, snapshot, }: {
+        workflowName: string;
+        runId: string;
+        snapshot: WorkflowRunState;
+    }): Promise<void>;
+    loadWorkflowSnapshot({ workflowName, runId, }: {
+        workflowName: string;
+        runId: string;
+    }): Promise<WorkflowRunState | null>;
+    abstract getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    __getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+}
+
+export { type WorkflowContext as $, Agent as A, type BaseStructuredOutputType as B, type CoreMessage as C, type LLMInnerStreamOptions as D, type EvalRow as E, type LLMStreamObjectOptions as F, type GenerateReturn as G, type MessageType as H, type IAction as I, type StorageThreadType as J, type MessageResponse as K, type LanguageModel as L, MastraMemory as M, type MemoryConfig as N, type OutputType as O, type SharedMemoryConfig as P, type CoreTool as Q, type RetryConfig as R, type StepExecutionContext as S, type ToolAction as T, type StepNode as U, type VariableReference as V, type WorkflowOptions as W, type BaseCondition as X, type ActionContext as Y, type StepDef as Z, type StepCondition as _, MastraStorage as a, type WorkflowLogMessage as a0, type WorkflowEvent as a1, type ResolverFunctionInput as a2, type ResolverFunctionOutput as a3, type SubscriberFunctionOutput as a4, type DependencyCheckOutput as a5, type WorkflowActors as a6, type WorkflowActionParams as a7, type WorkflowActions as a8, type WorkflowState as a9, type StepId as aa, type ExtractSchemaFromStep as ab, type ExtractStepResult as ac, type StepInputType as ad, type ExtractSchemaType as ae, type PathsToStringProps as af, type TABLE_NAMES as ag, TABLE_WORKFLOW_SNAPSHOT as ah, TABLE_EVALS as ai, TABLE_MESSAGES as aj, TABLE_THREADS as ak, TABLE_TRACES as al, type StepAction as b, type MastraPrimitives as c, type StepVariableType as d, type StepConfig as e, type StepResult as f, type WorkflowRunState as g, type StepGraph as h, type AgentConfig as i, type ToolExecutionContext as j, type StorageColumn as k, type WorkflowRow as l, type StorageGetMessagesArg as m, type CoreSystemMessage as n, type CoreAssistantMessage as o, type CoreUserMessage as p, type CoreToolMessage as q, type EmbedResult as r, type EmbedManyResult as s, type StructuredOutputType as t, type StructuredOutputArrayItem as u, type StructuredOutput as v, type StreamReturn as w, type LLMStreamOptions as x, type LLMTextOptions as y, type LLMTextObjectOptions as z };