@mastra/core 0.15.3-alpha.2 → 0.15.3-alpha.4

package/dist/{chunk-UG3KS3XV.cjs.map → chunk-F6XWBVVG.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/integration/integration.ts","../src/integration/openapi-toolset.ts"],"names":["createTool","z"],"mappings":";;;;;;AAGO,IAAM,cAAN,MAAwD;AAAA,EAC7D,IAAA,GAAe,aAAA;AAAA,EACP,SAAA;AAAA,EAER,WAAA,GAAc;AACZ,IAAA,IAAA,CAAK,YAAY,EAAC;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAA,CAAiB,MAAc,EAAA,EAAc;AAC3C,IAAA,IAAI,IAAA,CAAK,SAAA,CAAU,IAAI,CAAA,EAAG;AACxB,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,eAAA,EAAkB,IAAI,CAAA,oBAAA,CAAsB,CAAA;AAAA,IAC9D;AACA,IAAA,IAAA,CAAK,SAAA,CAAU,IAAI,CAAA,GAAI,EAAA;AAAA,EACzB;AAAA,EAEO,YAAA,CAAa,EAAE,UAAA,EAAW,EAAuD;AACtF,IAAA,IAAI,UAAA,EAAY;AACd,MAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,IAAA,CAAK,SAAS,CAAA,CAAE,MAAA,CAAO,CAAC,GAAA,EAAK,CAAC,CAAA,EAAG,CAAC,CAAA,KAAM;AAC5D,QAAA,OAAO;AAAA,UACL,GAAG,GAAA;AAAA,UACH,CAAC,CAAC,GAAG;AAAA,YACH,MAAM,CAAA,CAAE;AAAA;AACV,SACF;AAAA,MACF,CAAA,EAAG,EAAE,CAAA;AAAA,IACP;AACA,IAAA,OAAO,IAAA,CAAK,SAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,eAAe,OAAA,EAAkE;AAC/E,IAAA,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAM,SAAS,OAAA,EAA2E;AACxF,IAAA,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAM,YAAA,GAAmC;AACvC,IAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,EAC1C;AACF;AC7CO,IAAe,iBAAf,MAA8B;AAAA,EAInC,QAAA,GAAmB,SAAA;AAAA,EAEnB,WAAA,GAAc;AAAA,EAAC;AAAA,EAEf,IAAc,WAAA,GAAmB;AAC/B,IAAA,OAAO,EAAC;AAAA,EACV;AAAA,EAEA,IAAc,kBAAA,GAAwE;AACpF,IAAA,OAAO,EAAC;AAAA,EACV;AAAA,EAEA,IAAc,UAAA,GAAkB;AAC9B,IAAA,OAAO,EAAC;AAAA,EACV;AAAA,EAEA,MAAM,YAAA,GAA6B;AACjC,IAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,EACvC;AAAA,EAEU,yBAAA,GAA+B;AACvC,IAAA,MAAM,EAAE,MAAA,EAAQ,OAAA,EAAS,GAAG,aAAA,KAAkB,IAAA,CAAK,UAAA;AACnD,IAAA,MAAM,UAAU,IAAA,CAAK,WAAA;AACrB,IAAA,MAAM,iBAAiB,IAAA,CAAK,kBAAA;AAE5B,IAAA,MAAM,KAAA,GAAQ,OAAO,IAAA,CAAK,aAAa,EAAE,MAAA,CAAO,CAAC,KAAK,GAAA,KAAQ;AAC5D,MAAA,MAAM,OAAA,GAAU,cAAA,CAAe,GAAG,CAAA,EAAG,OAAA;AAErC,MAAA,MAAM,eAAA,GAAkB,WAAW,GAAG,CAAA,CAAA;AAEtC,MAAA,MAAM,OAAOA,4BAAA,CAAW;AAAA,QACtB,EAAA,EAAI,GAAA;AAAA,QACJ,aAAa,OAAA,CAAQ,GAAG,KAAKC,KAAA,CAAE,MAAA,CAAO,EAAE,CAAA;AAAA,QACxC,aAAa,OAAA,IAAW,eAAA;AAAA;AAAA,QAExB,OAAA,EAAS,OAAO,EAAE,OAAA,EAAQ,KAAM;AAC9B,UAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,YAAA,EAAa;AACvC,UAAA,MAAM,KAAA,GAAQ,OAAO,GAA0B,CAAA;AAC/C,UAAA,OAAQ,KAAA,CAAc;AAAA,YACpB,GAAI;AAAA,WACL,CAAA;AAAA,QACH;AAAA,OACD,CAAA;AAED,MAAA,OAAO,EAAE,GAAG,GAAA,EAAK,CAAC,GAAG,GAAG,IAAA,EAAK;AAAA,IAC/B,CAAA,EAAG,EAAE,CAAA;AAEL,IAAA,OAAO,KAAA;AAAA,EACT;AACF","file":"chunk-UG3KS3XV.cjs","sourcesContent":["import type { ToolAction } from '../tools';\nimport type { Workflow } from '../workflows';\n\nexport class Integration<ToolsParams = void, ApiClient = void> {\n  name: string = 'Integration';\n  private workflows: Record<string, Workflow>;\n\n  constructor() {\n    this.workflows = {};\n  }\n\n  /**\n   * Workflows\n   */\n\n  registerWorkflow(name: string, fn: Workflow) {\n    if (this.workflows[name]) {\n      throw new Error(`Sync function \"${name}\" already registered`);\n    }\n    this.workflows[name] = fn;\n  }\n\n  public getWorkflows({ serialized }: { serialized?: boolean }): Record<string, Workflow> {\n    if (serialized) {\n      return Object.entries(this.workflows).reduce((acc, [k, v]) => {\n        return {\n          ...acc,\n          [k]: {\n            name: v.name,\n          },\n        };\n      }, {});\n    }\n    return this.workflows;\n  }\n\n  /**\n   * TOOLS\n   */\n  getStaticTools(_params?: ToolsParams): Record<string, ToolAction<any, any, any>> {\n    throw new Error('Method not implemented.');\n  }\n\n  async getTools(_params?: ToolsParams): Promise<Record<string, ToolAction<any, any, any>>> {\n    throw new Error('Method not implemented.');\n  }\n\n  async getApiClient(): Promise<ApiClient> {\n    throw new Error('Method not implemented');\n  }\n}\n","import { z } from 'zod';\n\nimport { createTool } from '../tools';\nimport type { ToolAction } from '../tools';\n\nexport abstract class OpenAPIToolset {\n  abstract readonly name: string;\n  abstract readonly tools: Record<string, ToolAction<any, any, any>>;\n\n  authType: string = 'API_KEY';\n\n  constructor() {}\n\n  protected get toolSchemas(): any {\n    return {};\n  }\n\n  protected get toolDocumentations(): Record<string, { comment: string; doc?: string }> {\n    return {};\n  }\n\n  protected get baseClient(): any {\n    return {};\n  }\n\n  async getApiClient(): Promise<any> {\n    throw new Error('API not implemented');\n  }\n\n  protected _generateIntegrationTools<T>() {\n    const { client: _client, ...clientMethods } = this.baseClient;\n    const schemas = this.toolSchemas;\n    const documentations = this.toolDocumentations;\n\n    const tools = Object.keys(clientMethods).reduce((acc, key) => {\n      const comment = documentations[key]?.comment;\n      // const doc = documentations[key]?.doc;\n      const fallbackComment = `Execute ${key}`;\n\n      const tool = createTool({\n        id: key,\n        inputSchema: schemas[key] || z.object({}),\n        description: comment || fallbackComment,\n        // documentation: doc || fallbackComment,\n        execute: async ({ context }) => {\n          const client = await this.getApiClient();\n          const value = client[key as keyof typeof client];\n          return (value as any)({\n            ...(context as any),\n          });\n        },\n      });\n\n      return { ...acc, [key]: tool };\n    }, {});\n\n    return tools as T;\n  }\n}\n"]}
+{"version":3,"sources":["../src/integration/integration.ts","../src/integration/openapi-toolset.ts"],"names":["createTool","z"],"mappings":";;;;;;AAGO,IAAM,cAAN,MAAwD;AAAA,EAC7D,IAAA,GAAe,aAAA;AAAA,EACP,SAAA;AAAA,EAER,WAAA,GAAc;AACZ,IAAA,IAAA,CAAK,YAAY,EAAC;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA,EAMA,gBAAA,CAAiB,MAAc,EAAA,EAAc;AAC3C,IAAA,IAAI,IAAA,CAAK,SAAA,CAAU,IAAI,CAAA,EAAG;AACxB,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,eAAA,EAAkB,IAAI,CAAA,oBAAA,CAAsB,CAAA;AAAA,IAC9D;AACA,IAAA,IAAA,CAAK,SAAA,CAAU,IAAI,CAAA,GAAI,EAAA;AAAA,EACzB;AAAA,EAEO,YAAA,CAAa,EAAE,UAAA,EAAW,EAAuD;AACtF,IAAA,IAAI,UAAA,EAAY;AACd,MAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,IAAA,CAAK,SAAS,CAAA,CAAE,MAAA,CAAO,CAAC,GAAA,EAAK,CAAC,CAAA,EAAG,CAAC,CAAA,KAAM;AAC5D,QAAA,OAAO;AAAA,UACL,GAAG,GAAA;AAAA,UACH,CAAC,CAAC,GAAG;AAAA,YACH,MAAM,CAAA,CAAE;AAAA;AACV,SACF;AAAA,MACF,CAAA,EAAG,EAAE,CAAA;AAAA,IACP;AACA,IAAA,OAAO,IAAA,CAAK,SAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,eAAe,OAAA,EAAkE;AAC/E,IAAA,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAM,SAAS,OAAA,EAA2E;AACxF,IAAA,MAAM,IAAI,MAAM,yBAAyB,CAAA;AAAA,EAC3C;AAAA,EAEA,MAAM,YAAA,GAAmC;AACvC,IAAA,MAAM,IAAI,MAAM,wBAAwB,CAAA;AAAA,EAC1C;AACF;AC7CO,IAAe,iBAAf,MAA8B;AAAA,EAInC,QAAA,GAAmB,SAAA;AAAA,EAEnB,WAAA,GAAc;AAAA,EAAC;AAAA,EAEf,IAAc,WAAA,GAAmB;AAC/B,IAAA,OAAO,EAAC;AAAA,EACV;AAAA,EAEA,IAAc,kBAAA,GAAwE;AACpF,IAAA,OAAO,EAAC;AAAA,EACV;AAAA,EAEA,IAAc,UAAA,GAAkB;AAC9B,IAAA,OAAO,EAAC;AAAA,EACV;AAAA,EAEA,MAAM,YAAA,GAA6B;AACjC,IAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,EACvC;AAAA,EAEU,yBAAA,GAA+B;AACvC,IAAA,MAAM,EAAE,MAAA,EAAQ,OAAA,EAAS,GAAG,aAAA,KAAkB,IAAA,CAAK,UAAA;AACnD,IAAA,MAAM,UAAU,IAAA,CAAK,WAAA;AACrB,IAAA,MAAM,iBAAiB,IAAA,CAAK,kBAAA;AAE5B,IAAA,MAAM,KAAA,GAAQ,OAAO,IAAA,CAAK,aAAa,EAAE,MAAA,CAAO,CAAC,KAAK,GAAA,KAAQ;AAC5D,MAAA,MAAM,OAAA,GAAU,cAAA,CAAe,GAAG,CAAA,EAAG,OAAA;AAErC,MAAA,MAAM,eAAA,GAAkB,WAAW,GAAG,CAAA,CAAA;AAEtC,MAAA,MAAM,OAAOA,4BAAA,CAAW;AAAA,QACtB,EAAA,EAAI,GAAA;AAAA,QACJ,aAAa,OAAA,CAAQ,GAAG,KAAKC,KAAA,CAAE,MAAA,CAAO,EAAE,CAAA;AAAA,QACxC,aAAa,OAAA,IAAW,eAAA;AAAA;AAAA,QAExB,OAAA,EAAS,OAAO,EAAE,OAAA,EAAQ,KAAM;AAC9B,UAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,YAAA,EAAa;AACvC,UAAA,MAAM,KAAA,GAAQ,OAAO,GAA0B,CAAA;AAC/C,UAAA,OAAQ,KAAA,CAAc;AAAA,YACpB,GAAI;AAAA,WACL,CAAA;AAAA,QACH;AAAA,OACD,CAAA;AAED,MAAA,OAAO,EAAE,GAAG,GAAA,EAAK,CAAC,GAAG,GAAG,IAAA,EAAK;AAAA,IAC/B,CAAA,EAAG,EAAE,CAAA;AAEL,IAAA,OAAO,KAAA;AAAA,EACT;AACF","file":"chunk-F6XWBVVG.cjs","sourcesContent":["import type { ToolAction } from '../tools';\nimport type { Workflow } from '../workflows';\n\nexport class Integration<ToolsParams = void, ApiClient = void> {\n  name: string = 'Integration';\n  private workflows: Record<string, Workflow>;\n\n  constructor() {\n    this.workflows = {};\n  }\n\n  /**\n   * Workflows\n   */\n\n  registerWorkflow(name: string, fn: Workflow) {\n    if (this.workflows[name]) {\n      throw new Error(`Sync function \"${name}\" already registered`);\n    }\n    this.workflows[name] = fn;\n  }\n\n  public getWorkflows({ serialized }: { serialized?: boolean }): Record<string, Workflow> {\n    if (serialized) {\n      return Object.entries(this.workflows).reduce((acc, [k, v]) => {\n        return {\n          ...acc,\n          [k]: {\n            name: v.name,\n          },\n        };\n      }, {});\n    }\n    return this.workflows;\n  }\n\n  /**\n   * TOOLS\n   */\n  getStaticTools(_params?: ToolsParams): Record<string, ToolAction<any, any, any>> {\n    throw new Error('Method not implemented.');\n  }\n\n  async getTools(_params?: ToolsParams): Promise<Record<string, ToolAction<any, any, any>>> {\n    throw new Error('Method not implemented.');\n  }\n\n  async getApiClient(): Promise<ApiClient> {\n    throw new Error('Method not implemented');\n  }\n}\n","import { z } from 'zod';\n\nimport { createTool } from '../tools';\nimport type { ToolAction } from '../tools';\n\nexport abstract class OpenAPIToolset {\n  abstract readonly name: string;\n  abstract readonly tools: Record<string, ToolAction<any, any, any>>;\n\n  authType: string = 'API_KEY';\n\n  constructor() {}\n\n  protected get toolSchemas(): any {\n    return {};\n  }\n\n  protected get toolDocumentations(): Record<string, { comment: string; doc?: string }> {\n    return {};\n  }\n\n  protected get baseClient(): any {\n    return {};\n  }\n\n  async getApiClient(): Promise<any> {\n    throw new Error('API not implemented');\n  }\n\n  protected _generateIntegrationTools<T>() {\n    const { client: _client, ...clientMethods } = this.baseClient;\n    const schemas = this.toolSchemas;\n    const documentations = this.toolDocumentations;\n\n    const tools = Object.keys(clientMethods).reduce((acc, key) => {\n      const comment = documentations[key]?.comment;\n      // const doc = documentations[key]?.doc;\n      const fallbackComment = `Execute ${key}`;\n\n      const tool = createTool({\n        id: key,\n        inputSchema: schemas[key] || z.object({}),\n        description: comment || fallbackComment,\n        // documentation: doc || fallbackComment,\n        execute: async ({ context }) => {\n          const client = await this.getApiClient();\n          const value = client[key as keyof typeof client];\n          return (value as any)({\n            ...(context as any),\n          });\n        },\n      });\n\n      return { ...acc, [key]: tool };\n    }, {});\n\n    return tools as T;\n  }\n}\n"]}

package/dist/{chunk-WHEH32QU.cjs → chunk-FQRDHVZC.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunk7OSUKFNG_cjs = require('./chunk-7OSUKFNG.cjs');
+var chunkKUNWELBC_cjs = require('./chunk-KUNWELBC.cjs');
 
 // src/relevance/cohere/index.ts
 var CohereRelevanceScorer = class {
@@ -54,7 +54,7 @@ Relevance score (0-1):`;
 var MastraAgentRelevanceScorer = class {
   agent;
   constructor(name, model) {
-    this.agent = new chunk7OSUKFNG_cjs.Agent({
+    this.agent = new chunkKUNWELBC_cjs.Agent({
       name: `Relevance Scorer ${name}`,
       instructions: `You are a specialized agent for evaluating the relevance of text to queries.
 Your task is to rate how well a text passage answers a given query.
@@ -87,5 +87,5 @@ Always return just the number, no explanation.`,
 exports.CohereRelevanceScorer = CohereRelevanceScorer;
 exports.MastraAgentRelevanceScorer = MastraAgentRelevanceScorer;
 exports.createSimilarityPrompt = createSimilarityPrompt;
-//# sourceMappingURL=chunk-WHEH32QU.cjs.map
-//# sourceMappingURL=chunk-WHEH32QU.cjs.map
+//# sourceMappingURL=chunk-FQRDHVZC.cjs.map
+//# sourceMappingURL=chunk-FQRDHVZC.cjs.map

package/dist/{chunk-WHEH32QU.cjs.map → chunk-FQRDHVZC.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/relevance/cohere/index.ts","../src/relevance/relevance-score-provider.ts","../src/relevance/mastra-agent/index.ts"],"names":["Agent"],"mappings":";;;;;AAoBO,IAAM,wBAAN,MAA8D;AAAA,EAC3D,KAAA;AAAA,EACA,MAAA;AAAA,EACR,WAAA,CAAY,OAAe,MAAA,EAAiB;AAC1C,IAAA,OAAA,CAAQ,IAAA;AAAA,MACN;AAAA,KACF;AACA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AACd,IAAA,IAAA,CAAK,KAAA,GAAQ,KAAA;AAAA,EACf;AAAA,EACA,MAAM,iBAAA,CAAkB,KAAA,EAAe,IAAA,EAA+B;AACpE,IAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,CAAA,gCAAA,CAAA,EAAoC;AAAA,MAC/D,MAAA,EAAQ,MAAA;AAAA,MACR,OAAA,EAAS;AAAA,QACP,cAAA,EAAgB,kBAAA;AAAA,QAChB,aAAA,EAAe,CAAA,OAAA,EAAU,IAAA,CAAK,MAAM,CAAA;AAAA,OACtC;AAAA,MACA,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,QACnB,KAAA;AAAA,QACA,SAAA,EAAW,CAAC,IAAI,CAAA;AAAA,QAChB,OAAO,IAAA,CAAK,KAAA;AAAA,QACZ,KAAA,EAAO;AAAA,OACR;AAAA,KACF,CAAA;AAED,IAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,kBAAA,EAAqB,QAAA,CAAS,MAAM,IAAI,MAAM,QAAA,CAAS,IAAA,EAAM,CAAA,CAAE,CAAA;AAAA,IACjF;AAEA,IAAA,MAAM,IAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAClC,IAAA,MAAM,cAAA,GAAiB,IAAA,CAAK,OAAA,CAAQ,CAAC,CAAA,EAAG,eAAA;AAExC,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,MAAM,IAAI,MAAM,6CAA6C,CAAA;AAAA,IAC/D;AAEA,IAAA,OAAO,cAAA;AAAA,EACT;AACF;;;ACpDO,SAAS,sBAAA,CAAuB,OAAe,IAAA,EAAsB;AAC1E,EAAA,OAAO,CAAA;;AAAA,OAAA,EAEA,KAAK;;AAAA,MAAA,EAEN,IAAI;;AAAA,sBAAA,CAAA;AAGZ;;;ACRO,IAAM,6BAAN,MAAmE;AAAA,EAChE,KAAA;AAAA,EAER,WAAA,CAAY,MAAc,KAAA,EAA4B;AACpD,IAAA,IAAA,CAAK,KAAA,GAAQ,IAAIA,uBAAA,CAAM;AAAA,MACrB,IAAA,EAAM,oBAAoB,IAAI,CAAA,CAAA;AAAA,MAC9B,YAAA,EAAc,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,8CAAA,CAAA;AAAA,MAUd;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,iBAAA,CAAkB,KAAA,EAAe,IAAA,EAA+B;AACpE,IAAA,MAAM,MAAA,GAAS,sBAAA,CAAuB,KAAA,EAAO,IAAI,CAAA;AAEjD,IAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,KAAA,CAAM,QAAA,EAAS;AAExC,IAAA,IAAI,QAAA;AACJ,IAAA,IAAI,KAAA,CAAM,yBAAyB,IAAA,EAAM;AACvC,MAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,KAAA,CAAM,cAAc,MAAM,CAAA;AAC1D,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B,CAAA,MAAO;AACL,MAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,KAAA,CAAM,SAAS,MAAM,CAAA;AACrD,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B;AACA,IAAA,OAAO,WAAW,QAAQ,CAAA;AAAA,EAC5B;AACF","file":"chunk-WHEH32QU.cjs","sourcesContent":["import type { RelevanceScoreProvider } from '../relevance-score-provider';\n\ninterface CohereRerankingResponse {\n  results: Array<{\n    index: number;\n    relevance_score: number;\n  }>;\n  id: string;\n  meta: {\n    api_version: {\n      version: string;\n      is_experimental: boolean;\n    };\n    billed_units: {\n      search_units: number;\n    };\n  };\n}\n\n/** @deprecated Import from `@mastra/rag` instead */\nexport class CohereRelevanceScorer implements RelevanceScoreProvider {\n  private model: string;\n  private apiKey?: string;\n  constructor(model: string, apiKey?: string) {\n    console.warn(\n      'CohereRelevanceScorer exported from @mastra/core is deprecated. Please import from \"@mastra/rag\" instead.',\n    );\n    this.apiKey = apiKey;\n    this.model = model;\n  }\n  async getRelevanceScore(query: string, text: string): Promise<number> {\n    const response = await fetch(`https://api.cohere.com/v2/rerank`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n        Authorization: `Bearer ${this.apiKey}`,\n      },\n      body: JSON.stringify({\n        query,\n        documents: [text],\n        model: this.model,\n        top_n: 1,\n      }),\n    });\n\n    if (!response.ok) {\n      throw new Error(`Cohere API error: ${response.status} ${await response.text()}`);\n    }\n\n    const data = (await response.json()) as CohereRerankingResponse;\n    const relevanceScore = data.results[0]?.relevance_score;\n\n    if (!relevanceScore) {\n      throw new Error('No relevance score found on Cohere response');\n    }\n\n    return relevanceScore;\n  }\n}\n","// Provider interfaces\nexport interface RelevanceScoreProvider {\n  getRelevanceScore(text1: string, text2: string): Promise<number>;\n}\n\n// Helper function used by providers\nexport function createSimilarityPrompt(query: string, text: string): string {\n  return `Rate the semantic similarity between the following the query and the text on a scale from 0 to 1 (decimals allowed), where 1 means exactly the same meaning and 0 means completely different:\n\nQuery: ${query}\n\nText: ${text}\n\nRelevance score (0-1):`;\n}\n","import { Agent } from '../../agent';\nimport type { MastraLanguageModel } from '../../llm/model/shared.types';\nimport { createSimilarityPrompt } from '../relevance-score-provider';\nimport type { RelevanceScoreProvider } from '../relevance-score-provider';\n\n// Mastra Agent implementation\nexport class MastraAgentRelevanceScorer implements RelevanceScoreProvider {\n  private agent: Agent;\n\n  constructor(name: string, model: MastraLanguageModel) {\n    this.agent = new Agent({\n      name: `Relevance Scorer ${name}`,\n      instructions: `You are a specialized agent for evaluating the relevance of text to queries.\nYour task is to rate how well a text passage answers a given query.\nOutput only a number between 0 and 1, where:\n1.0 = Perfectly relevant, directly answers the query\n0.0 = Completely irrelevant\nConsider:\n- Direct relevance to the question\n- Completeness of information\n- Quality and specificity\nAlways return just the number, no explanation.`,\n      model,\n    });\n  }\n\n  async getRelevanceScore(query: string, text: string): Promise<number> {\n    const prompt = createSimilarityPrompt(query, text);\n\n    const model = await this.agent.getModel();\n\n    let response: string;\n    if (model.specificationVersion === 'v2') {\n      const responseText = await this.agent.generateVNext(prompt);\n      response = responseText.text;\n    } else {\n      const responseText = await this.agent.generate(prompt);\n      response = responseText.text;\n    }\n    return parseFloat(response);\n  }\n}\n"]}
+{"version":3,"sources":["../src/relevance/cohere/index.ts","../src/relevance/relevance-score-provider.ts","../src/relevance/mastra-agent/index.ts"],"names":["Agent"],"mappings":";;;;;AAoBO,IAAM,wBAAN,MAA8D;AAAA,EAC3D,KAAA;AAAA,EACA,MAAA;AAAA,EACR,WAAA,CAAY,OAAe,MAAA,EAAiB;AAC1C,IAAA,OAAA,CAAQ,IAAA;AAAA,MACN;AAAA,KACF;AACA,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AACd,IAAA,IAAA,CAAK,KAAA,GAAQ,KAAA;AAAA,EACf;AAAA,EACA,MAAM,iBAAA,CAAkB,KAAA,EAAe,IAAA,EAA+B;AACpE,IAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,CAAA,gCAAA,CAAA,EAAoC;AAAA,MAC/D,MAAA,EAAQ,MAAA;AAAA,MACR,OAAA,EAAS;AAAA,QACP,cAAA,EAAgB,kBAAA;AAAA,QAChB,aAAA,EAAe,CAAA,OAAA,EAAU,IAAA,CAAK,MAAM,CAAA;AAAA,OACtC;AAAA,MACA,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,QACnB,KAAA;AAAA,QACA,SAAA,EAAW,CAAC,IAAI,CAAA;AAAA,QAChB,OAAO,IAAA,CAAK,KAAA;AAAA,QACZ,KAAA,EAAO;AAAA,OACR;AAAA,KACF,CAAA;AAED,IAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,kBAAA,EAAqB,QAAA,CAAS,MAAM,IAAI,MAAM,QAAA,CAAS,IAAA,EAAM,CAAA,CAAE,CAAA;AAAA,IACjF;AAEA,IAAA,MAAM,IAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAClC,IAAA,MAAM,cAAA,GAAiB,IAAA,CAAK,OAAA,CAAQ,CAAC,CAAA,EAAG,eAAA;AAExC,IAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,MAAA,MAAM,IAAI,MAAM,6CAA6C,CAAA;AAAA,IAC/D;AAEA,IAAA,OAAO,cAAA;AAAA,EACT;AACF;;;ACpDO,SAAS,sBAAA,CAAuB,OAAe,IAAA,EAAsB;AAC1E,EAAA,OAAO,CAAA;;AAAA,OAAA,EAEA,KAAK;;AAAA,MAAA,EAEN,IAAI;;AAAA,sBAAA,CAAA;AAGZ;;;ACRO,IAAM,6BAAN,MAAmE;AAAA,EAChE,KAAA;AAAA,EAER,WAAA,CAAY,MAAc,KAAA,EAA4B;AACpD,IAAA,IAAA,CAAK,KAAA,GAAQ,IAAIA,uBAAA,CAAM;AAAA,MACrB,IAAA,EAAM,oBAAoB,IAAI,CAAA,CAAA;AAAA,MAC9B,YAAA,EAAc,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,8CAAA,CAAA;AAAA,MAUd;AAAA,KACD,CAAA;AAAA,EACH;AAAA,EAEA,MAAM,iBAAA,CAAkB,KAAA,EAAe,IAAA,EAA+B;AACpE,IAAA,MAAM,MAAA,GAAS,sBAAA,CAAuB,KAAA,EAAO,IAAI,CAAA;AAEjD,IAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,KAAA,CAAM,QAAA,EAAS;AAExC,IAAA,IAAI,QAAA;AACJ,IAAA,IAAI,KAAA,CAAM,yBAAyB,IAAA,EAAM;AACvC,MAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,KAAA,CAAM,cAAc,MAAM,CAAA;AAC1D,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B,CAAA,MAAO;AACL,MAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,KAAA,CAAM,SAAS,MAAM,CAAA;AACrD,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B;AACA,IAAA,OAAO,WAAW,QAAQ,CAAA;AAAA,EAC5B;AACF","file":"chunk-FQRDHVZC.cjs","sourcesContent":["import type { RelevanceScoreProvider } from '../relevance-score-provider';\n\ninterface CohereRerankingResponse {\n  results: Array<{\n    index: number;\n    relevance_score: number;\n  }>;\n  id: string;\n  meta: {\n    api_version: {\n      version: string;\n      is_experimental: boolean;\n    };\n    billed_units: {\n      search_units: number;\n    };\n  };\n}\n\n/** @deprecated Import from `@mastra/rag` instead */\nexport class CohereRelevanceScorer implements RelevanceScoreProvider {\n  private model: string;\n  private apiKey?: string;\n  constructor(model: string, apiKey?: string) {\n    console.warn(\n      'CohereRelevanceScorer exported from @mastra/core is deprecated. Please import from \"@mastra/rag\" instead.',\n    );\n    this.apiKey = apiKey;\n    this.model = model;\n  }\n  async getRelevanceScore(query: string, text: string): Promise<number> {\n    const response = await fetch(`https://api.cohere.com/v2/rerank`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n        Authorization: `Bearer ${this.apiKey}`,\n      },\n      body: JSON.stringify({\n        query,\n        documents: [text],\n        model: this.model,\n        top_n: 1,\n      }),\n    });\n\n    if (!response.ok) {\n      throw new Error(`Cohere API error: ${response.status} ${await response.text()}`);\n    }\n\n    const data = (await response.json()) as CohereRerankingResponse;\n    const relevanceScore = data.results[0]?.relevance_score;\n\n    if (!relevanceScore) {\n      throw new Error('No relevance score found on Cohere response');\n    }\n\n    return relevanceScore;\n  }\n}\n","// Provider interfaces\nexport interface RelevanceScoreProvider {\n  getRelevanceScore(text1: string, text2: string): Promise<number>;\n}\n\n// Helper function used by providers\nexport function createSimilarityPrompt(query: string, text: string): string {\n  return `Rate the semantic similarity between the following the query and the text on a scale from 0 to 1 (decimals allowed), where 1 means exactly the same meaning and 0 means completely different:\n\nQuery: ${query}\n\nText: ${text}\n\nRelevance score (0-1):`;\n}\n","import { Agent } from '../../agent';\nimport type { MastraLanguageModel } from '../../llm/model/shared.types';\nimport { createSimilarityPrompt } from '../relevance-score-provider';\nimport type { RelevanceScoreProvider } from '../relevance-score-provider';\n\n// Mastra Agent implementation\nexport class MastraAgentRelevanceScorer implements RelevanceScoreProvider {\n  private agent: Agent;\n\n  constructor(name: string, model: MastraLanguageModel) {\n    this.agent = new Agent({\n      name: `Relevance Scorer ${name}`,\n      instructions: `You are a specialized agent for evaluating the relevance of text to queries.\nYour task is to rate how well a text passage answers a given query.\nOutput only a number between 0 and 1, where:\n1.0 = Perfectly relevant, directly answers the query\n0.0 = Completely irrelevant\nConsider:\n- Direct relevance to the question\n- Completeness of information\n- Quality and specificity\nAlways return just the number, no explanation.`,\n      model,\n    });\n  }\n\n  async getRelevanceScore(query: string, text: string): Promise<number> {\n    const prompt = createSimilarityPrompt(query, text);\n\n    const model = await this.agent.getModel();\n\n    let response: string;\n    if (model.specificationVersion === 'v2') {\n      const responseText = await this.agent.generateVNext(prompt);\n      response = responseText.text;\n    } else {\n      const responseText = await this.agent.generate(prompt);\n      response = responseText.text;\n    }\n    return parseFloat(response);\n  }\n}\n"]}

package/dist/{chunk-4DKPMUAC.cjs → chunk-I7OAONIW.cjs}

@@ -11,7 +11,6 @@ var NoOpAISpan = class _NoOpAISpan {
   type;
   attributes;
   parent;
-  trace;
   traceId;
   startTime;
   endTime;
@@ -28,7 +27,6 @@ var NoOpAISpan = class _NoOpAISpan {
     this.attributes = options.attributes || {};
     this.metadata = options.metadata;
     this.parent = options.parent;
-    this.trace = options.parent ? options.parent.trace : this;
     this.traceId = "no-op-trace";
     this.startTime = /* @__PURE__ */ new Date();
     this.aiTracing = aiTracing;
@@ -902,6 +900,232 @@ var DefaultExporter = class {
   }
 };
 
+// src/ai-tracing/registry.ts
+var AITracingRegistry = class {
+  instances = /* @__PURE__ */ new Map();
+  defaultInstance;
+  selector;
+  /**
+   * Register a tracing instance
+   */
+  register(name, instance, isDefault = false) {
+    if (this.instances.has(name)) {
+      throw new Error(`AI Tracing instance '${name}' already registered`);
+    }
+    this.instances.set(name, instance);
+    if (isDefault || !this.defaultInstance) {
+      this.defaultInstance = instance;
+    }
+  }
+  /**
+   * Get a tracing instance by name
+   */
+  get(name) {
+    return this.instances.get(name);
+  }
+  /**
+   * Get the default tracing instance
+   */
+  getDefault() {
+    return this.defaultInstance;
+  }
+  /**
+   * Set the tracing selector function
+   */
+  setSelector(selector) {
+    this.selector = selector;
+  }
+  /**
+   * Get the selected tracing instance based on context
+   */
+  getSelected(context) {
+    if (this.selector) {
+      const selected = this.selector(context, this.instances);
+      if (selected && this.instances.has(selected)) {
+        return this.instances.get(selected);
+      }
+    }
+    return this.defaultInstance;
+  }
+  /**
+   * Unregister a tracing instance
+   */
+  unregister(name) {
+    return this.instances.delete(name);
+  }
+  /**
+   * Shutdown all instances and clear the registry
+   */
+  async shutdown() {
+    const shutdownPromises = Array.from(this.instances.values()).map((instance) => instance.shutdown());
+    await Promise.allSettled(shutdownPromises);
+    this.instances.clear();
+  }
+  /**
+   * Clear all instances without shutdown
+   */
+  clear() {
+    this.instances.clear();
+    this.defaultInstance = void 0;
+    this.selector = void 0;
+  }
+  /**
+   * Get all registered instances
+   */
+  getAll() {
+    return new Map(this.instances);
+  }
+};
+var aiTracingRegistry = new AITracingRegistry();
+function registerAITracing(name, instance, isDefault = false) {
+  aiTracingRegistry.register(name, instance, isDefault);
+}
+function getAITracing(name) {
+  return aiTracingRegistry.get(name);
+}
+function getDefaultAITracing() {
+  return aiTracingRegistry.getDefault();
+}
+function setAITracingSelector(selector) {
+  aiTracingRegistry.setSelector(selector);
+}
+function getSelectedAITracing(context) {
+  return aiTracingRegistry.getSelected(context);
+}
+function unregisterAITracing(name) {
+  return aiTracingRegistry.unregister(name);
+}
+async function shutdownAITracingRegistry() {
+  await aiTracingRegistry.shutdown();
+}
+function clearAITracingRegistry() {
+  aiTracingRegistry.clear();
+}
+function getAllAITracing() {
+  return aiTracingRegistry.getAll();
+}
+function hasAITracing(name) {
+  const tracing = getAITracing(name);
+  if (!tracing) return false;
+  const config = tracing.getConfig();
+  const sampling = config.sampling;
+  return sampling.type !== "never" /* NEVER */;
+}
+function isAITracingInstance(obj) {
+  return obj instanceof MastraAITracing;
+}
+function setupAITracing(config) {
+  const entries = Object.entries(config.instances);
+  entries.forEach(([name, tracingDef], index) => {
+    const instance = isAITracingInstance(tracingDef) ? tracingDef : new DefaultAITracing({ ...tracingDef, instanceName: name });
+    const isDefault = index === 0;
+    registerAITracing(name, instance, isDefault);
+  });
+  if (config.selector) {
+    setAITracingSelector(config.selector);
+  }
+}
+
+// src/ai-tracing/utils.ts
+function shallowCleanObject(obj) {
+  const cleaned = {};
+  for (const [key, value] of Object.entries(obj)) {
+    try {
+      JSON.stringify(value);
+      cleaned[key] = value;
+    } catch (error) {
+      cleaned[key] = `[${error instanceof Error ? error.message : String(error)}]`;
+    }
+  }
+  return cleaned;
+}
+function shallowCleanArray(arr) {
+  return arr.map((item) => {
+    if (item && typeof item === "object" && !Array.isArray(item)) {
+      return shallowCleanObject(item);
+    }
+    try {
+      JSON.stringify(item);
+      return item;
+    } catch (error) {
+      return `[${error instanceof Error ? error.message : String(error)}]`;
+    }
+  });
+}
+function shallowClean(value) {
+  if (value === null || value === void 0) {
+    return value;
+  }
+  if (Array.isArray(value)) {
+    return shallowCleanArray(value);
+  }
+  if (typeof value === "object") {
+    return shallowCleanObject(value);
+  }
+  try {
+    JSON.stringify(value);
+    return value;
+  } catch (error) {
+    return `[${error instanceof Error ? error.message : String(error)}]`;
+  }
+}
+function omitKeys(obj, keysToOmit) {
+  return Object.fromEntries(Object.entries(obj).filter(([key]) => !keysToOmit.includes(key)));
+}
+function selectFields(obj, fields) {
+  if (!obj || typeof obj !== "object") {
+    return obj;
+  }
+  const result = {};
+  for (const field of fields) {
+    const value = getNestedValue(obj, field);
+    if (value !== void 0) {
+      setNestedValue(result, field, value);
+    }
+  }
+  return result;
+}
+function getNestedValue(obj, path) {
+  return path.split(".").reduce((current, key) => {
+    return current && typeof current === "object" ? current[key] : void 0;
+  }, obj);
+}
+function setNestedValue(obj, path, value) {
+  const keys = path.split(".");
+  const lastKey = keys.pop();
+  if (!lastKey) {
+    return;
+  }
+  const target = keys.reduce((current, key) => {
+    if (!current[key] || typeof current[key] !== "object") {
+      current[key] = {};
+    }
+    return current[key];
+  }, obj);
+  target[lastKey] = value;
+}
+function getOrCreateSpan(options) {
+  const { type, attributes, tracingContext, runtimeContext, ...rest } = options;
+  if (tracingContext?.currentSpan) {
+    return tracingContext.currentSpan.createChildSpan({
+      type,
+      attributes,
+      ...rest
+    });
+  }
+  const aiTracing = getSelectedAITracing({
+    runtimeContext
+  });
+  return aiTracing?.startSpan({
+    type,
+    attributes,
+    startOptions: {
+      runtimeContext
+    },
+    ...rest
+  });
+}
+
 // src/ai-tracing/default.ts
 function generateSpanId() {
   const bytes = new Uint8Array(8);
@@ -931,7 +1155,6 @@ var DefaultAISpan = class {
   type;
   attributes;
   parent;
-  trace;
   traceId;
   startTime;
   endTime;
@@ -946,13 +1169,12 @@ var DefaultAISpan = class {
     this.id = generateSpanId();
     this.name = options.name;
     this.type = options.type;
-    this.attributes = options.attributes || {};
-    this.metadata = options.metadata;
+    this.attributes = shallowClean(options.attributes) || {};
+    this.metadata = shallowClean(options.metadata);
     this.parent = options.parent;
-    this.trace = options.parent ? options.parent.trace : this;
     this.startTime = /* @__PURE__ */ new Date();
     this.aiTracing = aiTracing;
-    this.input = options.input;
+    this.input = shallowClean(options.input);
     this.isEvent = options.isEvent;
     this.logger = new chunkV5WKCX3G_cjs.ConsoleLogger({
       name: "default-ai-span",
@@ -962,10 +1184,10 @@ var DefaultAISpan = class {
     if (!options.parent) {
       this.traceId = generateTraceId();
     } else {
-      this.traceId = options.parent.trace.traceId;
+      this.traceId = options.parent.traceId;
     }
     if (this.isEvent) {
-      this.output = options.output;
+      this.output = shallowClean(options.output);
     }
   }
   end(options) {
@@ -975,13 +1197,13 @@ var DefaultAISpan = class {
     }
     this.endTime = /* @__PURE__ */ new Date();
     if (options?.output !== void 0) {
-      this.output = options.output;
+      this.output = shallowClean(options.output);
     }
     if (options?.attributes) {
-      this.attributes = { ...this.attributes, ...options.attributes };
+      this.attributes = { ...this.attributes, ...shallowClean(options.attributes) };
     }
     if (options?.metadata) {
-      this.metadata = { ...this.metadata, ...options.metadata };
+      this.metadata = { ...this.metadata, ...shallowClean(options.metadata) };
     }
   }
   error(options) {
@@ -1000,10 +1222,10 @@ var DefaultAISpan = class {
       message: error.message
     };
     if (attributes) {
-      this.attributes = { ...this.attributes, ...attributes };
+      this.attributes = { ...this.attributes, ...shallowClean(attributes) };
     }
     if (metadata) {
-      this.metadata = { ...this.metadata, ...metadata };
+      this.metadata = { ...this.metadata, ...shallowClean(metadata) };
     }
     if (endSpan) {
       this.end();
@@ -1031,16 +1253,16 @@ var DefaultAISpan = class {
       return;
     }
     if (options?.input !== void 0) {
-      this.input = options.input;
+      this.input = shallowClean(options.input);
     }
     if (options?.output !== void 0) {
-      this.output = options.output;
+      this.output = shallowClean(options.output);
     }
     if (options?.attributes) {
-      this.attributes = { ...this.attributes, ...options.attributes };
+      this.attributes = { ...this.attributes, ...shallowClean(options.attributes) };
     }
     if (options?.metadata) {
-      this.metadata = { ...this.metadata, ...options.metadata };
+      this.metadata = { ...this.metadata, ...shallowClean(options.metadata) };
     }
   }
   get isRootSpan() {
@@ -1141,155 +1363,6 @@ var DefaultAITracing = class extends MastraAITracing {
   }
 };
 
-// src/ai-tracing/registry.ts
-var AITracingRegistry = class {
-  instances = /* @__PURE__ */ new Map();
-  defaultInstance;
-  selector;
-  /**
-   * Register a tracing instance
-   */
-  register(name, instance, isDefault = false) {
-    if (this.instances.has(name)) {
-      throw new Error(`AI Tracing instance '${name}' already registered`);
-    }
-    this.instances.set(name, instance);
-    if (isDefault || !this.defaultInstance) {
-      this.defaultInstance = instance;
-    }
-  }
-  /**
-   * Get a tracing instance by name
-   */
-  get(name) {
-    return this.instances.get(name);
-  }
-  /**
-   * Get the default tracing instance
-   */
-  getDefault() {
-    return this.defaultInstance;
-  }
-  /**
-   * Set the tracing selector function
-   */
-  setSelector(selector) {
-    this.selector = selector;
-  }
-  /**
-   * Get the selected tracing instance based on context
-   */
-  getSelected(context) {
-    if (this.selector) {
-      const selected = this.selector(context, this.instances);
-      if (selected && this.instances.has(selected)) {
-        return this.instances.get(selected);
-      }
-    }
-    return this.defaultInstance;
-  }
-  /**
-   * Unregister a tracing instance
-   */
-  unregister(name) {
-    return this.instances.delete(name);
-  }
-  /**
-   * Shutdown all instances and clear the registry
-   */
-  async shutdown() {
-    const shutdownPromises = Array.from(this.instances.values()).map((instance) => instance.shutdown());
-    await Promise.allSettled(shutdownPromises);
-    this.instances.clear();
-  }
-  /**
-   * Clear all instances without shutdown
-   */
-  clear() {
-    this.instances.clear();
-    this.defaultInstance = void 0;
-    this.selector = void 0;
-  }
-  /**
-   * Get all registered instances
-   */
-  getAll() {
-    return new Map(this.instances);
-  }
-};
-var aiTracingRegistry = new AITracingRegistry();
-function registerAITracing(name, instance, isDefault = false) {
-  aiTracingRegistry.register(name, instance, isDefault);
-}
-function getAITracing(name) {
-  return aiTracingRegistry.get(name);
-}
-function getDefaultAITracing() {
-  return aiTracingRegistry.getDefault();
-}
-function setAITracingSelector(selector) {
-  aiTracingRegistry.setSelector(selector);
-}
-function getSelectedAITracing(context) {
-  return aiTracingRegistry.getSelected(context);
-}
-function unregisterAITracing(name) {
-  return aiTracingRegistry.unregister(name);
-}
-async function shutdownAITracingRegistry() {
-  await aiTracingRegistry.shutdown();
-}
-function clearAITracingRegistry() {
-  aiTracingRegistry.clear();
-}
-function getAllAITracing() {
-  return aiTracingRegistry.getAll();
-}
-function hasAITracing(name) {
-  const tracing = getAITracing(name);
-  if (!tracing) return false;
-  const config = tracing.getConfig();
-  const sampling = config.sampling;
-  return sampling.type !== "never" /* NEVER */;
-}
-function isAITracingInstance(obj) {
-  return obj instanceof MastraAITracing;
-}
-function setupAITracing(config) {
-  const entries = Object.entries(config.instances);
-  entries.forEach(([name, tracingDef], index) => {
-    const instance = isAITracingInstance(tracingDef) ? tracingDef : new DefaultAITracing({ ...tracingDef, instanceName: name });
-    const isDefault = index === 0;
-    registerAITracing(name, instance, isDefault);
-  });
-  if (config.selector) {
-    setAITracingSelector(config.selector);
-  }
-}
-
-// src/ai-tracing/utils.ts
-function sanitizeMetadata(metadata) {
-  if (!metadata) return {};
-  const sanitized = {};
-  for (const [key, value] of Object.entries(metadata)) {
-    if (isSerializable(value)) {
-      sanitized[key] = value;
-    }
-  }
-  return sanitized;
-}
-function isSerializable(value) {
-  try {
-    JSON.stringify(value);
-    return true;
-  } catch {
-    return false;
-  }
-}
-function omitKeys(obj, keysToOmit) {
-  return Object.fromEntries(Object.entries(obj).filter(([key]) => !keysToOmit.includes(key)));
-}
-
 // src/ai-tracing/context.ts
 var AGENT_GETTERS = ["getAgent", "getAgentById"];
 var AGENT_METHODS_TO_WRAP = ["generate", "stream", "generateVNext", "streamVNext"];
@@ -1318,10 +1391,12 @@ function wrapMastra(mastra, tracingContext) {
               return wrapWorkflow(workflow, tracingContext);
             };
           }
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         } catch (error) {
           console.warn("AI Tracing: Failed to wrap method, falling back to original", error);
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         }
       }
     });
@@ -1346,10 +1421,12 @@ function wrapAgent(agent, tracingContext) {
               });
             };
           }
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         } catch (error) {
           console.warn("AI Tracing: Failed to wrap agent method, falling back to original", error);
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         }
       }
     });
@@ -1374,10 +1451,12 @@ function wrapWorkflow(workflow, tracingContext) {
               });
             };
           }
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         } catch (error) {
           console.warn("AI Tracing: Failed to wrap workflow method, falling back to original", error);
-          return target[prop];
+          const value = target[prop];
+          return typeof value === "function" ? value.bind(target) : value;
         }
       }
     });
