@mastra/core 1.2.0-alpha.0 → 1.2.0

package/dist/{chunk-XVHK5IAO.cjs → chunk-3LRUJTH3.cjs}

@@ -1,9 +1,9 @@
 'use strict';
 
-var chunkOAOMKZXU_cjs = require('./chunk-OAOMKZXU.cjs');
-var chunkMCI7M6AT_cjs = require('./chunk-MCI7M6AT.cjs');
-var chunkODNSWEMV_cjs = require('./chunk-ODNSWEMV.cjs');
-var chunkW57QS6F6_cjs = require('./chunk-W57QS6F6.cjs');
+var chunkAP3Z4EAP_cjs = require('./chunk-AP3Z4EAP.cjs');
+var chunkYL6ETQ5I_cjs = require('./chunk-YL6ETQ5I.cjs');
+var chunkVBOO4F4D_cjs = require('./chunk-VBOO4F4D.cjs');
+var chunkARQG4ZSD_cjs = require('./chunk-ARQG4ZSD.cjs');
 var chunk6BJ7XHRH_cjs = require('./chunk-6BJ7XHRH.cjs');
 var chunkMR7ZWBL6_cjs = require('./chunk-MR7ZWBL6.cjs');
 var chunkCYUP7QWT_cjs = require('./chunk-CYUP7QWT.cjs');
@@ -52,6 +52,8 @@ var Mastra = class {
   #serverCache;
   // Cache for stored agents to allow in-memory modifications (like model changes) to persist across requests
   #storedAgentsCache = /* @__PURE__ */ new Map();
+  // Editor instance for handling agent instantiation and configuration
+  #editor;
   get pubsub() {
     return this.#pubsub;
   }
@@ -64,12 +66,34 @@ var Mastra = class {
    *   idGenerator: () => `custom-${Date.now()}`
    * });
    * const generator = mastra.getIdGenerator();
