@mastra/core 0.24.1 → 0.24.2-alpha.0

package/dist/ai-tracing/index.cjs

@@ -1,148 +1,148 @@
 'use strict';
 
-var chunkO5PROZMW_cjs = require('../chunk-O5PROZMW.cjs');
+var chunkLZFCR2SE_cjs = require('../chunk-LZFCR2SE.cjs');
 
 
 
 Object.defineProperty(exports, "AISpanType", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.AISpanType; }
+  get: function () { return chunkLZFCR2SE_cjs.AISpanType; }
 });
 Object.defineProperty(exports, "AITracingEventType", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.AITracingEventType; }
+  get: function () { return chunkLZFCR2SE_cjs.AITracingEventType; }
 });
 Object.defineProperty(exports, "BaseAISpan", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.BaseAISpan; }
+  get: function () { return chunkLZFCR2SE_cjs.BaseAISpan; }
 });
 Object.defineProperty(exports, "BaseAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.BaseAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.BaseAITracing; }
 });
 Object.defineProperty(exports, "BaseExporter", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.BaseExporter; }
+  get: function () { return chunkLZFCR2SE_cjs.BaseExporter; }
 });
 Object.defineProperty(exports, "CloudExporter", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.CloudExporter; }
+  get: function () { return chunkLZFCR2SE_cjs.CloudExporter; }
 });
 Object.defineProperty(exports, "ConsoleExporter", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.ConsoleExporter; }
+  get: function () { return chunkLZFCR2SE_cjs.ConsoleExporter; }
 });
 Object.defineProperty(exports, "DefaultAISpan", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.DefaultAISpan; }
+  get: function () { return chunkLZFCR2SE_cjs.DefaultAISpan; }
 });
 Object.defineProperty(exports, "DefaultAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.DefaultAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.DefaultAITracing; }
 });
 Object.defineProperty(exports, "DefaultExporter", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.DefaultExporter; }
+  get: function () { return chunkLZFCR2SE_cjs.DefaultExporter; }
 });
 Object.defineProperty(exports, "InternalSpans", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.InternalSpans; }
+  get: function () { return chunkLZFCR2SE_cjs.InternalSpans; }
 });
 Object.defineProperty(exports, "ModelSpanTracker", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.ModelSpanTracker; }
+  get: function () { return chunkLZFCR2SE_cjs.ModelSpanTracker; }
 });
 Object.defineProperty(exports, "NoOpAISpan", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.NoOpAISpan; }
+  get: function () { return chunkLZFCR2SE_cjs.NoOpAISpan; }
 });
 Object.defineProperty(exports, "SamplingStrategyType", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.SamplingStrategyType; }
+  get: function () { return chunkLZFCR2SE_cjs.SamplingStrategyType; }
 });
 Object.defineProperty(exports, "SensitiveDataFilter", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.SensitiveDataFilter; }
+  get: function () { return chunkLZFCR2SE_cjs.SensitiveDataFilter; }
 });
 Object.defineProperty(exports, "clearAITracingRegistry", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.clearAITracingRegistry; }
+  get: function () { return chunkLZFCR2SE_cjs.clearAITracingRegistry; }
 });
 Object.defineProperty(exports, "deepClean", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.deepClean; }
+  get: function () { return chunkLZFCR2SE_cjs.deepClean; }
 });
 Object.defineProperty(exports, "getAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.getAITracing; }
 });
 Object.defineProperty(exports, "getAllAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getAllAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.getAllAITracing; }
 });
 Object.defineProperty(exports, "getDefaultAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getDefaultAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.getDefaultAITracing; }
 });
 Object.defineProperty(exports, "getNestedValue", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getNestedValue; }
+  get: function () { return chunkLZFCR2SE_cjs.getNestedValue; }
 });
 Object.defineProperty(exports, "getOrCreateSpan", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getOrCreateSpan; }
+  get: function () { return chunkLZFCR2SE_cjs.getOrCreateSpan; }
 });
 Object.defineProperty(exports, "getSelectedAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getSelectedAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.getSelectedAITracing; }
 });
 Object.defineProperty(exports, "getValidTraceId", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.getValidTraceId; }
+  get: function () { return chunkLZFCR2SE_cjs.getValidTraceId; }
 });
 Object.defineProperty(exports, "hasAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.hasAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.hasAITracing; }
 });
 Object.defineProperty(exports, "isMastra", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.isMastra; }
+  get: function () { return chunkLZFCR2SE_cjs.isMastra; }
 });
 Object.defineProperty(exports, "omitKeys", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.omitKeys; }
+  get: function () { return chunkLZFCR2SE_cjs.omitKeys; }
 });
 Object.defineProperty(exports, "registerAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.registerAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.registerAITracing; }
 });
 Object.defineProperty(exports, "selectFields", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.selectFields; }
