@mastra/core 0.8.3-alpha.0 → 0.8.3-alpha.2

package/{LICENSE → LICENSE.md}

@@ -1,4 +1,6 @@
-Elastic License 2.0 (ELv2)
+# Elastic License 2.0 (ELv2)
+
+Copyright (c) 2025 Mastra AI, Inc.
 
 **Acceptance**
 By using the software, you agree to all of the terms and conditions below.

package/dist/agent/index.cjs

@@ -1,10 +1,10 @@
 'use strict';
 
-var chunkMP3UIX44_cjs = require('../chunk-MP3UIX44.cjs');
+var chunkLRDD5WDF_cjs = require('../chunk-LRDD5WDF.cjs');
 
 
 
 Object.defineProperty(exports, "Agent", {
   enumerable: true,
-  get: function () { return chunkMP3UIX44_cjs.Agent; }
+  get: function () { return chunkLRDD5WDF_cjs.Agent; }
 });

package/dist/agent/index.d.cts

@@ -1,8 +1,9 @@
 export { Message as AiMessageType } from 'ai';
 import 'json-schema';
 import 'zod';
-export { j as Agent, k as AgentConfig, aM as AgentGenerateOptions, aN as AgentStreamOptions, aG as MastraLanguageModel, aH as ToolsInput, aL as ToolsetsInput } from '../base-VLg7uRIB.cjs';
+export { j as Agent, k as AgentConfig, aM as AgentGenerateOptions, aN as AgentStreamOptions, aG as MastraLanguageModel, aH as ToolsInput, aL as ToolsetsInput } from '../base-Ca8Hlcj4.cjs';
 import '../base-BihKcqDY.cjs';
+import '../di/index.cjs';
 import '../types-BtMyV38I.cjs';
 import 'sift';
 import '../index-CquI0inB.cjs';

package/dist/agent/index.d.ts

@@ -1,8 +1,9 @@
 export { Message as AiMessageType } from 'ai';
 import 'json-schema';
 import 'zod';
-export { j as Agent, k as AgentConfig, aM as AgentGenerateOptions, aN as AgentStreamOptions, aG as MastraLanguageModel, aH as ToolsInput, aL as ToolsetsInput } from '../base-BZEw2nXJ.js';
+export { j as Agent, k as AgentConfig, aM as AgentGenerateOptions, aN as AgentStreamOptions, aG as MastraLanguageModel, aH as ToolsInput, aL as ToolsetsInput } from '../base-Cb5pwtMe.js';
 import '../base-DT2poiVK.js';
+import '../di/index.js';
 import '../types-BtMyV38I.js';
 import 'sift';
 import '../index-CquI0inB.js';

package/dist/agent/index.js

@@ -1 +1 @@
-export { Agent } from '../chunk-FFOLLSHP.js';
+export { Agent } from '../chunk-KB4JVOEB.js';

package/dist/{base-VLg7uRIB.d.cts → base-Ca8Hlcj4.d.cts}

@@ -5,6 +5,7 @@ import { a as Metric, M as MetricResult, T as TestInfo } from './types-BtMyV38I.
 import { Query } from 'sift';
 import { z, ZodSchema } from 'zod';
 import { JSONSchema7 } from 'json-schema';
+import { Container } from './di/index.cjs';
 import { B as BaseLogMessage, R as RegisteredLogger, L as Logger, d as Run } from './index-CquI0inB.cjs';
 import { Span } from '@opentelemetry/api';
 import * as xstate from 'xstate';
