@mastra/core 1.0.0-beta.8 → 1.0.0-beta.9

package/dist/{chunk-Y7YCQDML.cjs → chunk-3DRGAKUH.cjs}

@@ -1,12 +1,12 @@
 'use strict';
 
-var chunkSD2BDUJ6_cjs = require('./chunk-SD2BDUJ6.cjs');
-var chunkQYNZAPFE_cjs = require('./chunk-QYNZAPFE.cjs');
+var chunkDWB7SBVC_cjs = require('./chunk-DWB7SBVC.cjs');
+var chunkF77YTHXI_cjs = require('./chunk-F77YTHXI.cjs');
 var chunk4D4KB75Y_cjs = require('./chunk-4D4KB75Y.cjs');
-var chunkHP6TRJ2Q_cjs = require('./chunk-HP6TRJ2Q.cjs');
+var chunkXMT5T4F2_cjs = require('./chunk-XMT5T4F2.cjs');
 var chunk6BJ7XHRH_cjs = require('./chunk-6BJ7XHRH.cjs');
 var chunkMR7ZWBL6_cjs = require('./chunk-MR7ZWBL6.cjs');
-var chunk3EYBRHB2_cjs = require('./chunk-3EYBRHB2.cjs');
+var chunkUVM27J4A_cjs = require('./chunk-UVM27J4A.cjs');
 var chunkTWH4PTDG_cjs = require('./chunk-TWH4PTDG.cjs');
 var chunkUIGRFDO6_cjs = require('./chunk-UIGRFDO6.cjs');
 var chunkAF74UXR5_cjs = require('./chunk-AF74UXR5.cjs');