+  get: function () { return chunkLZFCR2SE_cjs.selectFields; }
 });
 Object.defineProperty(exports, "setNestedValue", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.setNestedValue; }
+  get: function () { return chunkLZFCR2SE_cjs.setNestedValue; }
 });
 Object.defineProperty(exports, "setSelector", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.setSelector; }
+  get: function () { return chunkLZFCR2SE_cjs.setSelector; }
 });
 Object.defineProperty(exports, "setupAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.setupAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.setupAITracing; }
 });
 Object.defineProperty(exports, "shutdownAITracingRegistry", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.shutdownAITracingRegistry; }
+  get: function () { return chunkLZFCR2SE_cjs.shutdownAITracingRegistry; }
 });
 Object.defineProperty(exports, "unregisterAITracing", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.unregisterAITracing; }
+  get: function () { return chunkLZFCR2SE_cjs.unregisterAITracing; }
 });
 Object.defineProperty(exports, "wrapMastra", {
   enumerable: true,
-  get: function () { return chunkO5PROZMW_cjs.wrapMastra; }
+  get: function () { return chunkLZFCR2SE_cjs.wrapMastra; }
 });
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/ai-tracing/index.js

@@ -1,3 +1,3 @@
-export { AISpanType, AITracingEventType, BaseAISpan, BaseAITracing, BaseExporter, CloudExporter, ConsoleExporter, DefaultAISpan, DefaultAITracing, DefaultExporter, InternalSpans, ModelSpanTracker, NoOpAISpan, SamplingStrategyType, SensitiveDataFilter, clearAITracingRegistry, deepClean, getAITracing, getAllAITracing, getDefaultAITracing, getNestedValue, getOrCreateSpan, getSelectedAITracing, getValidTraceId, hasAITracing, isMastra, omitKeys, registerAITracing, selectFields, setNestedValue, setSelector, setupAITracing, shutdownAITracingRegistry, unregisterAITracing, wrapMastra } from '../chunk-MBMPTC7F.js';
+export { AISpanType, AITracingEventType, BaseAISpan, BaseAITracing, BaseExporter, CloudExporter, ConsoleExporter, DefaultAISpan, DefaultAITracing, DefaultExporter, InternalSpans, ModelSpanTracker, NoOpAISpan, SamplingStrategyType, SensitiveDataFilter, clearAITracingRegistry, deepClean, getAITracing, getAllAITracing, getDefaultAITracing, getNestedValue, getOrCreateSpan, getSelectedAITracing, getValidTraceId, hasAITracing, isMastra, omitKeys, registerAITracing, selectFields, setNestedValue, setSelector, setupAITracing, shutdownAITracingRegistry, unregisterAITracing, wrapMastra } from '../chunk-ZIHEKHUB.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/{chunk-EDVPIVAB.js → chunk-4OPPKDPO.js}