@@ -33,6 +34,7 @@ type CoreTool = {
 });
 interface ToolExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined> extends IExecutionContext<TSchemaIn> {
     mastra?: MastraUnion;
+    container: Container;
 }
 interface ToolAction<TSchemaIn extends z.ZodSchema | undefined = undefined, TSchemaOut extends z.ZodSchema | undefined = undefined, TContext extends ToolExecutionContext<TSchemaIn> = ToolExecutionContext<TSchemaIn>> extends IAction<string, TSchemaIn, TSchemaOut, TContext, ToolExecutionOptions> {
     description: string;
@@ -191,6 +193,7 @@ interface StepExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undef
     runId: string;
     emit: (event: string, data: any) => void;
     mastra?: MastraUnion;
+    container: Container;
 }
 interface StepAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined, TSchemaOut extends z.ZodSchema | undefined, TContext extends StepExecutionContext<TSchemaIn>> extends IAction<TId, TSchemaIn, TSchemaOut, TContext> {
     mastra?: Mastra;
@@ -565,9 +568,10 @@ declare class Machine<TSteps extends Step<any, any, any, any>[] = Step<any, any,
     #private;
     logger: Logger;
     name: string;
-    constructor({ logger, mastra, workflowInstance, executionSpan, name, runId, steps, stepGraph, retryConfig, startStepId, }: {
+    constructor({ logger, mastra, container, workflowInstance, executionSpan, name, runId, steps, stepGraph, retryConfig, startStepId, }: {
         logger: Logger;
         mastra?: Mastra;
+        container: Container;
         workflowInstance: WorkflowInstance;
         executionSpan?: Span;
         name: string;
@@ -939,13 +943,15 @@ interface WorkflowResultReturn<TResult extends z.ZodObject<any>, T extends z.Zod
     runId: string;
     start: (props?: {
         triggerData?: z.infer<T>;
+        container?: Container;
     } | undefined) => Promise<WorkflowRunResult<T, TSteps, TResult>>;
     watch: (onTransition: (state: Pick<WorkflowRunResult<T, TSteps, TResult>, 'results' | 'activePaths' | 'runId'>) => void) => () => void;
     resume: (props: {
         stepId: string;
         context?: Record<string, any>;
+        container?: Container;
     }) => Promise<Omit<WorkflowRunResult<T, TSteps, TResult>, 'runId'> | undefined>;
-    resumeWithEvent: (eventName: string, data: any) => Promise<Omit<WorkflowRunResult<T, TSteps, TResult>, 'runId'> | undefined>;
+    resumeWithEvent: (eventName: string, data: any, container?: Container) => Promise<Omit<WorkflowRunResult<T, TSteps, TResult>, 'runId'> | undefined>;
 }
 declare class WorkflowInstance<TSteps extends Step<any, any, any, any>[] = Step<any, any, any, any>[], TTriggerSchema extends z.ZodObject<any> = any, TResult extends z.ZodObject<any> = any> implements WorkflowResultReturn<TResult, TTriggerSchema, TSteps> {
     #private;
@@ -977,8 +983,9 @@ declare class WorkflowInstance<TSteps extends Step<any, any, any, any>[] = Step<
     get runId(): string;
     get executionSpan(): Span | undefined;
     watch(onTransition: (state: Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, 'results' | 'activePaths' | 'runId' | 'timestamp'>) => void): () => void;
-    start({ triggerData }?: {
+    start({ triggerData, container }?: {
         triggerData?: z.infer<TTriggerSchema>;
+        container?: Container;
     }): Promise<{
         runId: string;
         timestamp: number;
@@ -992,28 +999,31 @@ declare class WorkflowInstance<TSteps extends Step<any, any, any, any>[] = Step<
         }>;
     }>;
     private isCompoundDependencyMet;
-    execute({ triggerData, snapshot, stepId, resumeData, }?: {
+    execute({ triggerData, snapshot, stepId, resumeData, container, }?: {
         stepId?: string;
         triggerData?: z.infer<TTriggerSchema>;
         snapshot?: Snapshot<any>;
         resumeData?: any;
+        container: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, 'runId'>>;
     hasSubscribers(stepId: string): boolean;
-    runMachine(parentStepId: string, input: any): Promise<(Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId" | "timestamp" | "results" | "activePaths"> | undefined)[]>;
+    runMachine(parentStepId: string, input: any, container?: Container): Promise<(Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId" | "timestamp" | "results" | "activePaths"> | undefined)[]>;
     suspend(stepId: string, machine: Machine<TSteps, TTriggerSchema>): Promise<void>;
     /**
      * Persists the workflow state to the database
      */
     persistWorkflowSnapshot(): Promise<void>;
     getState(): Promise<WorkflowRunState | null>;
-    resumeWithEvent(eventName: string, data: any): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
-    resume({ stepId, context: resumeContext }: {
+    resumeWithEvent(eventName: string, data: any, container?: Container): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
+    resume({ stepId, context: resumeContext, container, }: {
         stepId: string;
         context?: Record<string, any>;
+        container?: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
-    _resume({ stepId, context: resumeContext }: {
+    _resume({ stepId, context: resumeContext, container, }: {
         stepId: string;
         context?: Record<string, any>;
+        container: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
 }
 
@@ -1082,10 +1092,11 @@ declare class Workflow<TSteps extends Step<string, any, any>[] = Step<string, an
     getWorkflowRuns(): Promise<WorkflowRuns>;
     getExecutionSpan(runId: string): Span | undefined;
     getState(runId: string): Promise<WorkflowRunState | null>;
-    resume({ runId, stepId, context: resumeContext, }: {
+    resume({ runId, stepId, context: resumeContext, container, }: {
         runId: string;
         stepId: string;
         context?: Record<string, any>;
+        container: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, any>, "runId"> | undefined>;
     watch(onTransition: (state: Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResultSchema>, 'results' | 'activePaths' | 'runId' | 'timestamp'>) => void): () => void;
     resumeWithEvent(runId: string, eventName: string, data: any): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, any>, "runId"> | undefined>;
@@ -1309,6 +1320,7 @@ type MastraCustomLLMOptions<Z extends ZodSchema | JSONSchema7 | undefined = unde
     telemetry?: TelemetrySettings;
     threadId?: string;
     resourceId?: string;
+    container: Container;
 } & Run;
 type LLMTextOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
     messages: CoreMessage[];
@@ -1568,6 +1580,8 @@ type AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefi
     };
     /** Telemetry settings */
     telemetry?: TelemetrySettings;
+    /** Container for dependency injection */
+    container?: Container;
 } & ({
     resourceId?: undefined;
     threadId?: undefined;
@@ -1609,6 +1623,8 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
     experimental_output?: Z;
     /** Telemetry settings */
     telemetry?: TelemetrySettings;
+    /** Container for dependency injection */
+    container?: Container;
 } & ({
     resourceId?: undefined;
     threadId?: undefined;
@@ -1758,10 +1774,10 @@ declare class MastraLLMBase extends MastraBase {
     __registerMastra(p: Mastra): void;
     __text<Z extends ZodSchema | JSONSchema7 | undefined>(input: LLMTextOptions<Z>): Promise<GenerateTextResult<any, any>>;
     __textObject<T extends ZodSchema | JSONSchema7 | undefined>(input: LLMTextObjectOptions<T>): Promise<GenerateObjectResult<T>>;
-    generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options?: LLMStreamOptions<Z>): Promise<GenerateReturn<Z>>;
+    generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options: LLMStreamOptions<Z>): Promise<GenerateReturn<Z>>;
     __stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(input: LLMInnerStreamOptions<Z>): Promise<StreamTextResult<any, any>>;
     __streamObject<T extends ZodSchema | JSONSchema7 | undefined>(input: LLMStreamObjectOptions<T>): Promise<StreamObjectResult<DeepPartial<T>, T, never>>;
-    stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options?: LLMStreamOptions<Z>): Promise<StreamReturn<Z>>;
+    stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options: LLMStreamOptions<Z>): Promise<StreamReturn<Z>>;
 }
 
 declare class Agent<TAgentId extends string = string, TTools extends ToolsInput = ToolsInput, TMetrics extends Record<string, Metric> = Record<string, Metric>> extends MastraBase {
@@ -1807,11 +1823,12 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
     }>;
     private getResponseMessages;
     sanitizeResponseMessages(messages: Array<CoreMessage$1>): Array<CoreMessage$1>;
-    convertTools({ toolsets, threadId, resourceId, runId, }: {
+    convertTools({ toolsets, threadId, resourceId, runId, container, }: {
         toolsets?: ToolsetsInput;
         threadId?: string;
         resourceId?: string;
         runId?: string;
+        container: Container;
     }): Record<string, CoreTool>;
     preExecute({ resourceId, runId, threadId, thread, memoryConfig, messages, systemMessage, }: {
         runId?: string;
@@ -1825,7 +1842,7 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
         coreMessages: CoreMessage$1[];
         threadIdToUse: string;
     }>;
-    __primitive({ instructions, messages, context, threadId, memoryConfig, resourceId, runId, toolsets, }: {
+    __primitive({ instructions, messages, context, threadId, memoryConfig, resourceId, runId, toolsets, container, }: {
         instructions?: string;
         toolsets?: ToolsetsInput;
         resourceId?: string;
@@ -1834,6 +1851,7 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
         context?: CoreMessage$1[];
         runId?: string;
         messages: CoreMessage$1[];
+        container: Container;
     }): {
         before: () => Promise<{
             messageObjects: CoreMessage$1[];

package/dist/{base-BZEw2nXJ.d.ts → base-Cb5pwtMe.d.ts}

@@ -5,6 +5,7 @@ import { a as Metric, M as MetricResult, T as TestInfo } from './types-BtMyV38I.
 import { Query } from 'sift';
 import { z, ZodSchema } from 'zod';
 import { JSONSchema7 } from 'json-schema';
+import { Container } from './di/index.js';
 import { B as BaseLogMessage, R as RegisteredLogger, L as Logger, d as Run } from './index-CquI0inB.js';
 import { Span } from '@opentelemetry/api';
 import * as xstate from 'xstate';
@@ -33,6 +34,7 @@ type CoreTool = {
 });
 interface ToolExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined> extends IExecutionContext<TSchemaIn> {
     mastra?: MastraUnion;
+    container: Container;
 }
 interface ToolAction<TSchemaIn extends z.ZodSchema | undefined = undefined, TSchemaOut extends z.ZodSchema | undefined = undefined, TContext extends ToolExecutionContext<TSchemaIn> = ToolExecutionContext<TSchemaIn>> extends IAction<string, TSchemaIn, TSchemaOut, TContext, ToolExecutionOptions> {
     description: string;
@@ -191,6 +193,7 @@ interface StepExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undef
     runId: string;
     emit: (event: string, data: any) => void;
     mastra?: MastraUnion;
+    container: Container;
 }
 interface StepAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined, TSchemaOut extends z.ZodSchema | undefined, TContext extends StepExecutionContext<TSchemaIn>> extends IAction<TId, TSchemaIn, TSchemaOut, TContext> {
     mastra?: Mastra;
@@ -565,9 +568,10 @@ declare class Machine<TSteps extends Step<any, any, any, any>[] = Step<any, any,
     #private;
     logger: Logger;
     name: string;
-    constructor({ logger, mastra, workflowInstance, executionSpan, name, runId, steps, stepGraph, retryConfig, startStepId, }: {
+    constructor({ logger, mastra, container, workflowInstance, executionSpan, name, runId, steps, stepGraph, retryConfig, startStepId, }: {
         logger: Logger;
         mastra?: Mastra;
+        container: Container;
         workflowInstance: WorkflowInstance;
         executionSpan?: Span;
         name: string;
@@ -939,13 +943,15 @@ interface WorkflowResultReturn<TResult extends z.ZodObject<any>, T extends z.Zod
     runId: string;
     start: (props?: {
         triggerData?: z.infer<T>;
+        container?: Container;
     } | undefined) => Promise<WorkflowRunResult<T, TSteps, TResult>>;
     watch: (onTransition: (state: Pick<WorkflowRunResult<T, TSteps, TResult>, 'results' | 'activePaths' | 'runId'>) => void) => () => void;
     resume: (props: {
         stepId: string;
         context?: Record<string, any>;
+        container?: Container;
     }) => Promise<Omit<WorkflowRunResult<T, TSteps, TResult>, 'runId'> | undefined>;
-    resumeWithEvent: (eventName: string, data: any) => Promise<Omit<WorkflowRunResult<T, TSteps, TResult>, 'runId'> | undefined>;
+    resumeWithEvent: (eventName: string, data: any, container?: Container) => Promise<Omit<WorkflowRunResult<T, TSteps, TResult>, 'runId'> | undefined>;
 }
 declare class WorkflowInstance<TSteps extends Step<any, any, any, any>[] = Step<any, any, any, any>[], TTriggerSchema extends z.ZodObject<any> = any, TResult extends z.ZodObject<any> = any> implements WorkflowResultReturn<TResult, TTriggerSchema, TSteps> {
     #private;
@@ -977,8 +983,9 @@ declare class WorkflowInstance<TSteps extends Step<any, any, any, any>[] = Step<
     get runId(): string;
     get executionSpan(): Span | undefined;
     watch(onTransition: (state: Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, 'results' | 'activePaths' | 'runId' | 'timestamp'>) => void): () => void;
-    start({ triggerData }?: {
+    start({ triggerData, container }?: {
         triggerData?: z.infer<TTriggerSchema>;
+        container?: Container;
     }): Promise<{
         runId: string;
         timestamp: number;
@@ -992,28 +999,31 @@ declare class WorkflowInstance<TSteps extends Step<any, any, any, any>[] = Step<
         }>;
     }>;
     private isCompoundDependencyMet;
-    execute({ triggerData, snapshot, stepId, resumeData, }?: {
+    execute({ triggerData, snapshot, stepId, resumeData, container, }?: {
         stepId?: string;
         triggerData?: z.infer<TTriggerSchema>;
         snapshot?: Snapshot<any>;
         resumeData?: any;
+        container: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, 'runId'>>;
     hasSubscribers(stepId: string): boolean;
-    runMachine(parentStepId: string, input: any): Promise<(Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId" | "timestamp" | "results" | "activePaths"> | undefined)[]>;
+    runMachine(parentStepId: string, input: any, container?: Container): Promise<(Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId" | "timestamp" | "results" | "activePaths"> | undefined)[]>;
     suspend(stepId: string, machine: Machine<TSteps, TTriggerSchema>): Promise<void>;
     /**
      * Persists the workflow state to the database
      */
     persistWorkflowSnapshot(): Promise<void>;
     getState(): Promise<WorkflowRunState | null>;
-    resumeWithEvent(eventName: string, data: any): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
-    resume({ stepId, context: resumeContext }: {
+    resumeWithEvent(eventName: string, data: any, container?: Container): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
+    resume({ stepId, context: resumeContext, container, }: {
         stepId: string;
         context?: Record<string, any>;
+        container?: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
-    _resume({ stepId, context: resumeContext }: {
+    _resume({ stepId, context: resumeContext, container, }: {
         stepId: string;
         context?: Record<string, any>;
+        container: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, TResult>, "runId"> | undefined>;
 }
 
@@ -1082,10 +1092,11 @@ declare class Workflow<TSteps extends Step<string, any, any>[] = Step<string, an
     getWorkflowRuns(): Promise<WorkflowRuns>;
     getExecutionSpan(runId: string): Span | undefined;
     getState(runId: string): Promise<WorkflowRunState | null>;
-    resume({ runId, stepId, context: resumeContext, }: {
+    resume({ runId, stepId, context: resumeContext, container, }: {
         runId: string;
         stepId: string;
         context?: Record<string, any>;
+        container: Container;
     }): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, any>, "runId"> | undefined>;
     watch(onTransition: (state: Pick<WorkflowRunResult<TTriggerSchema, TSteps, TResultSchema>, 'results' | 'activePaths' | 'runId' | 'timestamp'>) => void): () => void;
     resumeWithEvent(runId: string, eventName: string, data: any): Promise<Omit<WorkflowRunResult<TTriggerSchema, TSteps, any>, "runId"> | undefined>;
@@ -1309,6 +1320,7 @@ type MastraCustomLLMOptions<Z extends ZodSchema | JSONSchema7 | undefined = unde
     telemetry?: TelemetrySettings;
     threadId?: string;
     resourceId?: string;
+    container: Container;
 } & Run;
 type LLMTextOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
     messages: CoreMessage[];
@@ -1568,6 +1580,8 @@ type AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefi
     };
     /** Telemetry settings */
     telemetry?: TelemetrySettings;
+    /** Container for dependency injection */
+    container?: Container;
 } & ({
     resourceId?: undefined;
     threadId?: undefined;
@@ -1609,6 +1623,8 @@ type AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefine
     experimental_output?: Z;
     /** Telemetry settings */
     telemetry?: TelemetrySettings;
+    /** Container for dependency injection */
+    container?: Container;
 } & ({
     resourceId?: undefined;
     threadId?: undefined;
@@ -1758,10 +1774,10 @@ declare class MastraLLMBase extends MastraBase {
     __registerMastra(p: Mastra): void;
     __text<Z extends ZodSchema | JSONSchema7 | undefined>(input: LLMTextOptions<Z>): Promise<GenerateTextResult<any, any>>;
     __textObject<T extends ZodSchema | JSONSchema7 | undefined>(input: LLMTextObjectOptions<T>): Promise<GenerateObjectResult<T>>;
-    generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options?: LLMStreamOptions<Z>): Promise<GenerateReturn<Z>>;
+    generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options: LLMStreamOptions<Z>): Promise<GenerateReturn<Z>>;
     __stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(input: LLMInnerStreamOptions<Z>): Promise<StreamTextResult<any, any>>;
     __streamObject<T extends ZodSchema | JSONSchema7 | undefined>(input: LLMStreamObjectOptions<T>): Promise<StreamObjectResult<DeepPartial<T>, T, never>>;
-    stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options?: LLMStreamOptions<Z>): Promise<StreamReturn<Z>>;
+    stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options: LLMStreamOptions<Z>): Promise<StreamReturn<Z>>;
 }
 
 declare class Agent<TAgentId extends string = string, TTools extends ToolsInput = ToolsInput, TMetrics extends Record<string, Metric> = Record<string, Metric>> extends MastraBase {
@@ -1807,11 +1823,12 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
     }>;
     private getResponseMessages;
     sanitizeResponseMessages(messages: Array<CoreMessage$1>): Array<CoreMessage$1>;
-    convertTools({ toolsets, threadId, resourceId, runId, }: {
+    convertTools({ toolsets, threadId, resourceId, runId, container, }: {
         toolsets?: ToolsetsInput;
         threadId?: string;
         resourceId?: string;
         runId?: string;
+        container: Container;
     }): Record<string, CoreTool>;
     preExecute({ resourceId, runId, threadId, thread, memoryConfig, messages, systemMessage, }: {
         runId?: string;
@@ -1825,7 +1842,7 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
         coreMessages: CoreMessage$1[];
         threadIdToUse: string;
     }>;
-    __primitive({ instructions, messages, context, threadId, memoryConfig, resourceId, runId, toolsets, }: {
+    __primitive({ instructions, messages, context, threadId, memoryConfig, resourceId, runId, toolsets, container, }: {
         instructions?: string;
         toolsets?: ToolsetsInput;
         resourceId?: string;
@@ -1834,6 +1851,7 @@ declare class Agent<TAgentId extends string = string, TTools extends ToolsInput
         context?: CoreMessage$1[];
         runId?: string;
         messages: CoreMessage$1[];
+        container: Container;
     }): {
         before: () => Promise<{
             messageObjects: CoreMessage$1[];

package/dist/chunk-3PBEAM6M.js

@@ -0,0 +1,69 @@
+// src/di/index.ts
+var Container = class {
+  registry = /* @__PURE__ */ new Map();
+  constructor(iterable) {
+    this.registry = new Map(iterable);
+  }
+  /**
+   * set a value with strict typing if `Values` is a Record and the key exists in it.
+   */
+  set(key, value) {
+    this.registry.set(key, value);
+  }
+  /**
+   * Get a value with its type
+   */
+  get(key) {
+    return this.registry.get(key);
+  }
+  /**
+   * Check if a key exists in the container
+   */
+  has(key) {
+    return this.registry.has(key);
+  }
+  /**
+   * Delete a value by key
+   */
+  delete(key) {
+    return this.registry.delete(key);
+  }
+  /**
+   * Clear all values from the container
+   */
+  clear() {
+    this.registry.clear();
+  }
+  /**
+   * Get all keys in the container
+   */
+  keys() {
+    return this.registry.keys();
+  }
+  /**
+   * Get all values in the container
+   */
+  values() {
+    return this.registry.values();
+  }
+  /**
+   * Get all entries in the container
+   */
+  entries() {
+    return this.registry.entries();
+  }
+  /**
+   * Get the size of the container
+   */
+  size() {
+    return this.registry.size;
+  }
+  /**
+   * Execute a function for each entry in the container
+   */
+  forEach(callbackfn) {
+    this.registry.forEach(callbackfn);
+  }
+};
+
+export { Container };

package/dist/{chunk-OSY64FYD.cjs → chunk-B23KFOTM.cjs}

@@ -2,7 +2,7 @@
 
 var chunkAAMFAGE5_cjs = require('./chunk-AAMFAGE5.cjs');
 var chunk2LIFCMY6_cjs = require('./chunk-2LIFCMY6.cjs');
-var chunk64VPB7ZD_cjs = require('./chunk-64VPB7ZD.cjs');
+var chunkPY3DCT4A_cjs = require('./chunk-PY3DCT4A.cjs');
 var chunkD63P5O4Q_cjs = require('./chunk-D63P5O4Q.cjs');
 var fs = require('fs');
 var path = require('path');
@@ -226,7 +226,7 @@ var MastraMemory = class extends chunkD63P5O4Q_cjs.MastraBase {
     return { indexName };
   }
   getMergedThreadConfig(config) {
-    return chunk64VPB7ZD_cjs.deepMerge(this.threadConfig, config || {});
+    return chunkPY3DCT4A_cjs.deepMerge(this.threadConfig, config || {});
   }
   /**
    * Apply all configured message processors to a list of messages.

package/dist/{chunk-4RMSGSQN.js → chunk-EO7KZJF6.js}

@@ -1,4 +1,4 @@
-import { createTool } from './chunk-BC5B4EGI.js';
+import { createTool } from './chunk-K5TZKHR7.js';
 import { z } from 'zod';
 
 // src/integration/integration.ts

package/dist/{chunk-7I7AKQH3.cjs → chunk-IPQW4HAT.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunk64VPB7ZD_cjs = require('./chunk-64VPB7ZD.cjs');
+var chunkPY3DCT4A_cjs = require('./chunk-PY3DCT4A.cjs');
 var zod = require('zod');
 
 // src/integration/integration.ts
@@ -68,7 +68,7 @@ var OpenAPIToolset = class {
     const tools = Object.keys(clientMethods).reduce((acc, key) => {
       const comment = documentations[key]?.comment;
       const fallbackComment = `Execute ${key}`;
-      const tool = chunk64VPB7ZD_cjs.createTool({
+      const tool = chunkPY3DCT4A_cjs.createTool({
         id: key,
         inputSchema: schemas[key] || zod.z.object({}),
         description: comment || fallbackComment,

package/dist/{chunk-BC5B4EGI.js → chunk-K5TZKHR7.js}

@@ -1,3 +1,4 @@
+import { Container } from './chunk-3PBEAM6M.js';
 import { createHash } from 'crypto';
 import { convertToCoreMessages } from 'ai';
 import jsonSchemaToZod from 'json-schema-to-zod';
@@ -221,7 +222,7 @@ function createLogMessageOptions({ agentName, toolName, tool, type }) {
   };
 }
 function createExecute(tool, options, logType) {
-  const { logger, mastra: _mastra, memory: _memory, ...rest } = options;
+  const { logger, mastra: _mastra, memory: _memory, container, ...rest } = options;
   const { start, error } = createLogMessageOptions({
     agentName: options.agentName,
     toolName: options.name,
@@ -239,7 +240,8 @@ function createExecute(tool, options, logType) {
         resourceId: options.resourceId,
         mastra: options.mastra,
         memory: options.memory,
-        runId: options.runId
+        runId: options.runId,
+        container: container ?? new Container()
       },
       execOptions
     ) ?? void 0;