@@ -139,7 +139,7 @@ function createOnScorerHook(mastra) {
   };
 }
 async function validateAndSaveScore(storage, payload) {
-  const payloadToSave = chunkQYNZAPFE_cjs.saveScorePayloadSchema.parse(payload);
+  const payloadToSave = chunkF77YTHXI_cjs.saveScorePayloadSchema.parse(payload);
   await storage?.saveScore(payloadToSave);
 }
 async function findScorer(mastra, entityId, entityType, scorerId) {
@@ -193,6 +193,7 @@ var Mastra = class {
   #scorers;
   #tools;
   #processors;
+  #memory;
   #server;
   #serverAdapter;
   #mcpServers;
@@ -312,7 +313,7 @@ var Mastra = class {
         this.#events[topic] = config?.events?.[topic] ?? [];
       }
     }
-    const workflowEventProcessor = new chunkSD2BDUJ6_cjs.WorkflowEventProcessor({ mastra: this });
+    const workflowEventProcessor = new chunkDWB7SBVC_cjs.WorkflowEventProcessor({ mastra: this });
     const workflowEventCb = async (event, cb) => {
       try {
         await workflowEventProcessor.process(event, cb);
@@ -340,7 +341,7 @@ var Mastra = class {
     this.#idGenerator = config?.idGenerator;
     let storage = config?.storage;
     if (storage) {
-      storage = chunkHP6TRJ2Q_cjs.augmentWithInit(storage);
+      storage = chunkXMT5T4F2_cjs.augmentWithInit(storage);
     }
     if (config?.observability) {
       if (typeof config.observability.getDefaultInstance === "function") {
@@ -350,10 +351,10 @@ var Mastra = class {
         this.#logger?.warn(
           'Observability configuration error: Expected an Observability instance, but received a config object. Import and instantiate: import { Observability } from "@mastra/observability"; then pass: observability: new Observability({ default: { enabled: true } }). Observability has been disabled.'
         );
-        this.#observability = new chunk3EYBRHB2_cjs.NoOpObservability();
+        this.#observability = new chunkUVM27J4A_cjs.NoOpObservability();
       }
     } else {
-      this.#observability = new chunk3EYBRHB2_cjs.NoOpObservability();
+      this.#observability = new chunkUVM27J4A_cjs.NoOpObservability();
     }
     this.#storage = storage;
     this.#vectors = {};
@@ -363,6 +364,7 @@ var Mastra = class {
     this.#scorers = {};
     this.#tools = {};
     this.#processors = {};
+    this.#memory = {};
     this.#workflows = {};
     this.#gateways = {};
     if (config?.tools) {
@@ -379,6 +381,13 @@ var Mastra = class {
         }
       });
     }
+    if (config?.memory) {
+      Object.entries(config.memory).forEach(([key, memory]) => {
+        if (memory != null) {
+          this.addMemory(memory, key);
+        }
+      });
+    }
     if (config?.vectors) {
       Object.entries(config.vectors).forEach(([key, vector]) => {
         if (vector != null) {
@@ -549,6 +558,241 @@ var Mastra = class {
   listAgents() {
     return this.#agents;
   }
+  async getStoredAgentById(id, options) {
+    const storage = this.#storage;
+    if (!storage) {
+      const error = new chunkTWH4PTDG_cjs.MastraError({
+        id: "MASTRA_GET_STORED_AGENT_STORAGE_NOT_CONFIGURED",
+        domain: "MASTRA" /* MASTRA */,
+        category: "USER" /* USER */,
+        text: "Storage is not configured",
+        details: { status: 400 }
+      });
+      this.#logger?.trackException(error);
+      throw error;
+    }
+    if (!storage.supports.agents) {
+      const error = new chunkTWH4PTDG_cjs.MastraError({
+        id: "MASTRA_GET_STORED_AGENT_NOT_SUPPORTED",
+        domain: "MASTRA" /* MASTRA */,
+        category: "USER" /* USER */,
+        text: "Storage does not support agents",
+        details: { status: 400 }
+      });
+      this.#logger?.trackException(error);
+      throw error;
+    }
+    const storedAgent = await storage.getAgentById({ id });
+    if (!storedAgent) {
+      return null;
+    }
+    if (options?.raw) {
+      return storedAgent;
+    }
+    return this.#createAgentFromStoredConfig(storedAgent);
+  }
+  async listStoredAgents(args) {
+    const storage = this.#storage;
+    if (!storage) {
+      const error = new chunkTWH4PTDG_cjs.MastraError({
+        id: "MASTRA_LIST_STORED_AGENTS_STORAGE_NOT_CONFIGURED",
+        domain: "MASTRA" /* MASTRA */,
+        category: "USER" /* USER */,
+        text: "Storage is not configured",
+        details: { status: 400 }
+      });
+      this.#logger?.trackException(error);
+      throw error;
+    }
+    if (!storage.supports.agents) {
+      const error = new chunkTWH4PTDG_cjs.MastraError({
+        id: "MASTRA_LIST_STORED_AGENTS_NOT_SUPPORTED",
+        domain: "MASTRA" /* MASTRA */,
+        category: "USER" /* USER */,
+        text: "Storage does not support agents",
+        details: { status: 400 }
+      });
+      this.#logger?.trackException(error);
+      throw error;
+    }
+    const result = await storage.listAgents({
+      page: args?.page,
+      perPage: args?.perPage,
+      orderBy: args?.orderBy
+    });
+    if (args?.raw) {
+      return result;
+    }
+    const agents = result.agents.map((storedAgent) => this.#createAgentFromStoredConfig(storedAgent));
+    return {
+      agents,
+      total: result.total,
+      page: result.page,
+      perPage: result.perPage,
+      hasMore: result.hasMore
+    };
+  }
+  /**
+   * Creates an Agent instance from a stored agent configuration.
+   * @private
+   */
+  #createAgentFromStoredConfig(storedAgent) {
+    const modelConfig = storedAgent.model;
+    if (!modelConfig.provider || !modelConfig.name) {
+      throw new chunkTWH4PTDG_cjs.MastraError({
+        id: "MASTRA_STORED_AGENT_INVALID_MODEL",
+        domain: "MASTRA" /* MASTRA */,
+        category: "USER" /* USER */,
+        text: `Stored agent "${storedAgent.id}" has invalid model configuration. Both provider and name are required.`,
+        details: { agentId: storedAgent.id, model: JSON.stringify(storedAgent.model) }
+      });
+    }
+    const model = `${modelConfig.provider}/${modelConfig.name}`;
+    const tools = this.#resolveStoredTools(storedAgent.tools);
+    const workflows = this.#resolveStoredWorkflows(storedAgent.workflows);
+    const agents = this.#resolveStoredAgents(storedAgent.agents);
+    const memory = this.#resolveStoredMemory(storedAgent.memory);
+    const scorers = this.#resolveStoredScorers(storedAgent.scorers);
+    const agent = new chunkXMT5T4F2_cjs.Agent({
+      id: storedAgent.id,
+      name: storedAgent.name,
+      description: storedAgent.description,
+      instructions: storedAgent.instructions,
+      model,
+      tools,
+      workflows,
+      agents,
+      memory,
+      scorers,
+      defaultOptions: storedAgent.defaultOptions
+    });
+    agent.__setLogger(this.#logger);
+    agent.__registerMastra(this);
+    agent.__registerPrimitives({
+      logger: this.getLogger(),
+      storage: this.getStorage(),
+      agents: this.#agents,
+      tts: this.#tts,
+      vectors: this.#vectors
+    });
+    return agent;
+  }
+  /**
+   * Resolves tool references from stored configuration to actual tool instances.
+   * @private
+   */
+  #resolveStoredTools(storedTools) {
+    if (!storedTools || storedTools.length === 0) {
+      return {};
+    }
+    const resolvedTools = {};
+    const registeredTools = this.#tools;
+    for (const toolKey of storedTools) {
+      if (registeredTools && registeredTools[toolKey]) {
+        resolvedTools[toolKey] = registeredTools[toolKey];
+      } else {
+        this.#logger?.warn(`Tool "${toolKey}" referenced in stored agent but not registered in Mastra`);
+      }
+    }
+    return resolvedTools;
+  }
+  /**
+   * Resolves workflow references from stored configuration to actual workflow instances.
+   * @private
+   */
+  #resolveStoredWorkflows(storedWorkflows) {
+    if (!storedWorkflows || storedWorkflows.length === 0) {
+      return {};
+    }
+    const resolvedWorkflows = {};
+    for (const workflowKey of storedWorkflows) {
+      try {
+        const workflow = this.getWorkflow(workflowKey);
+        resolvedWorkflows[workflowKey] = workflow;
+      } catch {
+        try {
+          const workflow = this.getWorkflowById(workflowKey);
+          resolvedWorkflows[workflowKey] = workflow;
+        } catch {
+          this.#logger?.warn(`Workflow "${workflowKey}" referenced in stored agent but not registered in Mastra`);
+        }
+      }
+    }
+    return resolvedWorkflows;
+  }
+  /**
+   * Resolves agent references from stored configuration to actual agent instances.
+   * @private
+   */
+  #resolveStoredAgents(storedAgents) {
+    if (!storedAgents || storedAgents.length === 0) {
+      return {};
+    }
+    const resolvedAgents = {};
+    for (const agentKey of storedAgents) {
+      try {
+        const agent = this.getAgent(agentKey);
+        resolvedAgents[agentKey] = agent;
+      } catch {
+        try {
+          const agent = this.getAgentById(agentKey);
+          resolvedAgents[agentKey] = agent;
+        } catch {
+          this.#logger?.warn(`Agent "${agentKey}" referenced in stored agent but not registered in Mastra`);
+        }
+      }
+    }
+    return resolvedAgents;
+  }
+  /**
+   * Resolves memory reference from stored configuration to actual memory instance.
+   * @private
+   */
+  #resolveStoredMemory(storedMemory) {
+    if (!storedMemory) {
+      return void 0;
+    }
+    try {
+      return this.getMemory(storedMemory);
+    } catch {
+      try {
+        return this.getMemoryById(storedMemory);
+      } catch {
+        this.#logger?.warn(`Memory "${storedMemory}" referenced in stored agent but not registered in Mastra`);
+      }
+    }
+    return void 0;
+  }
+  /**
+   * Resolves scorer references from stored configuration to actual scorer instances.
+   * @private
+   */
+  #resolveStoredScorers(storedScorers) {
+    if (!storedScorers) {
+      return void 0;
+    }
+    const resolvedScorers = {};
+    for (const [scorerKey, scorerConfig] of Object.entries(storedScorers)) {
+      try {
+        const scorer = this.getScorer(scorerKey);
+        resolvedScorers[scorerKey] = {
+          scorer,
+          sampling: scorerConfig.sampling
+        };
+      } catch {
+        try {
+          const scorer = this.getScorerById(scorerKey);
+          resolvedScorers[scorerKey] = {
+            scorer,
+            sampling: scorerConfig.sampling
+          };
+        } catch {
+          this.#logger?.warn(`Scorer "${scorerKey}" referenced in stored agent but not registered in Mastra`);
+        }
+      }
+    }
+    return Object.keys(resolvedScorers).length > 0 ? resolvedScorers : void 0;
+  }
   /**
    * Adds a new agent to the Mastra instance.
    *
@@ -1376,6 +1620,131 @@ var Mastra = class {
     }
     processors[processorKey] = processor;
   }
+  /**
+   * Retrieves a registered memory instance by its registration key.
+   *
+   * @throws {MastraError} When the memory instance with the specified key is not found
+   *
+   * @example
+   * ```typescript
+   * const mastra = new Mastra({
+   *   memory: {
+   *     chat: new Memory({ storage })
+   *   }
+   * });
+   *
+   * const chatMemory = mastra.getMemory('chat');
+   * ```
+   */
+  getMemory(name) {
+    if (!this.#memory || !this.#memory[name]) {
+      const error = new chunkTWH4PTDG_cjs.MastraError({
+        id: "MASTRA_GET_MEMORY_BY_KEY_NOT_FOUND",
+        domain: "MASTRA" /* MASTRA */,
+        category: "USER" /* USER */,
+        text: `Memory with key ${String(name)} not found`,
+        details: {
+          status: 404,
+          memoryKey: String(name),
+          memory: Object.keys(this.#memory ?? {}).join(", ")
+        }
+      });
+      this.#logger?.trackException(error);
+      throw error;
+    }
+    return this.#memory[name];
+  }
+  /**
+   * Retrieves a registered memory instance by its ID.
+   *
+   * Searches through all registered memory instances and returns the one whose ID matches.
+   *
+   * @throws {MastraError} When no memory instance with the specified ID is found
+   *
+   * @example
+   * ```typescript
+   * const mastra = new Mastra({
+   *   memory: {
+   *     chat: new Memory({ id: 'chat-memory', storage })
+   *   }
+   * });
+   *
+   * const memory = mastra.getMemoryById('chat-memory');
+   * ```
+   */
+  getMemoryById(id) {
+    const allMemory = this.#memory;
+    if (allMemory) {
+      for (const [, memory] of Object.entries(allMemory)) {
+        if (memory.id === id) {
+          return memory;
+        }
+      }
+    }
+    const error = new chunkTWH4PTDG_cjs.MastraError({
+      id: "MASTRA_GET_MEMORY_BY_ID_NOT_FOUND",
+      domain: "MASTRA" /* MASTRA */,
+      category: "USER" /* USER */,
+      text: `Memory with id ${id} not found`,
+      details: {
+        status: 404,
+        memoryId: id,
+        availableIds: Object.values(allMemory ?? {}).map((m) => m.id).join(", ")
+      }
+    });
+    this.#logger?.trackException(error);
+    throw error;
+  }
+  /**
+   * Returns all registered memory instances as a record keyed by their names.
+   *
+   * @example
+   * ```typescript
+   * const mastra = new Mastra({
+   *   memory: {
+   *     chat: new Memory({ storage }),
+   *     longTerm: new Memory({ storage })
+   *   }
+   * });
+   *
+   * const allMemory = mastra.listMemory();
+   * console.log(Object.keys(allMemory)); // ['chat', 'longTerm']
+   * ```
+   */
+  listMemory() {
+    return this.#memory;
+  }
+  /**
+   * Adds a new memory instance to the Mastra instance.
+   *
+   * This method allows dynamic registration of memory instances after the Mastra instance
+   * has been created.
+   *
+   * @example
+   * ```typescript
+   * const mastra = new Mastra();
+   * const chatMemory = new Memory({
+   *   id: 'chat-memory',
+   *   storage: mastra.getStorage()
+   * });
+   * mastra.addMemory(chatMemory); // Uses memory.id as key
+   * // or
+   * mastra.addMemory(chatMemory, 'customKey'); // Uses custom key
+   * ```
+   */
+  addMemory(memory, key) {
+    if (!memory) {
+      throw createUndefinedPrimitiveError("memory", memory, key);
+    }
+    const memoryKey = key || memory.id;
+    const memoryRegistry = this.#memory;
+    if (memoryRegistry[memoryKey]) {
+      const logger = this.getLogger();
+      logger.debug(`Memory with key ${memoryKey} already exists. Skipping addition.`);
+      return;
+    }
+    memoryRegistry[memoryKey] = memory;
+  }
   /**
    * Returns all registered workflows as a record keyed by their IDs.
    *
@@ -1472,7 +1841,7 @@ var Mastra = class {
    * ```
    */
   setStorage(storage) {
-    this.#storage = chunkHP6TRJ2Q_cjs.augmentWithInit(storage);
+    this.#storage = chunkXMT5T4F2_cjs.augmentWithInit(storage);
   }
   setLogger({ logger }) {
     this.#logger = logger;
@@ -2134,7 +2503,7 @@ var Mastra = class {
       if (process.env.MASTRA_DEV !== "true" && process.env.MASTRA_DEV !== "1") {
         return;
       }
-      import('./provider-registry-V32PGH6F.cjs').then(async ({ GatewayRegistry }) => {
+      import('./provider-registry-Q6T32VC5.cjs').then(async ({ GatewayRegistry }) => {
         const registry = GatewayRegistry.getInstance();
         const customGateways = Object.values(this.#gateways || {});
         registry.registerCustomGateways(customGateways);
@@ -2188,5 +2557,5 @@ var Mastra = class {
 };
 
 exports.Mastra = Mastra;
-//# sourceMappingURL=chunk-Y7YCQDML.cjs.map
-//# sourceMappingURL=chunk-Y7YCQDML.cjs.map
+//# sourceMappingURL=chunk-3DRGAKUH.cjs.map
+//# sourceMappingURL=chunk-3DRGAKUH.cjs.map