@@ -1,4 +1,4 @@
-import { Agent } from './chunk-ZVAWVX5J.js';
+import { Agent } from './chunk-SHPLHQFH.js';
 
 // src/relevance/cohere/index.ts
 var CohereRelevanceScorer = class {
@@ -83,5 +83,5 @@ Always return just the number, no explanation.`,
 };
 
 export { CohereRelevanceScorer, MastraAgentRelevanceScorer, createSimilarityPrompt };
-//# sourceMappingURL=chunk-EDVPIVAB.js.map
-//# sourceMappingURL=chunk-EDVPIVAB.js.map
+//# sourceMappingURL=chunk-4OPPKDPO.js.map
+//# sourceMappingURL=chunk-4OPPKDPO.js.map

package/dist/{chunk-EDVPIVAB.js.map → chunk-4OPPKDPO.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/relevance/cohere/index.ts","../src/relevance/relevance-score-provider.ts","../src/relevance/mastra-agent/index.ts"],"names":[],"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,EAA0B;AAClD,IAAA,IAAA,CAAK,KAAA,GAAQ,IAAI,KAAA,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,SAAS,MAAM,CAAA;AACrD,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B,CAAA,MAAO;AACL,MAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,KAAA,CAAM,eAAe,MAAM,CAAA;AAC3D,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B;AACA,IAAA,OAAO,WAAW,QAAQ,CAAA;AAAA,EAC5B;AACF","file":"chunk-EDVPIVAB.js","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 { MastraModelConfig } 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: MastraModelConfig) {\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.generate(prompt);\n      response = responseText.text;\n    } else {\n      const responseText = await this.agent.generateLegacy(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":[],"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,EAA0B;AAClD,IAAA,IAAA,CAAK,KAAA,GAAQ,IAAI,KAAA,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,SAAS,MAAM,CAAA;AACrD,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B,CAAA,MAAO;AACL,MAAA,MAAM,YAAA,GAAe,MAAM,IAAA,CAAK,KAAA,CAAM,eAAe,MAAM,CAAA;AAC3D,MAAA,QAAA,GAAW,YAAA,CAAa,IAAA;AAAA,IAC1B;AACA,IAAA,OAAO,WAAW,QAAQ,CAAA;AAAA,EAC5B;AACF","file":"chunk-4OPPKDPO.js","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 { MastraModelConfig } 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: MastraModelConfig) {\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.generate(prompt);\n      response = responseText.text;\n    } else {\n      const responseText = await this.agent.generateLegacy(prompt);\n      response = responseText.text;\n    }\n    return parseFloat(response);\n  }\n}\n"]}

package/dist/{chunk-BCFLCIAX.cjs → chunk-B6RCPK2E.cjs}

@@ -2204,7 +2204,7 @@ function saveLastRefreshTimeToDisk(date) {
 }
 function getPackageRoot() {
   try {
-    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-BCFLCIAX.cjs', document.baseURI).href)) || "file://");
+    const require2 = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-B6RCPK2E.cjs', document.baseURI).href)) || "file://");
     const packageJsonPath = require2.resolve("@mastra/core/package.json");
     return path__default.default.dirname(packageJsonPath);
   } catch {
@@ -2474,6 +2474,137 @@ var autoRefreshEnabled = process.env.MASTRA_AUTO_REFRESH_PROVIDERS === "true" ||
 if (autoRefreshEnabled) {
   GatewayRegistry.getInstance({ useDynamicLoading: isDev }).startAutoRefresh();
 }
+var AISDKV5LanguageModel = class {
+  /**
+   * The language model must specify which language model interface version it implements.
+   */
+  specificationVersion = "v2";
+  /**
+   * Name of the provider for logging purposes.
+   */
+  provider;
+  /**
+   * Provider-specific model ID for logging purposes.
+   */
+  modelId;
+  /**
+   * Supported URL patterns by media type for the provider.
+   *
+   * The keys are media type patterns or full media types (e.g. `*\/*` for everything, `audio/*`, `video/*`, or `application/pdf`).
+   * and the values are arrays of regular expressions that match the URL paths.
+   * The matching should be against lower-case URLs.
+   * Matched URLs are supported natively by the model and are not downloaded.
+   * @returns A map of supported URL patterns by media type (as a promise or a plain object).
+   */
+  supportedUrls;
+  #model;
+  constructor(config) {
+    this.#model = config;
+    this.provider = this.#model.provider;
+    this.modelId = this.#model.modelId;
+    this.supportedUrls = this.#model.supportedUrls;
+  }
+  async doGenerate(options) {
+    const result = await this.#model.doGenerate(options);
+    return {
+      request: result.request,
+      response: result.response,
+      stream: new ReadableStream({
+        start(controller) {
+          controller.enqueue({ type: "stream-start", warnings: result.warnings });
+          controller.enqueue({
+            type: "response-metadata",
+            id: result.response?.id,
+            modelId: result.response?.modelId,
+            timestamp: result.response?.timestamp
+          });
+          for (const message of result.content) {
+            if (message.type === "tool-call") {
+              const toolCall = message;
+              controller.enqueue({
+                type: "tool-input-start",
+                id: toolCall.toolCallId,
+                toolName: toolCall.toolName
+              });
+              controller.enqueue({
+                type: "tool-input-delta",
+                id: toolCall.toolCallId,
+                delta: toolCall.input
+              });
+              controller.enqueue({
+                type: "tool-input-end",
+                id: toolCall.toolCallId
+              });
+              controller.enqueue(toolCall);
+            } else if (message.type === "text") {
+              const text = message;
+              const id = `msg_${crypto.randomUUID()}`;
+              controller.enqueue({
+                type: "text-start",
+                id,
+                providerMetadata: text.providerMetadata
+              });
+              controller.enqueue({
+                type: "text-delta",
+                id,
+                delta: text.text
+              });
+              controller.enqueue({
+                type: "text-end",
+                id
+              });
+            } else if (message.type === "reasoning") {
+              const id = `reasoning_${crypto.randomUUID()}`;
+              const reasoning = message;
+              controller.enqueue({
+                type: "reasoning-start",
+                id,
+                providerMetadata: reasoning.providerMetadata
+              });
+              controller.enqueue({
+                type: "reasoning-delta",
+                id,
+                delta: reasoning.text,
+                providerMetadata: reasoning.providerMetadata
+              });
+              controller.enqueue({
+                type: "reasoning-end",
+                id,
+                providerMetadata: reasoning.providerMetadata
+              });
+            } else if (message.type === "file") {
+              const file = message;
+              controller.enqueue({
+                type: "file",
+                mediaType: file.mediaType,
+                data: file.data
+              });
+            } else if (message.type === "source") {
+              const source = message;
+              controller.enqueue({
+                type: "source",
+                id: source.id,
+                sourceType: source.sourceType,
+                title: source.title,
+                providerMetadata: source.providerMetadata
+              });
+            }
+          }
+          controller.enqueue({
+            type: "finish",
+            finishReason: result.finishReason,
+            usage: result.usage,
+            providerMetadata: result.providerMetadata
+          });
+          controller.close();
+        }
+      })
+    };
+  }
+  async doStream(options) {
+    return await this.#model.doStream(options);
+  }
+};
 
 // src/llm/model/gateways/index.ts
 function findGatewayForModel(gatewayId, gateways2) {
@@ -2540,10 +2671,33 @@ var ModelRouterLanguageModel = class _ModelRouterLanguageModel {
     this.modelId = parsedConfig.id;
     this.config = parsedConfig;
   }
-  async doGenerate() {
-    throw new Error(
-      "doGenerate is not supported by Mastra model router. Mastra only uses streaming (doStream) for all LLM calls."
-    );
+  async doGenerate(options) {
+    let apiKey;
+    try {
+      if (this.config.url) {
+        apiKey = this.config.apiKey || "";
+      } else {
+        apiKey = this.config.apiKey || await this.gateway.getApiKey(this.config.routerId);
+      }
+    } catch (error) {
+      return {
+        stream: new ReadableStream({
+          start(controller) {
+            controller.enqueue({
+              type: "error",
+              error
+            });
+            controller.close();
+          }
+        })
+      };
+    }
+    const model = await this.resolveLanguageModel({
+      apiKey,
+      ...chunkIJRERAWQ_cjs.parseModelRouterId(this.config.routerId, this.gateway.prefix)
+    });
+    const aiSDKV5Model = new AISDKV5LanguageModel(model);
+    return aiSDKV5Model.doGenerate(options);
   }
   async doStream(options) {
     let apiKey;
@@ -2561,6 +2715,7 @@ var ModelRouterLanguageModel = class _ModelRouterLanguageModel {
               type: "error",
               error
             });
+            controller.close();
           }
         })
       };
