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

package/dist/{chunk-SPIUKQDK.js → chunk-LB7UOP2X.js}

@@ -1,10 +1,10 @@
-import { WorkflowEventProcessor } from './chunk-7PXCGGMO.js';
-import { saveScorePayloadSchema } from './chunk-RWT3XPKO.js';
+import { WorkflowEventProcessor } from './chunk-QTEIZ2XG.js';
+import { saveScorePayloadSchema } from './chunk-4GLDYGNN.js';
 import { PubSub } from './chunk-BVUMKER5.js';
-import { augmentWithInit } from './chunk-EDNHZZUP.js';
+import { augmentWithInit, Agent } from './chunk-ARSSSQQP.js';
 import { noopLogger } from './chunk-PSCMWPLC.js';
 import { registerHook } from './chunk-L54GIUCB.js';
-import { NoOpObservability } from './chunk-FNSFXWDN.js';
+import { NoOpObservability } from './chunk-PIHOAJTA.js';
 import { MastraError } from './chunk-JJ5O45LH.js';
 import { InMemoryServerCache } from './chunk-3CKZSDTQ.js';
 import { ConsoleLogger, LogLevel } from './chunk-NRUZYMHE.js';
@@ -186,6 +186,7 @@ var Mastra = class {
   #scorers;
   #tools;
   #processors;
+  #memory;
   #server;
   #serverAdapter;
   #mcpServers;
@@ -356,6 +357,7 @@ var Mastra = class {
     this.#scorers = {};
     this.#tools = {};
     this.#processors = {};
+    this.#memory = {};
     this.#workflows = {};
     this.#gateways = {};
     if (config?.tools) {
@@ -372,6 +374,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) {
@@ -542,6 +551,241 @@ var Mastra = class {
   listAgents() {
     return this.#agents;
   }
+  async getStoredAgentById(id, options) {
+    const storage = this.#storage;
+    if (!storage) {
+      const error = new 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 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 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 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 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 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.
    *
@@ -1369,6 +1613,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 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 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.
    *
@@ -2127,7 +2496,7 @@ var Mastra = class {
       if (process.env.MASTRA_DEV !== "true" && process.env.MASTRA_DEV !== "1") {
         return;
       }
-      import('./provider-registry-A5QZFI2X.js').then(async ({ GatewayRegistry }) => {
+      import('./provider-registry-FM3IVY6C.js').then(async ({ GatewayRegistry }) => {
         const registry = GatewayRegistry.getInstance();
         const customGateways = Object.values(this.#gateways || {});
         registry.registerCustomGateways(customGateways);
@@ -2181,5 +2550,5 @@ var Mastra = class {
 };
 
 export { Mastra };
-//# sourceMappingURL=chunk-SPIUKQDK.js.map
-//# sourceMappingURL=chunk-SPIUKQDK.js.map
+//# sourceMappingURL=chunk-LB7UOP2X.js.map
+//# sourceMappingURL=chunk-LB7UOP2X.js.map