-   * console.log(generator?.()); // "custom-1234567890"
+   * console.log(generator?.()); // \"custom-1234567890\"
    * ```
    */
   getIdGenerator() {
     return this.#idGenerator;
   }
+  /**
+   * Gets the currently configured editor instance.
+   * The editor is responsible for handling agent instantiation and configuration.
+   *
+   * @example
+   * ```typescript
+   * const mastra = new Mastra({
+   *   editor: new MastraEditor({ logger })
+   * });
+   * const editor = mastra.getEditor();
+   * ```
+   */
+  getEditor() {
+    return this.#editor;
+  }
+  /**
+   * Gets the stored agents cache
+   * @internal
+   */
+  getStoredAgentCache() {
+    return this.#storedAgentsCache;
+  }
   /**
    * Generates a unique identifier using the configured generator or defaults to `crypto.randomUUID()`.
    *
@@ -160,10 +184,14 @@ var Mastra = class {
    */
   constructor(config) {
     this.#serverCache = new chunkQ2D7LERO_cjs.InMemoryServerCache();
+    this.#editor = config?.editor;
+    if (this.#editor && typeof this.#editor.registerWithMastra === "function") {
+      this.#editor.registerWithMastra(this);
+    }
     if (config?.pubsub) {
       this.#pubsub = config.pubsub;
     } else {
-      this.#pubsub = new chunkW57QS6F6_cjs.EventEmitterPubSub();
+      this.#pubsub = new chunkARQG4ZSD_cjs.EventEmitterPubSub();
     }
     this.#events = {};
     for (const topic in config?.events ?? {}) {
@@ -173,7 +201,7 @@ var Mastra = class {
         this.#events[topic] = config?.events?.[topic] ?? [];
       }
     }
-    const workflowEventProcessor = new chunkOAOMKZXU_cjs.WorkflowEventProcessor({ mastra: this });
+    const workflowEventProcessor = new chunkAP3Z4EAP_cjs.WorkflowEventProcessor({ mastra: this });
     const workflowEventCb = async (event, cb) => {
       try {
         await workflowEventProcessor.process(event, cb);
@@ -201,7 +229,7 @@ var Mastra = class {
     this.#idGenerator = config?.idGenerator;
     let storage = config?.storage;
     if (storage) {
-      storage = chunkW57QS6F6_cjs.augmentWithInit(storage);
+      storage = chunkARQG4ZSD_cjs.augmentWithInit(storage);
     }
     if (config?.observability) {
       if (typeof config.observability.getDefaultInstance === "function") {
@@ -303,7 +331,7 @@ var Mastra = class {
     if (config?.server) {
       this.#server = config.server;
     }
-    chunkMR7ZWBL6_cjs.registerHook("onScorerRun" /* ON_SCORER_RUN */, chunkODNSWEMV_cjs.createOnScorerHook(this));
+    chunkMR7ZWBL6_cjs.registerHook("onScorerRun" /* ON_SCORER_RUN */, chunkVBOO4F4D_cjs.createOnScorerHook(this));
     this.#observability.setMastraContext({ mastra: this });
     this.setLogger({ logger });
   }
@@ -421,323 +449,6 @@ var Mastra = class {
   listAgents() {
     return this.#agents;
   }
-  async getStoredAgentById(id, options) {
-    const storage = this.#storage;
-    if (!storage) {
-      const error = new chunk4U7ZLI36_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;
-    }
-    const agentsStore = await storage.getStore("agents");
-    if (!agentsStore) {
-      const error = new chunk4U7ZLI36_cjs.MastraError({
-        id: "MASTRA_GET_STORED_AGENT_NOT_SUPPORTED",
-        domain: "MASTRA" /* MASTRA */,
-        category: "USER" /* USER */,
-        text: "Agents storage is not available",
-        details: { status: 400 }
-      });
-      this.#logger?.trackException(error);
-      throw error;
-    }
-    if (options?.versionId && options?.versionNumber !== void 0) {
-      this.#logger?.warn(`Both versionId and versionNumber provided for agent "${id}". Using versionId.`);
-    }
-    if (options?.versionId) {
-      const version = await agentsStore.getVersion(options.versionId);
-      if (!version) {
-        return null;
-      }
-      if (version.agentId !== id) {
-        return null;
-      }
-      const {
-        id: _versionId,
-        agentId: _agentId,
-        versionNumber: _versionNumber,
-        changedFields: _changedFields,
-        changeMessage: _changeMessage,
-        createdAt: _createdAt,
-        ...snapshotConfig
-      } = version;
-      const agentRecord = await agentsStore.getAgentById({ id });
-      if (!agentRecord) {
-        return null;
-      }
-      const resolvedAgent = { ...agentRecord, ...snapshotConfig };
-      if (options?.raw) {
-        return resolvedAgent;
-      }
-      return this.#createAgentFromStoredConfig(resolvedAgent);
-    }
-    if (options?.versionNumber !== void 0) {
-      const version = await agentsStore.getVersionByNumber(id, options.versionNumber);
-      if (!version) {
-        return null;
-      }
-      const {
-        id: _versionId,
-        agentId: _agentId,
-        versionNumber: _versionNumber,
-        changedFields: _changedFields,
-        changeMessage: _changeMessage,
-        createdAt: _createdAt,
-        ...snapshotConfig
-      } = version;
-      const agentRecord = await agentsStore.getAgentById({ id });
-      if (!agentRecord) {
-        return null;
-      }
-      const resolvedAgent = { ...agentRecord, ...snapshotConfig };
-      if (options?.raw) {
-        return resolvedAgent;
-      }
-      return this.#createAgentFromStoredConfig(resolvedAgent);
-    }
-    if (!options?.raw) {
-      const cachedAgent = this.#storedAgentsCache.get(id);
-      if (cachedAgent) {
-        this.#logger?.debug(`[getStoredAgentById] Returning cached agent "${id}"`);
-        return cachedAgent;
-      }
-      this.#logger?.debug(`[getStoredAgentById] Cache miss for agent "${id}", fetching from storage`);
-    }
-    const storedAgent = await agentsStore.getAgentByIdResolved({ id });
-    if (!storedAgent) {
-      return null;
-    }
-    if (options?.raw) {
-      return storedAgent;
-    }
-    const agent = this.#createAgentFromStoredConfig(storedAgent);
-    this.#storedAgentsCache.set(id, agent);
-    return agent;
-  }
-  /**
-   * Clears the cached stored agent instance, forcing a fresh load from storage on next access.
-   * This should be called when an agent is updated via the Edit dialog to ensure
-   * the cached instance is refreshed with the new configuration.
-   *
-   * @param id - The ID of the agent to clear from cache
-   */
-  clearStoredAgentCache(id) {
-    this.#storedAgentsCache.delete(id);
-  }
-  async listStoredAgents(args) {
-    const storage = this.#storage;
-    if (!storage) {
-      const error = new chunk4U7ZLI36_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;
-    }
-    const agentsStore = await storage.getStore("agents");
-    if (!agentsStore) {
-      const error = new chunk4U7ZLI36_cjs.MastraError({
-        id: "MASTRA_LIST_STORED_AGENTS_NOT_SUPPORTED",
-        domain: "MASTRA" /* MASTRA */,
-        category: "USER" /* USER */,
-        text: "Agents storage is not available",
-        details: { status: 400 }
-      });
-      this.#logger?.trackException(error);
-      throw error;
-    }
-    const result = await agentsStore.listAgentsResolved({
-      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 chunk4U7ZLI36_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 chunkW57QS6F6_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
-    });
-    agent.source = "stored";
-    return agent;
-  }
-  /**
-   * Resolves tool references from stored configuration to actual tool instances.
-   * @private
-   */
-  #resolveStoredTools(storedTools) {
-    if (!storedTools || storedTools.length === 0) {
-      return {};
-    }
-    const resolvedTools = {};
-    for (const toolKey of storedTools) {
-      try {
-        const tool = this.getToolById(toolKey);
-        resolvedTools[toolKey] = tool;
-      } catch {
-        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(memoryConfig) {
-    if (!memoryConfig) {
-      return void 0;
-    }
-    const memoryKey = memoryConfig.key;
-    if (!memoryKey) {
-      this.#logger?.warn(`Stored agent memory config missing "key" field: ${JSON.stringify(memoryConfig)}`);
-      return void 0;
-    }
-    try {
-      return this.getMemory(memoryKey);
-    } catch {
-      try {
-        return this.getMemoryById(memoryKey);
-      } catch {
-        this.#logger?.warn(`Memory "${memoryKey}" 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 || Object.keys(storedScorers).length === 0) {
-      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.
    *
@@ -759,13 +470,13 @@ var Mastra = class {
    * mastra.addAgent(newAgent, 'customKey'); // Uses custom key
    * ```
    */
-  addAgent(agent, key) {
+  addAgent(agent, key, options) {
     if (!agent) {
       throw createUndefinedPrimitiveError("agent", agent, key);
     }
     let mastraAgent;
-    if (chunkMCI7M6AT_cjs.isToolLoopAgentLike(agent)) {
-      mastraAgent = chunkMCI7M6AT_cjs.toolLoopAgentToMastraAgent(agent, { fallbackName: key });
+    if (chunkYL6ETQ5I_cjs.isToolLoopAgentLike(agent)) {
+      mastraAgent = chunkYL6ETQ5I_cjs.toolLoopAgentToMastraAgent(agent, { fallbackName: key });
     } else {
       mastraAgent = agent;
     }
@@ -785,6 +496,9 @@ var Mastra = class {
       tts: this.#tts,
       vectors: this.#vectors
     });
+    if (options?.source) {
+      mastraAgent.source = options.source;
+    }
     agents[agentKey] = mastraAgent;
     mastraAgent.getConfiguredProcessorWorkflows().then((processorWorkflows) => {
       for (const workflow of processorWorkflows) {
@@ -1900,7 +1614,7 @@ var Mastra = class {
    * ```
    */
   setStorage(storage) {
-    this.#storage = chunkW57QS6F6_cjs.augmentWithInit(storage);
+    this.#storage = chunkARQG4ZSD_cjs.augmentWithInit(storage);
   }
   setLogger({ logger }) {
     this.#logger = logger;
@@ -1933,6 +1647,9 @@ var Mastra = class {
     if (this.#serverAdapter) {
       this.#serverAdapter.__setLogger(this.#logger);
     }
+    if (this.#workspace) {
+      this.#workspace.__setLogger(this.#logger);
+    }
     this.#observability.setLogger({ logger: this.#logger });
   }
   /**
@@ -2562,7 +2279,7 @@ var Mastra = class {
       if (process.env.MASTRA_DEV !== "true" && process.env.MASTRA_DEV !== "1") {
         return;
       }
-      import('./provider-registry-ML2VWWLB.cjs').then(async ({ GatewayRegistry }) => {
+      import('./provider-registry-NR7FXV2Q.cjs').then(async ({ GatewayRegistry }) => {
         const registry = GatewayRegistry.getInstance();
         const customGateways = Object.values(this.#gateways || {});
         registry.registerCustomGateways(customGateways);
@@ -2616,5 +2333,5 @@ var Mastra = class {
 };
 
 exports.Mastra = Mastra;
-//# sourceMappingURL=chunk-XVHK5IAO.cjs.map
-//# sourceMappingURL=chunk-XVHK5IAO.cjs.map
+//# sourceMappingURL=chunk-3LRUJTH3.cjs.map
+//# sourceMappingURL=chunk-3LRUJTH3.cjs.map