@mastra/core 0.9.4 → 0.10.0-alpha.1

package/dist/{chunk-FW5RS6WV.js → chunk-E7CLY6SK.js}

@@ -1,11 +1,11 @@
-import { DefaultVoice } from './chunk-NOPY74PV.js';
-import { MastraLLM } from './chunk-WIKOYZ63.js';
+import { DefaultVoice } from './chunk-KJQFFEEQ.js';
+import { executeHook } from './chunk-BB4KXGBU.js';
+import { MastraLLM } from './chunk-BPTSLJHA.js';
 import { InstrumentClass } from './chunk-BFNYQFG2.js';
-import { ensureToolProperties, ensureAllMessagesAreCoreMessages, makeCoreTool, createMastraProxy } from './chunk-S34QVMKE.js';
+import { ensureToolProperties, ensureAllMessagesAreCoreMessages, makeCoreTool, createMastraProxy } from './chunk-2PW6UJMW.js';
+import { MastraBase } from './chunk-FI7R232B.js';
+import { RegisteredLogger } from './chunk-VHLL4AZK.js';
 import { RuntimeContext } from './chunk-SGGPJWRQ.js';
-import { executeHook } from './chunk-BB4KXGBU.js';
-import { MastraBase } from './chunk-235X76GC.js';
-import { RegisteredLogger } from './chunk-CMPPXW4U.js';
 import { __decoratorStart, __decorateElement, __runInitializers } from './chunk-C6A6W6XS.js';
 import { context, trace } from '@opentelemetry/api';
 import { z } from 'zod';
@@ -15,8 +15,8 @@ import EventEmitter from 'node:events';
 import sift from 'sift';
 import { createActor, assign, fromPromise, setup } from 'xstate';
 