@@ -1400,18 +1479,21 @@ exports.clearAITracingRegistry = clearAITracingRegistry;
 exports.getAITracing = getAITracing;
 exports.getAllAITracing = getAllAITracing;
 exports.getDefaultAITracing = getDefaultAITracing;
+exports.getOrCreateSpan = getOrCreateSpan;
 exports.getSelectedAITracing = getSelectedAITracing;
 exports.hasAITracing = hasAITracing;
-exports.isSerializable = isSerializable;
 exports.omitKeys = omitKeys;
 exports.registerAITracing = registerAITracing;
-exports.sanitizeMetadata = sanitizeMetadata;
+exports.selectFields = selectFields;
 exports.setAITracingSelector = setAITracingSelector;
 exports.setupAITracing = setupAITracing;
+exports.shallowClean = shallowClean;
+exports.shallowCleanArray = shallowCleanArray;
+exports.shallowCleanObject = shallowCleanObject;
 exports.shutdownAITracingRegistry = shutdownAITracingRegistry;
 exports.unregisterAITracing = unregisterAITracing;
 exports.wrapAgent = wrapAgent;
 exports.wrapMastra = wrapMastra;
 exports.wrapWorkflow = wrapWorkflow;
-//# sourceMappingURL=chunk-4DKPMUAC.cjs.map
-//# sourceMappingURL=chunk-4DKPMUAC.cjs.map
+//# sourceMappingURL=chunk-I7OAONIW.cjs.map
+//# sourceMappingURL=chunk-I7OAONIW.cjs.map