@mastra/core 0.0.0-commonjs-20250227130920

package/dist/chunk-SY5244IR.js

@@ -0,0 +1,1499 @@
+import { InstrumentClass } from './chunk-W5HVJX45.js';
+import { delay } from './chunk-WIBGG4X6.js';
+import { MastraBase } from './chunk-OZ4XVJ6F.js';
+import { RegisteredLogger } from './chunk-PHMSPCTC.js';
+import { executeHook } from './chunk-BB4KXGBU.js';
+import { __decoratorStart, __decorateElement, __runInitializers } from './chunk-C6A6W6XS.js';
+import { randomUUID } from 'crypto';
+import { z } from 'zod';
+import { jsonSchema, generateText, Output, generateObject, streamText, streamObject } from 'ai';
+
+// src/llm/model/base.ts
+var MastraLLMBase = class extends MastraBase {
+  // @ts-ignore
+  #mastra;
+  #model;
+  constructor({
+    name,
+    model
+  }) {
+    super({
+      component: RegisteredLogger.LLM,
+      name
+    });
+    this.#model = model;
+  }
+  getProvider() {
+    return this.#model.provider;
+  }
+  getModelId() {
+    return this.#model.modelId;
+  }
+  getModel() {
+    return this.#model;
+  }
+  convertToMessages(messages) {
+    if (Array.isArray(messages)) {
+      return messages.map(m => {
+        if (typeof m === "string") {
+          return {
+            role: "user",
+            content: m
+          };
+        }
+        return m;
+      });
+    }
+    return [{
+      role: "user",
+      content: messages
+    }];
+  }
+  __registerPrimitives(p) {
+    if (p.telemetry) {
+      this.__setTelemetry(p.telemetry);
+    }
+    if (p.logger) {
+      this.__setLogger(p.logger);
+    }
+    this.#mastra = p;
+  }
+  async __text(input) {
+    this.logger.debug(`[LLMs:${this.name}] Generating text.`, {
+      input
+    });
+    throw new Error("Method not implemented.");
+  }
+  async __textObject(input) {
+    this.logger.debug(`[LLMs:${this.name}] Generating object.`, {
+      input
+    });
+    throw new Error("Method not implemented.");
+  }
+  async generate(messages, options = {}) {
+    this.logger.debug(`[LLMs:${this.name}] Generating text.`, {
+      messages,
+      options
+    });
+    throw new Error("Method not implemented.");
+  }
+  async __stream(input) {
+    this.logger.debug(`[LLMs:${this.name}] Streaming text.`, {
+      input
+    });
+    throw new Error("Method not implemented.");
+  }
+  async __streamObject(input) {
+    this.logger.debug(`[LLMs:${this.name}] Streaming object.`, {
+      input
+    });
+    throw new Error("Method not implemented.");
+  }
+  async stream(messages, options = {}) {
+    this.logger.debug(`[LLMs:${this.name}] Streaming text.`, {
+      messages,
+      options
+    });
+    throw new Error("Method not implemented.");
+  }
+};
+var MastraLLM = class extends MastraLLMBase {
+  #model;
+  #mastra;
+  constructor({
+    model,
+    mastra
+  }) {
+    super({
+      name: "aisdk",
+      model
+    });
+    this.#model = model;
+    if (mastra) {
+      this.#mastra = mastra;
+      if (mastra.logger) {
+        this.__setLogger(mastra.logger);
+      }
+    }
+  }
+  __registerPrimitives(p) {
+    if (p.telemetry) {
+      this.__setTelemetry(p.telemetry);
+    }
+    if (p.logger) {
+      this.__setLogger(p.logger);
+    }
+    this.#mastra = p;
+  }
+  getProvider() {
+    return this.#model.provider;
+  }
+  getModelId() {
+    return this.#model.modelId;
+  }
+  getModel() {
+    return this.#model;
+  }
+  convertTools({
+    tools,
+    runId
+  } = {}) {
+    this.logger.debug("Starting tool conversion for LLM");
+    const converted = Object.entries(tools || {}).reduce((memo, value) => {
+      const k = value[0];
+      const tool = value[1];
+      if (tool) {
+        memo[k] = {
+          description: tool.description,
+          parameters: tool.inputSchema,
+          execute: async props => {
+            try {
+              this.logger.debug("Executing tool", {
+                tool: k,
+                props
+              });
+              return tool?.execute?.({
+                context: props,
+                mastra: this.#mastra,
+                runId
+              }) ?? void 0;
+            } catch (error) {
+              this.logger.error("Error executing tool", {
+                tool: k,
+                props,
+                error
+              });
+              throw error;
+            }
+          }
+        };
+      }
+      return memo;
+    }, {});
+    this.logger.debug(`Converted tools for LLM`);
+    return converted;
+  }
+  async __text({
+    runId,
+    messages,
+    maxSteps,
+    tools,
+    convertedTools,
+    temperature,
+    toolChoice = "auto",
+    onStepFinish,
+    experimental_output,
+    telemetry
+  }) {
+    const model = this.#model;
+    this.logger.debug(`[LLM] - Generating text`, {
+      runId,
+      messages,
+      maxSteps,
+      tools: Object.keys(tools || convertedTools || {})
+    });
+    const finalTools = convertedTools || this.convertTools({
+      tools,
+      runId
+    });
+    const argsForExecute = {
+      model,
+      temperature,
+      tools: {
+        ...finalTools
+      },
+      toolChoice,
+      maxSteps,
+      onStepFinish: async props => {
+        onStepFinish?.(JSON.stringify(props, null, 2));
+        this.logger.debug("[LLM] - Step Change:", {
+          text: props?.text,
+          toolCalls: props?.toolCalls,
+          toolResults: props?.toolResults,
+          finishReason: props?.finishReason,
+          usage: props?.usage,
+          runId
+        });
+        if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
+          this.logger.warn("Rate limit approaching, waiting 10 seconds", {
+            runId
+          });
+          await delay(10 * 1e3);
+        }
+      }
+    };
+    let schema;
+    if (experimental_output) {
+      this.logger.debug("[LLM] - Using experimental output", {
+        runId
+      });
+      if (typeof experimental_output.parse === "function") {
+        schema = experimental_output;
+        if (schema instanceof z.ZodArray) {
+          schema = schema._def.type;
+        }
+      } else {
+        schema = jsonSchema(experimental_output);
+      }
+    }
+    return await generateText({
+      messages,
+      ...argsForExecute,
+      experimental_telemetry: {
+        ...this.experimental_telemetry,
+        ...telemetry
+      },
+      experimental_output: schema ? Output.object({
+        schema
+      }) : void 0
+    });
+  }
+  async __textObject({
+    messages,
+    onStepFinish,
+    maxSteps = 5,
+    tools,
+    convertedTools,
+    structuredOutput,
+    runId,
+    temperature,
+    toolChoice = "auto",
+    telemetry
+  }) {
+    const model = this.#model;
+    this.logger.debug(`[LLM] - Generating a text object`, {
+      runId
+    });
+    const finalTools = convertedTools || this.convertTools({
+      tools,
+      runId
+    });
+    const argsForExecute = {
+      model,
+      temperature,
+      tools: {
+        ...finalTools
+      },
+      maxSteps,
+      toolChoice,
+      onStepFinish: async props => {
+        onStepFinish?.(JSON.stringify(props, null, 2));
+        this.logger.debug("[LLM] - Step Change:", {
+          text: props?.text,
+          toolCalls: props?.toolCalls,
+          toolResults: props?.toolResults,
+          finishReason: props?.finishReason,
+          usage: props?.usage,
+          runId
+        });
+        if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
+          this.logger.warn("Rate limit approaching, waiting 10 seconds", {
+            runId
+          });
+          await delay(10 * 1e3);
+        }
+      }
+    };
+    let schema;
+    let output = "object";
+    if (typeof structuredOutput.parse === "function") {
+      schema = structuredOutput;
+      if (schema instanceof z.ZodArray) {
+        output = "array";
+        schema = schema._def.type;
+      }
+    } else {
+      schema = jsonSchema(structuredOutput);
+    }
+    return await generateObject({
+      messages,
+      ...argsForExecute,
+      output,
+      schema,
+      experimental_telemetry: {
+        ...this.experimental_telemetry,
+        ...telemetry
+      }
+    });
+  }
+  async __stream({
+    messages,
+    onStepFinish,
+    onFinish,
+    maxSteps = 5,
+    tools,
+    convertedTools,
+    runId,
+    temperature,
+    toolChoice = "auto",
+    experimental_output,
+    telemetry
+  }) {
+    const model = this.#model;
+    this.logger.debug(`[LLM] - Streaming text`, {
+      runId,
+      messages,
+      maxSteps,
+      tools: Object.keys(tools || convertedTools || {})
+    });
+    const finalTools = convertedTools || this.convertTools({
+      tools,
+      runId
+    });
+    const argsForExecute = {
+      model,
+      temperature,
+      tools: {
+        ...finalTools
+      },
+      maxSteps,
+      toolChoice,
+      onStepFinish: async props => {
+        onStepFinish?.(JSON.stringify(props, null, 2));
+        this.logger.debug("[LLM] - Stream Step Change:", {
+          text: props?.text,
+          toolCalls: props?.toolCalls,
+          toolResults: props?.toolResults,
+          finishReason: props?.finishReason,
+          usage: props?.usage,
+          runId
+        });
+        if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
+          this.logger.warn("Rate limit approaching, waiting 10 seconds", {
+            runId
+          });
+          await delay(10 * 1e3);
+        }
+      },
+      onFinish: async props => {
+        onFinish?.(JSON.stringify(props, null, 2));
+        this.logger.debug("[LLM] - Stream Finished:", {
+          text: props?.text,
+          toolCalls: props?.toolCalls,
+          toolResults: props?.toolResults,
+          finishReason: props?.finishReason,
+          usage: props?.usage,
+          runId
+        });
+      }
+    };
+    let schema;
+    if (experimental_output) {
+      this.logger.debug("[LLM] - Using experimental output", {
+        runId
+      });
+      if (typeof experimental_output.parse === "function") {
+        schema = experimental_output;
+        if (schema instanceof z.ZodArray) {
+          schema = schema._def.type;
+        }
+      } else {
+        schema = jsonSchema(experimental_output);
+      }
+    }
+    return await streamText({
+      messages,
+      ...argsForExecute,
+      experimental_telemetry: {
+        ...this.experimental_telemetry,
+        ...telemetry
+      },
+      experimental_output: schema ? Output.object({
+        schema
+      }) : void 0
+    });
+  }
+  async __streamObject({
+    messages,
+    onStepFinish,
+    onFinish,
+    maxSteps = 5,
+    tools,
+    convertedTools,
+    structuredOutput,
+    runId,
+    temperature,
+    toolChoice = "auto",
+    telemetry
+  }) {
+    const model = this.#model;
+    this.logger.debug(`[LLM] - Streaming structured output`, {
+      runId,
+      messages,
+      maxSteps,
+      tools: Object.keys(tools || convertedTools || {})
+    });
+    const finalTools = convertedTools || this.convertTools({
+      tools,
+      runId
+    });
+    const argsForExecute = {
+      model,
+      temperature,
+      tools: {
+        ...finalTools
+      },
+      maxSteps,
+      toolChoice,
+      onStepFinish: async props => {
+        onStepFinish?.(JSON.stringify(props, null, 2));
+        this.logger.debug("[LLM] - Stream Step Change:", {
+          text: props?.text,
+          toolCalls: props?.toolCalls,
+          toolResults: props?.toolResults,
+          finishReason: props?.finishReason,
+          usage: props?.usage,
+          runId
+        });
+        if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
+          this.logger.warn("Rate limit approaching, waiting 10 seconds", {
+            runId
+          });
+          await delay(10 * 1e3);
+        }
+      },
+      onFinish: async props => {
+        onFinish?.(JSON.stringify(props, null, 2));
+        this.logger.debug("[LLM] - Stream Finished:", {
+          text: props?.text,
+          toolCalls: props?.toolCalls,
+          toolResults: props?.toolResults,
+          finishReason: props?.finishReason,
+          usage: props?.usage,
+          runId
+        });
+      }
+    };
+    let schema;
+    let output = "object";
+    if (typeof structuredOutput.parse === "function") {
+      schema = structuredOutput;
+      if (schema instanceof z.ZodArray) {
+        output = "array";
+        schema = schema._def.type;
+      }
+    } else {
+      schema = jsonSchema(structuredOutput);
+    }
+    return streamObject({
+      messages,
+      ...argsForExecute,
+      output,
+      schema,
+      experimental_telemetry: {
+        ...this.experimental_telemetry,
+        ...telemetry
+      }
+    });
+  }
+  async generate(messages, {
+    maxSteps = 5,
+    onStepFinish,
+    tools,
+    convertedTools,
+    runId,
+    output = "text",
+    temperature,
+    telemetry
+  } = {}) {
+    const msgs = this.convertToMessages(messages);
+    if (output === "text") {
+      return await this.__text({
+        messages: msgs,
+        onStepFinish,
+        maxSteps,
+        tools,
+        convertedTools,
+        runId,
+        temperature
+      });
+    }
+    return await this.__textObject({
+      messages: msgs,
+      structuredOutput: output,
+      onStepFinish,
+      maxSteps,
+      tools,
+      convertedTools,
+      runId,
+      telemetry
+    });
+  }
+  async stream(messages, {
+    maxSteps = 5,
+    onFinish,
+    onStepFinish,
+    tools,
+    convertedTools,
+    runId,
+    output = "text",
+    temperature,
+    telemetry
+  } = {}) {
+    const msgs = this.convertToMessages(messages);
+    if (output === "text") {
+      return await this.__stream({
+        messages: msgs,
+        onStepFinish,
+        onFinish,
+        maxSteps,
+        tools,
+        convertedTools,
+        runId,
+        temperature,
+        telemetry
+      });
+    }
+    return await this.__streamObject({
+      messages: msgs,
+      structuredOutput: output,
+      onStepFinish,
+      onFinish,
+      maxSteps,
+      tools,
+      convertedTools,
+      runId,
+      temperature,
+      telemetry
+    });
+  }
+};
+
+// src/agent/index.ts
+var _Agent_decorators, _init, _a;
+_Agent_decorators = [InstrumentClass({
+  prefix: "agent",
+  excludeMethods: ["__setTools", "__setLogger", "__setTelemetry", "log"]
+})];
+var Agent = class extends (_a = MastraBase) {
+  name;
+  llm;
+  instructions;
+  model;
+  #mastra;
+  #memory;
+  tools;
+  /** @deprecated This property is deprecated. Use evals instead. */
+  metrics;
+  evals;
+  voice;
+  constructor(config) {
+    super({
+      component: RegisteredLogger.AGENT
+    });
+    this.name = config.name;
+    this.instructions = config.instructions;
+    if (!config.model) {
+      throw new Error(`LanguageModel is required to create an Agent. Please provide the 'model'.`);
+    }
+    this.llm = new MastraLLM({
+      model: config.model
+    });
+    this.tools = {};
+    this.metrics = {};
+    this.evals = {};
+    if (config.tools) {
+      this.tools = config.tools;
+    }
+    if (config.mastra) {
+      this.#mastra = config.mastra;
+    }
+    if (config.metrics) {
+      this.logger.warn("The metrics property is deprecated. Please use evals instead to add evaluation metrics.");
+      this.metrics = config.metrics;
+      this.evals = config.metrics;
+    }
+    if (config.evals) {
+      this.evals = config.evals;
+    }
+    if (config.memory) {
+      this.#memory = config.memory;
+    }
+    if (config.voice) {
+      this.voice = config.voice;
+    }
+  }
+  hasOwnMemory() {
+    return Boolean(this.#memory);
+  }
+  getMemory() {
+    return this.#memory ?? this.#mastra?.memory;
+  }
+  __updateInstructions(newInstructions) {
+    this.instructions = newInstructions;
+    this.logger.debug(`[Agents:${this.name}] Instructions updated.`, {
+      model: this.model,
+      name: this.name
+    });
+  }
+  __registerPrimitives(p) {
+    if (p.telemetry) {
+      this.__setTelemetry(p.telemetry);
+    }
+    if (p.logger) {
+      this.__setLogger(p.logger);
+    }
+    this.llm.__registerPrimitives(p);
+    this.#mastra = p;
+    this.logger.debug(`[Agents:${this.name}] initialized.`, {
+      model: this.model,
+      name: this.name
+    });
+  }
+  /**
+   * Set the concrete tools for the agent
+   * @param tools
+   */
+  __setTools(tools) {
+    this.tools = tools;
+    this.logger.debug(`[Agents:${this.name}] Tools set for agent ${this.name}`, {
+      model: this.model,
+      name: this.name
+    });
+  }
+  async generateTitleFromUserMessage({
+    message
+  }) {
+    const {
+      object
+    } = await this.llm.__textObject({
+      messages: [{
+        role: "system",
+        content: `
+
+      - you will generate a short title based on the first message a user begins a conversation with
+      - ensure it is not more than 80 characters long
+      - the title should be a summary of the user's message
+      - do not use quotes or colons`
+      }, {
+        role: "user",
+        content: JSON.stringify(message)
+      }],
+      structuredOutput: z.object({
+        title: z.string()
+      })
+    });
+    return object.title;
+  }
+  getMostRecentUserMessage(messages) {
+    const userMessages = messages.filter(message => message.role === "user");
+    return userMessages.at(-1);
+  }
+  async genTitle(userMessage) {
+    let title = "New Thread";
+    try {
+      if (userMessage) {
+        title = await this.generateTitleFromUserMessage({
+          message: userMessage
+        });
+      }
+    } catch (e) {
+      console.error("Error generating title:", e);
+    }
+    return title;
+  }
+  async saveMemory({
+    threadId,
+    memoryConfig,
+    resourceId,
+    userMessages,
+    runId
+  }) {
+    const userMessage = this.getMostRecentUserMessage(userMessages);
+    const memory = this.getMemory();
+    if (memory) {
+      let thread;
+      if (!threadId) {
+        this.logger.debug(`No threadId, creating new thread for agent ${this.name}`, {
+          runId: runId || this.name
+        });
+        const title = await this.genTitle(userMessage);
+        thread = await memory.createThread({
+          threadId,
+          resourceId,
+          title,
+          memoryConfig
+        });
+      } else {
+        thread = await memory.getThreadById({
+          threadId
+        });
+        if (!thread) {
+          this.logger.debug(`Thread with id ${threadId} not found, creating new thread for agent ${this.name}`, {
+            runId: runId || this.name
+          });
+          const title = await this.genTitle(userMessage);
+          thread = await memory.createThread({
+            threadId,
+            resourceId,
+            title,
+            memoryConfig
+          });
+        }
+      }
+      const newMessages = userMessage ? [userMessage] : userMessages;
+      if (thread) {
+        const messages = newMessages.map(u => {
+          return {
+            id: this.getMemory()?.generateId(),
+            createdAt: /* @__PURE__ */new Date(),
+            threadId: thread.id,
+            ...u,
+            content: u.content,
+            role: u.role,
+            type: "text"
+          };
+        });
+        const memoryMessages = threadId && memory ? (await memory.rememberMessages({
+          threadId,
+          config: memoryConfig,
+          vectorMessageSearch: messages.slice(-1).map(m => {
+            if (typeof m === `string`) {
+              return m;
+            }
+            return m?.content || ``;
+          }).join(`
+`)
+        })).messages : [];
+        if (memory) {
+          await memory.saveMessages({
+            messages,
+            memoryConfig
+          });
+        }
+        this.logger.debug("Saved messages to memory", {
+          threadId: thread.id,
+          runId
+        });
+        const memorySystemMessage = memory && threadId ? await memory.getSystemMessage({
+          threadId,
+          memoryConfig
+        }) : null;
+        return {
+          threadId: thread.id,
+          messages: [memorySystemMessage ? {
+            role: "system",
+            content: memorySystemMessage
+          } : null, ...this.sanitizeResponseMessages(memoryMessages), ...newMessages].filter(message => Boolean(message))
+        };
+      }
+      return {
+        threadId: thread?.id || threadId || "",
+        messages: userMessages
+      };
+    }
+    return {
+      threadId: threadId || "",
+      messages: userMessages
+    };
+  }
+  async saveResponse({
+    result,
+    threadId,
+    resourceId,
+    runId,
+    memoryConfig
+  }) {
+    const {
+      response
+    } = result;
+    try {
+      if (response.messages) {
+        const ms = Array.isArray(response.messages) ? response.messages : [response.messages];
+        const responseMessagesWithoutIncompleteToolCalls = this.sanitizeResponseMessages(ms);
+        const memory = this.getMemory();
+        if (memory) {
+          this.logger.debug(`[Agent:${this.name}] - Memory persistence: store=${this.getMemory()?.constructor.name} threadId=${threadId}`, {
+            runId,
+            resourceId,
+            threadId,
+            memoryStore: this.getMemory()?.constructor.name
+          });
+          await memory.saveMessages({
+            memoryConfig,
+            messages: responseMessagesWithoutIncompleteToolCalls.map(message => {
+              const messageId = randomUUID();
+              let toolCallIds;
+              let toolCallArgs;
+              let toolNames;
+              let type = "text";
+              if (message.role === "tool") {
+                toolCallIds = message.content.map(content => content.toolCallId);
+                type = "tool-result";
+              }
+              if (message.role === "assistant") {
+                const assistantContent = message.content;
+                const assistantToolCalls = assistantContent.map(content => {
+                  if (content.type === "tool-call") {
+                    return {
+                      toolCallId: content.toolCallId,
+                      toolArgs: content.args,
+                      toolName: content.toolName
+                    };
+                  }
+                  return void 0;
+                })?.filter(Boolean);
+                toolCallIds = assistantToolCalls?.map(toolCall => toolCall.toolCallId);
+                toolCallArgs = assistantToolCalls?.map(toolCall => toolCall.toolArgs);
+                toolNames = assistantToolCalls?.map(toolCall => toolCall.toolName);
+                type = assistantContent?.[0]?.type;
+              }
+              return {
+                id: messageId,
+                threadId,
+                role: message.role,
+                content: message.content,
+                createdAt: /* @__PURE__ */new Date(),
+                toolCallIds: toolCallIds?.length ? toolCallIds : void 0,
+                toolCallArgs: toolCallArgs?.length ? toolCallArgs : void 0,
+                toolNames: toolNames?.length ? toolNames : void 0,
+                type
+              };
+            })
+          });
+        }
+      }
+    } catch (err) {
+      this.logger.error(`[Agent:${this.name}] - Failed to save assistant response`, {
+        error: err,
+        runId
+      });
+    }
+  }
+  sanitizeResponseMessages(messages) {
+    let toolResultIds = [];
+    let toolCallIds = [];
+    for (const message of messages) {
+      if (message.role === "tool") {
+        for (const content of message.content) {
+          if (content.type === "tool-result") {
+            toolResultIds.push(content.toolCallId);
+          }
+        }
+      }
+      if (message.role === "assistant" || message.role === "user") {
+        for (const content of message.content) {
+          if (typeof content !== `string`) {
+            if (content.type === `tool-call`) {
+              toolCallIds.push(content.toolCallId);
+            }
+          }
+        }
+      }
+    }
+    const messagesBySanitizedContent = messages.map(message => {
+      if (message.role !== "assistant" && message.role !== `tool` && message.role !== `user`) return message;
+      if (typeof message.content === "string") return message;
+      const sanitizedContent = message.content.filter(content => {
+        if (content.type === `tool-call`) {
+          return toolResultIds.includes(content.toolCallId);
+        }
+        if (content.type === `text`) {
+          return content.text.trim() !== ``;
+        }
+        if (content.type === `tool-result`) {
+          return toolCallIds.includes(content.toolCallId);
+        }
+        return true;
+      });
+      return {
+        ...message,
+        content: sanitizedContent
+      };
+    });
+    return messagesBySanitizedContent.filter(message => {
+      if (typeof message.content === `string`) {
+        return message.content !== "";
+      }
+      if (Array.isArray(message.content)) {
+        return message.content.length && message.content.every(c => {
+          if (c.type === `text`) {
+            return c.text && c.text !== "";
+          }
+          return true;
+        });
+      }
+      return true;
+    });
+  }
+  convertTools({
+    toolsets,
+    // threadId,
+    runId
+  }) {
+    this.logger.debug(`[Agents:${this.name}] - Assigning tools`, {
+      runId
+    });
+    const converted = Object.entries(this.tools || {}).reduce((memo, value) => {
+      const k = value[0];
+      const tool = this.tools[k];
+      if (tool) {
+        memo[k] = {
+          description: tool.description,
+          parameters: tool.inputSchema,
+          execute: async args => {
+            try {
+              this.logger.debug(`[Agent:${this.name}] - Executing tool ${k}`, {
+                name: k,
+                description: tool.description,
+                args,
+                runId
+              });
+              return tool?.execute?.({
+                context: args,
+                mastra: this.#mastra,
+                runId
+              }) ?? void 0;
+            } catch (err) {
+              this.logger.error(`[Agent:${this.name}] - Failed execution`, {
+                error: err,
+                runId
+              });
+              throw err;
+            }
+          }
+        };
+      }
+      return memo;
+    }, {});
+    const toolsFromToolsetsConverted = {
+      ...converted
+    };
+    const toolsFromToolsets = Object.values(toolsets || {});
+    if (toolsFromToolsets.length > 0) {
+      this.logger.debug(`[Agent:${this.name}] - Adding tools from toolsets ${Object.keys(toolsets || {}).join(", ")}`, {
+        runId
+      });
+      toolsFromToolsets.forEach(toolset => {
+        Object.entries(toolset).forEach(([toolName, tool]) => {
+          const toolObj = tool;
+          toolsFromToolsetsConverted[toolName] = {
+            description: toolObj.description || "",
+            parameters: toolObj.inputSchema,
+            execute: async args => {
+              try {
+                this.logger.debug(`[Agent:${this.name}] - Executing tool ${toolName}`, {
+                  name: toolName,
+                  description: toolObj.description,
+                  args,
+                  runId
+                });
+                return toolObj.execute({
+                  context: args,
+                  runId
+                });
+              } catch (err) {
+                this.logger.error(`[Agent:${this.name}] - Failed toolset execution`, {
+                  error: err,
+                  runId
+                });
+                throw err;
+              }
+            }
+          };
+        });
+      });
+    }
+    return toolsFromToolsetsConverted;
+  }
+  async preExecute({
+    resourceId,
+    runId,
+    threadId,
+    memoryConfig,
+    messages
+  }) {
+    let coreMessages = [];
+    let threadIdToUse = threadId;
+    this.logger.debug(`Saving user messages in memory for agent ${this.name}`, {
+      runId
+    });
+    const saveMessageResponse = await this.saveMemory({
+      threadId,
+      resourceId,
+      userMessages: messages,
+      memoryConfig
+    });
+    coreMessages = saveMessageResponse.messages;
+    threadIdToUse = saveMessageResponse.threadId;
+    return {
+      coreMessages,
+      threadIdToUse
+    };
+  }
+  __primitive({
+    messages,
+    context,
+    threadId,
+    memoryConfig,
+    resourceId,
+    runId,
+    toolsets
+  }) {
+    return {
+      before: async () => {
+        if (process.env.NODE_ENV !== "test") {
+          this.logger.debug(`[Agents:${this.name}] - Starting generation`, {
+            runId
+          });
+        }
+        const systemMessage = {
+          role: "system",
+          content: `${this.instructions}.`
+        };
+        let coreMessages = messages;
+        let threadIdToUse = threadId;
+        const memory = this.getMemory();
+        if (threadId && memory && !resourceId) {
+          throw new Error(`A resourceId must be provided when passing a threadId and using Memory. Saw threadId ${threadId} but resourceId is ${resourceId}`);
+        }
+        if (memory && resourceId) {
+          this.logger.debug(`[Agent:${this.name}] - Memory persistence enabled: store=${this.getMemory()?.constructor.name}, resourceId=${resourceId}`, {
+            runId,
+            resourceId,
+            threadId: threadIdToUse,
+            memoryStore: this.getMemory()?.constructor.name
+          });
+          const preExecuteResult = await this.preExecute({
+            resourceId,
+            runId,
+            threadId: threadIdToUse,
+            memoryConfig,
+            messages
+          });
+          coreMessages = preExecuteResult.coreMessages;
+          threadIdToUse = preExecuteResult.threadIdToUse;
+        }
+        let convertedTools;
+        if (toolsets && Object.keys(toolsets || {}).length > 0 || this.getMemory() && resourceId) {
+          const reasons = [];
+          if (toolsets && Object.keys(toolsets || {}).length > 0) {
+            reasons.push(`toolsets present (${Object.keys(toolsets || {}).length} tools)`);
+          }
+          if (this.getMemory() && resourceId) {
+            reasons.push("memory and resourceId available");
+          }
+          this.logger.debug(`[Agent:${this.name}] - Enhancing tools: ${reasons.join(", ")}`, {
+            runId,
+            toolsets: toolsets ? Object.keys(toolsets) : void 0,
+            hasMemory: !!this.getMemory(),
+            hasResourceId: !!resourceId
+          });
+          convertedTools = this.convertTools({
+            toolsets,
+            threadId: threadIdToUse,
+            runId
+          });
+        }
+        const messageObjects = [systemMessage, ...(context || []), ...coreMessages];
+        return {
+          messageObjects,
+          convertedTools,
+          threadId: threadIdToUse
+        };
+      },
+      after: async ({
+        result,
+        threadId: threadId2,
+        memoryConfig: memoryConfig2,
+        outputText,
+        runId: runId2
+      }) => {
+        const resToLog = {
+          text: result?.text,
+          object: result?.object,
+          toolResults: result?.toolResults,
+          toolCalls: result?.toolCalls,
+          usage: result?.usage,
+          steps: result?.steps?.map(s => {
+            return {
+              stepType: s?.stepType,
+              text: result?.text,
+              object: result?.object,
+              toolResults: result?.toolResults,
+              toolCalls: result?.toolCalls,
+              usage: result?.usage
+            };
+          })
+        };
+        this.logger.debug(`[Agent:${this.name}] - Post processing LLM response`, {
+          runId: runId2,
+          result: resToLog,
+          threadId: threadId2
+        });
+        if (this.getMemory() && resourceId) {
+          try {
+            await this.saveResponse({
+              result,
+              threadId: threadId2,
+              resourceId,
+              memoryConfig: memoryConfig2,
+              runId: runId2
+            });
+          } catch (e) {
+            this.logger.error("Error saving response", {
+              error: e,
+              runId: runId2,
+              result: resToLog,
+              threadId: threadId2
+            });
+          }
+        }
+        if (Object.keys(this.evals || {}).length > 0) {
+          const input = messages.map(message => message.content).join("\n");
+          const runIdToUse = runId2 || crypto.randomUUID();
+          for (const metric of Object.values(this.evals || {})) {
+            executeHook("onGeneration" /* ON_GENERATION */, {
+              input,
+              output: outputText,
+              runId: runIdToUse,
+              metric,
+              agentName: this.name,
+              instructions: this.instructions
+            });
+          }
+        }
+      }
+    };
+  }
+  async generate(messages, {
+    context,
+    threadId: threadIdInFn,
+    memoryOptions,
+    resourceId,
+    maxSteps = 5,
+    onStepFinish,
+    runId,
+    toolsets,
+    output = "text",
+    temperature,
+    toolChoice = "auto",
+    experimental_output,
+    telemetry
+  } = {}) {
+    let messagesToUse = [];
+    if (typeof messages === `string`) {
+      messagesToUse = [{
+        role: "user",
+        content: messages
+      }];
+    } else {
+      messagesToUse = messages.map(message => {
+        if (typeof message === `string`) {
+          return {
+            role: "user",
+            content: message
+          };
+        }
+        return message;
+      });
+    }
+    const runIdToUse = runId || randomUUID();
+    const {
+      before,
+      after
+    } = this.__primitive({
+      messages: messagesToUse,
+      context,
+      threadId: threadIdInFn,
+      memoryConfig: memoryOptions,
+      resourceId,
+      runId: runIdToUse,
+      toolsets
+    });
+    const {
+      threadId,
+      messageObjects,
+      convertedTools
+    } = await before();
+    if (output === "text" && experimental_output) {
+      const result2 = await this.llm.__text({
+        messages: messageObjects,
+        tools: this.tools,
+        convertedTools,
+        onStepFinish,
+        maxSteps,
+        runId: runIdToUse,
+        temperature,
+        toolChoice,
+        experimental_output
+      });
+      const outputText2 = result2.text;
+      await after({
+        result: result2,
+        threadId,
+        memoryConfig: memoryOptions,
+        outputText: outputText2,
+        runId: runIdToUse
+      });
+      const newResult = result2;
+      newResult.object = result2.experimental_output;
+      return newResult;
+    }
+    if (output === "text") {
+      const result2 = await this.llm.__text({
+        messages: messageObjects,
+        tools: this.tools,
+        convertedTools,
+        onStepFinish,
+        maxSteps,
+        runId: runIdToUse,
+        temperature,
+        toolChoice,
+        telemetry
+      });
+      const outputText2 = result2.text;
+      await after({
+        result: result2,
+        threadId,
+        memoryConfig: memoryOptions,
+        outputText: outputText2,
+        runId: runIdToUse
+      });
+      return result2;
+    }
+    const result = await this.llm.__textObject({
+      messages: messageObjects,
+      tools: this.tools,
+      structuredOutput: output,
+      convertedTools,
+      onStepFinish,
+      maxSteps,
+      runId: runIdToUse,
+      temperature,
+      toolChoice,
+      telemetry
+    });
+    const outputText = JSON.stringify(result.object);
+    await after({
+      result,
+      threadId,
+      memoryConfig: memoryOptions,
+      outputText,
+      runId: runIdToUse
+    });
+    return result;
+  }
+  async stream(messages, {
+    context,
+    threadId: threadIdInFn,
+    memoryOptions,
+    resourceId,
+    maxSteps = 5,
+    onFinish,
+    onStepFinish,
+    runId,
+    toolsets,
+    output = "text",
+    temperature,
+    toolChoice = "auto",
+    experimental_output,
+    telemetry
+  } = {}) {
+    const runIdToUse = runId || randomUUID();
+    let messagesToUse = [];
+    if (typeof messages === `string`) {
+      messagesToUse = [{
+        role: "user",
+        content: messages
+      }];
+    } else {
+      messagesToUse = messages.map(message => {
+        if (typeof message === `string`) {
+          return {
+            role: "user",
+            content: message
+          };
+        }
+        return message;
+      });
+    }
+    const {
+      before,
+      after
+    } = this.__primitive({
+      messages: messagesToUse,
+      context,
+      threadId: threadIdInFn,
+      memoryConfig: memoryOptions,
+      resourceId,
+      runId: runIdToUse,
+      toolsets
+    });
+    const {
+      threadId,
+      messageObjects,
+      convertedTools
+    } = await before();
+    if (output === "text" && experimental_output) {
+      this.logger.debug(`Starting agent ${this.name} llm stream call`, {
+        runId
+      });
+      const streamResult = await this.llm.__stream({
+        messages: messageObjects,
+        temperature,
+        tools: this.tools,
+        convertedTools,
+        onStepFinish,
+        onFinish: async result => {
+          try {
+            const res = JSON.parse(result) || {};
+            const outputText = res.text;
+            await after({
+              result: res,
+              threadId,
+              memoryConfig: memoryOptions,
+              outputText,
+              runId: runIdToUse
+            });
+          } catch (e) {
+            this.logger.error("Error saving memory on finish", {
+              error: e,
+              runId
+            });
+          }
+          onFinish?.(result);
+        },
+        maxSteps,
+        runId: runIdToUse,
+        toolChoice,
+        experimental_output
+      });
+      const newStreamResult = streamResult;
+      newStreamResult.partialObjectStream = streamResult.experimental_partialOutputStream;
+      return newStreamResult;
+    } else if (output === "text") {
+      this.logger.debug(`Starting agent ${this.name} llm stream call`, {
+        runId
+      });
+      return this.llm.__stream({
+        messages: messageObjects,
+        temperature,
+        tools: this.tools,
+        convertedTools,
+        onStepFinish,
+        onFinish: async result => {
+          try {
+            const res = JSON.parse(result) || {};
+            const outputText = res.text;
+            await after({
+              result: res,
+              threadId,
+              memoryConfig: memoryOptions,
+              outputText,
+              runId: runIdToUse
+            });
+          } catch (e) {
+            this.logger.error("Error saving memory on finish", {
+              error: e,
+              runId
+            });
+          }
+          onFinish?.(result);
+        },
+        maxSteps,
+        runId: runIdToUse,
+        toolChoice,
+        telemetry
+      });
+    }
+    this.logger.debug(`Starting agent ${this.name} llm streamObject call`, {
+      runId
+    });
+    return this.llm.__streamObject({
+      messages: messageObjects,
+      tools: this.tools,
+      temperature,
+      structuredOutput: output,
+      convertedTools,
+      onStepFinish,
+      onFinish: async result => {
+        try {
+          const res = JSON.parse(result) || {};
+          const outputText = JSON.stringify(res.object);
+          await after({
+            result: res,
+            threadId,
+            memoryConfig: memoryOptions,
+            outputText,
+            runId: runIdToUse
+          });
+        } catch (e) {
+          this.logger.error("Error saving memory on finish", {
+            error: e,
+            runId
+          });
+        }
+        onFinish?.(result);
+      },
+      maxSteps,
+      runId: runIdToUse,
+      toolChoice,
+      telemetry
+    });
+  }
+  /**
+   * Convert text to speech using the configured voice provider
+   * @param input Text or text stream to convert to speech
+   * @param options Speech options including speaker and provider-specific options
+   * @returns Audio stream
+   */
+  async speak(input, options) {
+    if (!this.voice) {
+      throw new Error("No voice provider configured");
+    }
+    try {
+      return this.voice.speak(input, options);
+    } catch (e) {
+      this.logger.error("Error during agent speak", {
+        error: e
+      });
+      throw e;
+    }
+  }
+  /**
+   * Convert speech to text using the configured voice provider
+   * @param audioStream Audio stream to transcribe
+   * @param options Provider-specific transcription options
+   * @returns Text or text stream
+   */
+  async listen(audioStream, options) {
+    if (!this.voice) {
+      throw new Error("No voice provider configured");
+    }
+    try {
+      return this.voice.listen(audioStream, options);
+    } catch (e) {
+      this.logger.error("Error during agent listen", {
+        error: e
+      });
+      throw e;
+    }
+  }
+  /**
+   * Get a list of available speakers from the configured voice provider
+   * @throws {Error} If no voice provider is configured
+   * @returns {Promise<Array<{voiceId: string}>>} List of available speakers
+   */
+  async getSpeakers() {
+    if (!this.voice) {
+      throw new Error("No voice provider configured");
+    }
+    try {
+      return await this.voice.getSpeakers();
+    } catch (e) {
+      this.logger.error("Error during agent getSpeakers", {
+        error: e
+      });
+      throw e;
+    }
+  }
+};
+Agent = /*@__PURE__*/(_ => {
+  _init = __decoratorStart(_a);
+  Agent = __decorateElement(_init, 0, "Agent", _Agent_decorators, Agent);
+  __runInitializers(_init, 1, Agent);
+  return Agent;
+})();
+
+export { Agent };