-// src/workflows/step.ts
-var Step = class {
+// src/workflows/legacy/step.ts
+var LegacyStep = class {
   id;
   description;
   inputSchema;
@@ -44,7 +44,7 @@ var Step = class {
   }
 };
 
-// src/workflows/types.ts
+// src/workflows/legacy/types.ts
 var WhenConditionReturnValue = /* @__PURE__ */(WhenConditionReturnValue2 => {
   WhenConditionReturnValue2["CONTINUE"] = "continue";
   WhenConditionReturnValue2["CONTINUE_FAILED"] = "continue_failed";
@@ -131,7 +131,14 @@ var Agent = class extends (_a = MastraBase) {
     return Boolean(this.#memory);
   }
   getMemory() {
-    return this.#memory ?? this.#mastra?.memory;
+    const memory = this.#memory;
+    if (memory && !memory.hasOwnStorage && this.#mastra) {
+      const storage = this.#mastra.getStorage();
+      if (storage) {
+        memory.setStorage(storage);
+      }
+    }
+    return memory;
   }
   get voice() {
     if (typeof this.#instructions === "function") {
@@ -1136,7 +1143,7 @@ var Agent = class extends (_a = MastraBase) {
         messages: messageObjects,
         tools: convertedTools,
         onStepFinish: result3 => {
-          void onStepFinish?.(result3);
+          return onStepFinish?.(result3);
         },
         maxSteps,
         runId: runIdToUse,
@@ -1168,7 +1175,7 @@ var Agent = class extends (_a = MastraBase) {
         messages: messageObjects,
         tools: convertedTools,
         onStepFinish: result3 => {
-          void onStepFinish?.(result3);
+          return onStepFinish?.(result3);
         },
         maxSteps,
         runId: runIdToUse,
@@ -1198,7 +1205,7 @@ var Agent = class extends (_a = MastraBase) {
       tools: convertedTools,
       structuredOutput: output,
       onStepFinish: result2 => {
-        void onStepFinish?.(result2);
+        return onStepFinish?.(result2);
       },
       maxSteps,
       runId: runIdToUse,
@@ -1296,7 +1303,7 @@ var Agent = class extends (_a = MastraBase) {
         temperature,
         tools: convertedTools,
         onStepFinish: result => {
-          void onStepFinish?.(result);
+          return onStepFinish?.(result);
         },
         onFinish: async result => {
           try {
@@ -1316,7 +1323,7 @@ var Agent = class extends (_a = MastraBase) {
               runId
             });
           }
-          void onFinish?.(result);
+          await onFinish?.(result);
         },
         maxSteps,
         runId: runIdToUse,
@@ -1338,7 +1345,7 @@ var Agent = class extends (_a = MastraBase) {
         temperature,
         tools: convertedTools,
         onStepFinish: result => {
-          void onStepFinish?.(result);
+          return onStepFinish?.(result);
         },
         onFinish: async result => {
           try {
@@ -1358,7 +1365,7 @@ var Agent = class extends (_a = MastraBase) {
               runId
             });
           }
-          void onFinish?.(result);
+          await onFinish?.(result);
         },
         maxSteps,
         runId: runIdToUse,
@@ -1378,7 +1385,7 @@ var Agent = class extends (_a = MastraBase) {
       temperature,
       structuredOutput: output,
       onStepFinish: result => {
-        void onStepFinish?.(result);
+        return onStepFinish?.(result);
       },
       onFinish: async result => {
         try {
@@ -1398,7 +1405,7 @@ var Agent = class extends (_a = MastraBase) {
             runId
           });
         }
-        void onFinish?.(result);
+        await onFinish?.(result);
       },
       runId: runIdToUse,
       toolChoice,
@@ -1472,7 +1479,7 @@ var Agent = class extends (_a = MastraBase) {
   }
   toStep() {
     const x = agentToStep(this);
-    return new Step(x);
+    return new LegacyStep(x);
   }
 };
 Agent = /*@__PURE__*/(_ => {
@@ -1480,10 +1487,10 @@ Agent = /*@__PURE__*/(_ => {
   Agent = __decorateElement(_init, 0, "Agent", _Agent_decorators, Agent);
   __runInitializers(_init, 1, Agent);
 
-  // src/workflows/utils.ts
+  // src/workflows/legacy/utils.ts
   return Agent;
 })();
-// src/workflows/utils.ts
+// src/workflows/legacy/utils.ts
 function isErrorEvent(stateEvent) {
   return stateEvent.type.startsWith("xstate.error.actor.");
 }
@@ -1602,7 +1609,7 @@ function getResultActivePaths(state) {
   return activePathsAndStatus;
 }
 function isWorkflow(step) {
-  return step instanceof Workflow;
+  return step instanceof LegacyWorkflow;
 }
 function isAgent(step) {
   return step instanceof Agent;
@@ -1716,7 +1723,7 @@ function workflowToStep(workflow, {
       });
       unwatch();
       if (!awaitedResult) {
-        throw new Error("Workflow run failed");
+        throw new Error("LegacyWorkflow run failed");
       }
       if (awaitedResult.activePaths?.size > 0) {
         const suspendedStep = [...awaitedResult.activePaths.entries()].find(([, {
@@ -2868,7 +2875,7 @@ var Machine = class extends EventEmitter {
   }
 };
 
-// src/workflows/workflow-instance.ts
+// src/workflows/legacy/workflow-instance.ts
 var WorkflowInstance = class {
   name;
   #mastra;
@@ -3452,8 +3459,8 @@ var WorkflowInstance = class {
   }
 };
 
-// src/workflows/workflow.ts
-var Workflow = class extends MastraBase {
+// src/workflows/legacy/workflow.ts
+var LegacyWorkflow = class extends MastraBase {
   name;
   triggerSchema;
   resultSchema;
@@ -3480,7 +3487,7 @@ var Workflow = class extends MastraBase {
   #steps = {};
   #ifCount = 0;
   /**
-   * Creates a new Workflow instance
+   * Creates a new LegacyWorkflow instance
    * @param name - Identifier for the workflow (not necessarily unique)
    * @param logger - Optional logger instance
    */
@@ -3525,7 +3532,7 @@ var Workflow = class extends MastraBase {
       });
       nextSteps.forEach(step2 => this.step(step2, config));
       this.after(nextSteps);
-      this.step(new Step({
+      this.step(new LegacyStep({
         id: `__after_${next.map(step2 => config?.id ?? step2?.id ?? step2?.name).join("_")}`,
         execute: async () => {
           return {
@@ -3604,7 +3611,7 @@ var Workflow = class extends MastraBase {
       });
       nextSteps.forEach(step2 => this.#__internalStep(step2, config, internalUse));
       this.after(nextSteps);
-      this.#__internalStep(new Step({
+      this.#__internalStep(new LegacyStep({
         id: `__after_${next.map(step2 => step2?.id ?? step2?.name).join("_")}`,
         execute: async () => {
           return {
@@ -3688,7 +3695,7 @@ var Workflow = class extends MastraBase {
         return step2;
       });
       nextSteps.forEach(step2 => this.step(step2, config));
-      this.step(new Step({
+      this.step(new LegacyStep({
         // @ts-ignore
         id: `__after_${next.map(step2 => step2?.id ?? step2?.name).join("_")}`,
         execute: async () => {
@@ -4002,7 +4009,7 @@ var Workflow = class extends MastraBase {
       } else {
         this.after(_ifStep);
       }
-      this.step(new Step({
+      this.step(new LegacyStep({
         id: `${lastStep.id}_if_else`,
         execute: async () => {
           return {
@@ -4083,7 +4090,7 @@ var Workflow = class extends MastraBase {
       throw new Error("Condition requires a step to be executed after");
     }
     const eventStepKey = `__${eventName}_event`;
-    const eventStep = new Step({
+    const eventStep = new LegacyStep({
       id: eventStepKey,
       execute: async ({
         context,
@@ -4429,8 +4436,8 @@ var Workflow = class extends MastraBase {
     const x = workflowToStep(this, {
       mastra: this.#mastra
     });
-    return new Step(x);
+    return new LegacyStep(x);
   }
 };
 
-export { Agent, Step, WhenConditionReturnValue, Workflow, agentToStep, getActivePathsAndStatus, getResultActivePaths, getStepResult, getSuspendedPaths, isAgent, isConditionalKey, isErrorEvent, isFinalState, isLimboState, isTransitionEvent, isVariableReference, isWorkflow, mergeChildValue, recursivelyCheckForFinalState, resolveVariables, updateStepInHierarchy, workflowToStep };
+export { Agent, LegacyStep, LegacyWorkflow, WhenConditionReturnValue, agentToStep, getActivePathsAndStatus, getResultActivePaths, getStepResult, getSuspendedPaths, isAgent, isConditionalKey, isErrorEvent, isFinalState, isLimboState, isTransitionEvent, isVariableReference, isWorkflow, mergeChildValue, recursivelyCheckForFinalState, resolveVariables, updateStepInHierarchy, workflowToStep };

package/dist/{chunk-235X76GC.js → chunk-FI7R232B.js}

@@ -1,4 +1,4 @@
-import { RegisteredLogger, createLogger } from './chunk-CMPPXW4U.js';
+import { RegisteredLogger, ConsoleLogger } from './chunk-VHLL4AZK.js';
 
 // src/base.ts
 var MastraBase = class {
@@ -9,7 +9,7 @@ var MastraBase = class {
   constructor({ component, name }) {
     this.component = component || RegisteredLogger.LLM;
     this.name = name;
-    this.logger = createLogger({ name: `${this.component} - ${this.name}` });
+    this.logger = new ConsoleLogger({ name: `${this.component} - ${this.name}` });
   }
   /**
    * Set the logger for the agent

package/dist/chunk-GQ2XQ4UN.js

@@ -0,0 +1,321 @@
+import { augmentWithInit } from './chunk-75Z4BZEE.js';
+import { deepMerge } from './chunk-2PW6UJMW.js';
+import { MastraBase } from './chunk-FI7R232B.js';
+
+// src/memory/memory.ts
+var MemoryProcessor = class extends MastraBase {
+  /**
+   * Process a list of messages and return a filtered or transformed list.
+   * @param messages The messages to process
+   * @returns The processed messages
+   */
+  process(messages, _opts) {
+    return messages;
+  }
+};
+var memoryDefaultOptions = {
+  lastMessages: 10,
+  semanticRecall: false,
+  threads: {
+    generateTitle: false
+  },
+  workingMemory: {
+    enabled: false,
+    template: `
+# User Information
+- **First Name**: 
+- **Last Name**: 
+- **Location**: 
+- **Occupation**: 
+- **Interests**: 
+- **Goals**: 
+- **Events**: 
+- **Facts**: 
+- **Projects**: 
+`
+  }
+};
+var MastraMemory = class extends MastraBase {
+  MAX_CONTEXT_TOKENS;
+  _storage;
+  vector;
+  embedder;
+  processors = [];
+  threadConfig = { ...memoryDefaultOptions };
+  constructor(config) {
+    super({ component: "MEMORY", name: config.name });
+    if (config.options) this.threadConfig = this.getMergedThreadConfig(config.options);
+    if (config.processors) this.processors = config.processors;
+    if (config.storage) {
+      this._storage = augmentWithInit(config.storage);
+      this._hasOwnStorage = true;
+    }
+    if (this.threadConfig.semanticRecall) {
+      if (!config.vector) {
+        throw new Error(
+          `Semantic recall requires a vector store to be configured.
+
+https://mastra.ai/en/docs/memory/semantic-recall`
+        );
+      }
+      this.vector = config.vector;
+      if (!config.embedder) {
+        throw new Error(
+          `Semantic recall requires an embedder to be configured.
+
+https://mastra.ai/en/docs/memory/semantic-recall`
+        );
+      }
+      this.embedder = config.embedder;
+    }
+  }
+  _hasOwnStorage = false;
+  get hasOwnStorage() {
+    return this._hasOwnStorage;
+  }
+  get storage() {
+    if (!this._storage) {
+      throw new Error(
+        `Memory requires a storage provider to function. Add a storage configuration to Memory or to your Mastra instance.
+
+https://mastra.ai/en/docs/memory/overview`
+      );
+    }
+    return this._storage;
+  }
+  setStorage(storage) {
+    this._storage = storage;
+  }
+  setVector(vector) {
+    this.vector = vector;
+  }
+  setEmbedder(embedder) {
+    this.embedder = embedder;
+  }
+  /**
+   * 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.
+   */
+  async getSystemMessage(_input) {
+    return null;
+  }
+  /**
+   * Get tools that should be available to the agent.
+   * This will be called when converting tools for the agent.
+   * Implementations can override this to provide additional tools.
+   */
+  getTools(_config) {
+    return {};
+  }
+  async createEmbeddingIndex(dimensions) {
+    const defaultDimensions = 1536;
+    const isDefault = dimensions === defaultDimensions;
+    const usedDimensions = dimensions ?? defaultDimensions;
+    const separator = this.vector?.indexSeparator ?? "_";
+    const indexName = isDefault ? `memory${separator}messages` : `memory${separator}messages${separator}${usedDimensions}`;
+    if (typeof this.vector === `undefined`) {
+      throw new Error(`Tried to create embedding index but no vector db is attached to this Memory instance.`);
+    }
+    await this.vector.createIndex({
+      indexName,
+      dimension: usedDimensions
+    });
+    return { indexName };
+  }
+  getMergedThreadConfig(config) {
+    if (config?.workingMemory && "use" in config.workingMemory) {
+      throw new Error("The workingMemory.use option has been removed. Working memory always uses tool-call mode.");
+    }
+    return deepMerge(this.threadConfig, config || {});
+  }
+  /**
+   * Apply all configured message processors to a list of messages.
+   * @param messages The messages to process
+   * @returns The processed messages
+   */
+  applyProcessors(messages, opts) {
+    const processors = opts.processors || this.processors;
+    if (!processors || processors.length === 0) {
+      return messages;
+    }
+    let processedMessages = [...messages];
+    for (const processor of processors) {
+      processedMessages = processor.process(processedMessages, {
+        systemMessage: opts.systemMessage,
+        newMessages: opts.newMessages,
+        memorySystemMessage: opts.memorySystemMessage
+      });
+    }
+    return processedMessages;
+  }
+  processMessages({
+    messages,
+    processors,
+    ...opts
+  }) {
+    return this.applyProcessors(messages, { processors: processors || this.processors, ...opts });
+  }
+  estimateTokens(text) {
+    return Math.ceil(text.split(" ").length * 1.3);
+  }
+  parseMessages(messages) {
+    return messages.map((msg) => {
+      let content = msg.content;
+      if (typeof content === "string" && (content.startsWith("[") || content.startsWith("{"))) {
+        try {
+          content = JSON.parse(content);
+        } catch {
+        }
+      } else if (typeof content === "number") {
+        content = String(content);
+      }
+      return {
+        ...msg,
+        content
+      };
+    });
+  }
+  convertToUIMessages(messages) {
+    function addToolMessageToChat({
+      toolMessage,
+      messages: messages2,
+      toolResultContents
+    }) {
+      const chatMessages2 = messages2.map((message) => {
+        if (message.toolInvocations) {
+          return {
+            ...message,
+            toolInvocations: message.toolInvocations.map((toolInvocation) => {
+              const toolResult = toolMessage.content.find((tool) => tool.toolCallId === toolInvocation.toolCallId);
+              if (toolResult) {
+                return {
+                  ...toolInvocation,
+                  state: "result",
+                  result: toolResult.result
+                };
+              }
+              return toolInvocation;
+            })
+          };
+        }
+        return message;
+      });
+      const resultContents = [...toolResultContents, ...toolMessage.content];
+      return { chatMessages: chatMessages2, toolResultContents: resultContents };
+    }
+    const { chatMessages } = messages.reduce(
+      (obj, message) => {
+        if (message.role === "tool") {
+          return addToolMessageToChat({
+            toolMessage: message,
+            messages: obj.chatMessages,
+            toolResultContents: obj.toolResultContents
+          });
+        }
+        let textContent = "";
+        let toolInvocations = [];
+        if (typeof message.content === "string") {
+          textContent = message.content;
+        } else if (typeof message.content === "number") {
+          textContent = String(message.content);
+        } else if (Array.isArray(message.content)) {
+          for (const content of message.content) {
+            if (content.type === "text") {
+              textContent += content.text;
+            } else if (content.type === "tool-call") {
+              const toolResult = obj.toolResultContents.find((tool) => tool.toolCallId === content.toolCallId);
+              toolInvocations.push({
+                state: toolResult ? "result" : "call",
+                toolCallId: content.toolCallId,
+                toolName: content.toolName,
+                args: content.args,
+                result: toolResult?.result
+              });
+            }
+          }
+        }
+        obj.chatMessages.push({
+          id: message.id,
+          role: message.role,
+          content: textContent,
+          toolInvocations,
+          createdAt: message.createdAt
+        });
+        return obj;
+      },
+      { chatMessages: [], toolResultContents: [] }
+    );
+    return chatMessages;
+  }
+  /**
+   * 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
+   */
+  async createThread({
+    threadId,
+    resourceId,
+    title,
+    metadata,
+    memoryConfig
+  }) {
+    await this.storage.init();
+    const thread = {
+      id: threadId || this.generateId(),
+      title: title || `New Thread ${(/* @__PURE__ */ new Date()).toISOString()}`,
+      resourceId,
+      createdAt: /* @__PURE__ */ new Date(),
+      updatedAt: /* @__PURE__ */ new Date(),
+      metadata
+    };
+    return this.saveThread({ thread, memoryConfig });
+  }
+  /**
+   * 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
+   */
+  async addMessage({
+    threadId,
+    resourceId,
+    config,
+    content,
+    role,
+    type,
+    toolNames,
+    toolCallArgs,
+    toolCallIds
+  }) {
+    const message = {
+      id: this.generateId(),
+      content,
+      role,
+      createdAt: /* @__PURE__ */ new Date(),
+      threadId,
+      resourceId,
+      type,
+      toolNames,
+      toolCallArgs,
+      toolCallIds
+    };
+    const savedMessages = await this.saveMessages({ messages: [message], memoryConfig: config });
+    return savedMessages[0];
+  }
+  /**
+   * Generates a unique identifier
+   * @returns A unique string ID
+   */
+  generateId() {
+    return crypto.randomUUID();
+  }
+};
+
+export { MastraMemory, MemoryProcessor, memoryDefaultOptions };

package/dist/{chunk-H4CEZSRM.js → chunk-J7GDGV6F.js}

@@ -1,4 +1,4 @@
-import { Agent } from './chunk-FW5RS6WV.js';
+import { Agent } from './chunk-E7CLY6SK.js';
 import { CohereClient } from 'cohere-ai';
 
 var CohereRelevanceScorer = class {

package/dist/{chunk-NLBJOFNG.js → chunk-JW5TMK2L.js}

@@ -1,4 +1,4 @@
-import { MastraBase } from './chunk-235X76GC.js';
+import { MastraBase } from './chunk-FI7R232B.js';
 import { readFile } from 'fs/promises';
 import { parse } from 'dotenv';
 

package/dist/{chunk-NOPY74PV.js → chunk-KJQFFEEQ.js}

@@ -1,5 +1,5 @@
 import { InstrumentClass } from './chunk-BFNYQFG2.js';
-import { MastraBase } from './chunk-235X76GC.js';
+import { MastraBase } from './chunk-FI7R232B.js';
 import { __decoratorStart, __decorateElement, __runInitializers } from './chunk-C6A6W6XS.js';
 
 // src/voice/voice.ts

package/dist/{chunk-V2Q2FEVO.cjs → chunk-KSPHQSZX.cjs}

@@ -1,7 +1,7 @@
 'use strict';
 
 var chunkUVRGQEMD_cjs = require('./chunk-UVRGQEMD.cjs');
-var chunkEG74VBSP_cjs = require('./chunk-EG74VBSP.cjs');
+var chunkYEULQPUY_cjs = require('./chunk-YEULQPUY.cjs');
 var chunkRWTSGWWL_cjs = require('./chunk-RWTSGWWL.cjs');
 
 // src/tts/index.ts
@@ -10,7 +10,7 @@ _MastraTTS_decorators = [chunkUVRGQEMD_cjs.InstrumentClass({
   prefix: "tts",
   excludeMethods: ["__setTools", "__setLogger", "__setTelemetry", "#log"]
 })];
-exports.MastraTTS = class MastraTTS extends (_a = chunkEG74VBSP_cjs.MastraBase) {
+exports.MastraTTS = class MastraTTS extends (_a = chunkYEULQPUY_cjs.MastraBase) {
   model;
   constructor({
     model

package/dist/{chunk-NZV4OA6P.js → chunk-LHRZV5VG.js}

@@ -1,4 +1,4 @@
-import { MastraBase } from './chunk-235X76GC.js';
+import { MastraBase } from './chunk-FI7R232B.js';
 
 // src/vector/vector.ts
 var MastraVector = class extends MastraBase {
@@ -8,31 +8,10 @@ var MastraVector = class extends MastraBase {
   get indexSeparator() {
     return "_";
   }
-  baseKeys = {
-    query: ["queryVector", "topK", "filter", "includeVector"],
-    upsert: ["vectors", "metadata", "ids"],
-    createIndex: ["dimension", "metric"]
-  };
-  normalizeArgs(method, [first, ...rest], extendedKeys = []) {
-    if (typeof first === "object") {
-      return first;
-    }
-    this.logger.warn(
-      `Deprecation Warning: Passing individual arguments to ${method}() is deprecated.
-      Please use an object parameter instead.
-      Individual arguments will be removed on May 20th, 2025.`
-    );
-    const baseKeys = this.baseKeys[method] || [];
-    const paramKeys = [...baseKeys, ...extendedKeys].slice(0, rest.length);
-    return {
-      indexName: first,
-      ...Object.fromEntries(paramKeys.map((key, i) => [key, rest[i]]))
-    };
-  }
   async validateExistingIndex(indexName, dimension, metric) {
     let info;
     try {
-      info = await this.describeIndex(indexName);
+      info = await this.describeIndex({ indexName });
     } catch (infoError) {
       const message = `Index "${indexName}" already exists, but failed to fetch index info for dimension check: ${infoError}`;
       this.logger?.error(message);