@absolutejs/absolute 0.19.0-beta.552 → 0.19.0-beta.553

package/dist/ai/index.js

@@ -2551,6 +2551,7 @@ var resolveRAGHybridSearchOptions = (retrieval) => {
       fusionConstant: DEFAULT_FUSION_CONSTANT,
       lexicalTopK: undefined,
       lexicalWeight: 2,
+      sourceBalanceStrategy: "cap",
       mode: "vector",
       vectorWeight: 1
     };
@@ -2561,6 +2562,7 @@ var resolveRAGHybridSearchOptions = (retrieval) => {
       fusionConstant: DEFAULT_FUSION_CONSTANT,
       lexicalTopK: undefined,
       lexicalWeight: 2,
+      sourceBalanceStrategy: "cap",
       mode: retrieval,
       vectorWeight: 1
     };
@@ -2572,6 +2574,7 @@ var resolveRAGHybridSearchOptions = (retrieval) => {
     maxResultsPerSource: typeof retrieval.maxResultsPerSource === "number" ? Math.max(1, Math.floor(retrieval.maxResultsPerSource)) : undefined,
     lexicalWeight: Math.max(0, retrieval.lexicalWeight ?? 2),
     mode: retrieval.mode ?? "vector",
+    sourceBalanceStrategy: retrieval.sourceBalanceStrategy === "round_robin" ? "round_robin" : "cap",
     vectorWeight: Math.max(0, retrieval.vectorWeight ?? 1)
   };
 };
@@ -4319,10 +4322,47 @@ var getRAGSourceDiversityKey = (result) => {
   const documentId = typeof result.metadata?.documentId === "string" ? result.metadata.documentId : undefined;
   return result.source ?? documentId ?? result.title ?? result.chunkId;
 };
-var applyRAGSourceDiversity = (results, maxResultsPerSource) => {
+var applyRAGSourceDiversity = (results, maxResultsPerSource, strategy = "cap") => {
   if (typeof maxResultsPerSource !== "number") {
     return results;
   }
+  if (strategy === "round_robin") {
+    const grouped = new Map;
+    const order = [];
+    for (const result of results) {
+      const key = getRAGSourceDiversityKey(result);
+      const entries = grouped.get(key);
+      if (entries) {
+        entries.push(result);
+        continue;
+      }
+      grouped.set(key, [result]);
+      order.push(key);
+    }
+    const counts2 = new Map;
+    const balanced = [];
+    let remaining = true;
+    while (remaining) {
+      remaining = false;
+      for (const key of order) {
+        const entries = grouped.get(key);
+        if (!entries || entries.length === 0) {
+          continue;
+        }
+        if ((counts2.get(key) ?? 0) >= maxResultsPerSource) {
+          continue;
+        }
+        const next = entries.shift();
+        if (!next) {
+          continue;
+        }
+        counts2.set(key, (counts2.get(key) ?? 0) + 1);
+        balanced.push(next);
+        remaining = true;
+      }
+    }
+    return balanced;
+  }
   const limited = [];
   const counts = new Map;
   for (const result of results) {
@@ -4395,6 +4435,7 @@ var createRAGCollection = (options) => {
           mode: retrieval.mode,
           runLexical,
           runVector,
+          sourceBalanceStrategy: retrieval.sourceBalanceStrategy,
           topK
         },
         stage: "input"
@@ -4509,13 +4550,14 @@ var createRAGCollection = (options) => {
       },
       stage: "rerank"
     });
-    const diversified = applyRAGSourceDiversity(reranked, retrieval.maxResultsPerSource);
+    const diversified = applyRAGSourceDiversity(reranked, retrieval.maxResultsPerSource, retrieval.sourceBalanceStrategy);
     if (typeof retrieval.maxResultsPerSource === "number") {
       steps.push({
         count: diversified.length,
         label: "Balanced candidates across sources",
         metadata: {
-          maxResultsPerSource: retrieval.maxResultsPerSource
+          maxResultsPerSource: retrieval.maxResultsPerSource,
+          strategy: retrieval.sourceBalanceStrategy
         },
         stage: "source_balance"
       });
@@ -4547,6 +4589,7 @@ var createRAGCollection = (options) => {
           },
           runLexical,
           runVector,
+          sourceBalanceStrategy: retrieval.sourceBalanceStrategy,
           steps,
           topK,
           transformedQuery: transformed.query,
@@ -4590,6 +4633,7 @@ var createRAGCollection = (options) => {
         runLexical,
         runVector,
         scoreThreshold,
+        sourceBalanceStrategy: retrieval.sourceBalanceStrategy,
         steps,
         topK,
         transformedQuery: transformed.query,
@@ -12541,5 +12585,5 @@ export {
   aiChat
 };
 
-//# debugId=FD250EE2F99EB57964756E2164756E21
+//# debugId=BD4F3B903BF6366564756E2164756E21
 //# sourceMappingURL=index.js.map