@mastra/core 0.1.27-alpha.57 → 0.1.27-alpha.58

package/dist/core.cjs.development.js

@@ -30,6 +30,7 @@ var radash = require('radash');
 var sift = require('sift');
 var xstate = require('xstate');
 var voyageAiProvider = require('voyage-ai-provider');
+var cohereAi = require('cohere-ai');
 
 function _arrayLikeToArray(r, a) {
   (null == a || a > r.length) && (a = r.length);
@@ -6559,6 +6560,93 @@ function _embed() {
   return _embed.apply(this, arguments);
 }
 
+// Cohere implementation
+var CohereRelevanceScorer = /*#__PURE__*/function () {
+  function CohereRelevanceScorer(apiKey, model) {
+    this.client = void 0;
+    this.model = void 0;
+    this.client = new cohereAi.CohereClient({
+      token: apiKey
+    });
+    this.model = model || 'rerank-v3.5';
+  }
+  var _proto = CohereRelevanceScorer.prototype;
+  _proto.getRelevanceScore = /*#__PURE__*/function () {
+    var _getRelevanceScore = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(query, text) {
+      var response;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            _context.next = 2;
+            return this.client.rerank({
+              query: query,
+              documents: [text],
+              model: this.model,
+              topN: 1
+            });
+          case 2:
+            response = _context.sent;
+            return _context.abrupt("return", response.results[0].relevanceScore);
+          case 4:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function getRelevanceScore(_x, _x2) {
+      return _getRelevanceScore.apply(this, arguments);
+    }
+    return getRelevanceScore;
+  }();
+  return CohereRelevanceScorer;
+}();
+
+// Helper function used by providers
+function createSimilarityPrompt(query, text) {
+  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):";
+}
+
+// Mastra Agent implementation
+var MastraAgentRelevanceScorer = /*#__PURE__*/function () {
+  function MastraAgentRelevanceScorer(provider, name) {
+    this.agent = void 0;
+    var modelConfig = {
+      provider: provider,
+      name: name
+    };
+    this.agent = new exports.Agent({
+      name: "Relevance Scorer " + provider + " " + name,
+      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.",
+      model: modelConfig
+    });
+  }
+  var _proto = MastraAgentRelevanceScorer.prototype;
+  _proto.getRelevanceScore = /*#__PURE__*/function () {
+    var _getRelevanceScore = /*#__PURE__*/_asyncToGenerator(/*#__PURE__*/_regeneratorRuntime().mark(function _callee(query, text) {
+      var prompt, response;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            prompt = createSimilarityPrompt(query, text);
+            _context.next = 3;
+            return this.agent.generate(prompt);
+          case 3:
+            response = _context.sent;
+            return _context.abrupt("return", parseFloat(response.text));
+          case 5:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, this);
+    }));
+    function getRelevanceScore(_x, _x2) {
+      return _getRelevanceScore.apply(this, arguments);
+    }
+    return getRelevanceScore;
+  }();
+  return MastraAgentRelevanceScorer;
+}();
+
 exports.MastraTTS = /*#__PURE__*/function (_MastraBase) {
   function MastraTTS(_ref) {
     var _this;
@@ -6589,11 +6677,13 @@ exports.MastraTTS = /*#__PURE__*/tslib.__decorate([/*#__PURE__*/InstrumentClass(
 }), /*#__PURE__*/tslib.__metadata("design:paramtypes", [Object])], exports.MastraTTS);
 
 exports.BaseLogger = BaseLogger;
+exports.CohereRelevanceScorer = CohereRelevanceScorer;
 exports.ConsoleLogger = ConsoleLogger;
 exports.FileLogger = FileLogger;
 exports.InstrumentClass = InstrumentClass;
 exports.Integration = Integration;
 exports.LogLevel = LogLevel;
+exports.MastraAgentRelevanceScorer = MastraAgentRelevanceScorer;
 exports.MastraEngine = MastraEngine;
 exports.MastraMemory = MastraMemory;
 exports.MastraVector = MastraVector;
@@ -6609,6 +6699,7 @@ exports.UpstashRedisLogger = UpstashRedisLogger;
 exports.Workflow = Workflow;
 exports.combineLoggers = combineLoggers;
 exports.createLogger = createLogger;
+exports.createSimilarityPrompt = createSimilarityPrompt;
 exports.createStep = createStep;
 exports.createSync = createSync;
 exports.createTool = createTool;