@@ -2569,7 +2724,8 @@ var ModelRouterLanguageModel = class _ModelRouterLanguageModel {
       apiKey,
       ...chunkIJRERAWQ_cjs.parseModelRouterId(this.config.routerId, this.gateway.prefix)
     });
-    return model.doStream(options);
+    const aiSDKV5Model = new AISDKV5LanguageModel(model);
+    return aiSDKV5Model.doStream(options);
   }
   async resolveLanguageModel({
     modelId,
@@ -2606,19 +2762,21 @@ function isOpenAICompatibleObjectConfig(modelConfig) {
   return false;
 }
 async function resolveModelConfig(modelConfig, runtimeContext = new chunkGPWMM745_cjs.RuntimeContext(), mastra) {
+  if (typeof modelConfig === "function") {
+    modelConfig = await modelConfig({ runtimeContext, mastra });
+  }
+  if (modelConfig instanceof ModelRouterLanguageModel || modelConfig instanceof AISDKV5LanguageModel) {
+    return modelConfig;
+  }
   if (typeof modelConfig === "object" && "specificationVersion" in modelConfig) {
+    if (modelConfig.specificationVersion === "v2") {
+      return new AISDKV5LanguageModel(modelConfig);
+    }
     return modelConfig;
   }
   if (typeof modelConfig === "string" || isOpenAICompatibleObjectConfig(modelConfig)) {
     return new ModelRouterLanguageModel(modelConfig);
   }
-  if (typeof modelConfig === "function") {
-    const fromDynamic = await modelConfig({ runtimeContext, mastra });
-    if (typeof fromDynamic === "string" || isOpenAICompatibleObjectConfig(fromDynamic)) {
-      return new ModelRouterLanguageModel(fromDynamic);
-    }
-    return fromDynamic;
-  }
   throw new Error("Invalid model configuration provided");
 }
 var ModelRouterEmbeddingModel = class {
@@ -2728,5 +2886,5 @@ exports.PROVIDER_REGISTRY = PROVIDER_REGISTRY;
 exports.getProviderConfig = getProviderConfig;
 exports.parseModelString = parseModelString;
 exports.resolveModelConfig = resolveModelConfig;
-//# sourceMappingURL=chunk-BCFLCIAX.cjs.map
-//# sourceMappingURL=chunk-BCFLCIAX.cjs.map
+//# sourceMappingURL=chunk-B6RCPK2E.cjs.map
+//# sourceMappingURL=chunk-B6RCPK2E.cjs.map