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

package/dist/ai/index.js

@@ -2569,6 +2569,7 @@ var resolveRAGHybridSearchOptions = (retrieval) => {
     fusion: retrieval.fusion ?? "rrf",
     fusionConstant: Math.max(1, Math.floor(retrieval.fusionConstant ?? DEFAULT_FUSION_CONSTANT)),
     lexicalTopK: retrieval.lexicalTopK,
+    maxResultsPerSource: typeof retrieval.maxResultsPerSource === "number" ? Math.max(1, Math.floor(retrieval.maxResultsPerSource)) : undefined,
     lexicalWeight: Math.max(0, retrieval.lexicalWeight ?? 2),
     mode: retrieval.mode ?? "vector",
     vectorWeight: Math.max(0, retrieval.vectorWeight ?? 1)
@@ -4314,6 +4315,27 @@ var mergeQueryResults = (results) => {
     return left.chunkId.localeCompare(right.chunkId);
   });
 };
+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) => {
+  if (typeof maxResultsPerSource !== "number") {
+    return results;
+  }
+  const limited = [];
+  const counts = new Map;
+  for (const result of results) {
+    const key = getRAGSourceDiversityKey(result);
+    const count = counts.get(key) ?? 0;
+    if (count >= maxResultsPerSource) {
+      continue;
+    }
+    counts.set(key, count + 1);
+    limited.push(result);
+  }
+  return limited;
+};
 var weightQueryResults = (results, queryIndex) => {
   if (queryIndex === 0) {
     return results;
@@ -4369,6 +4391,7 @@ var createRAGCollection = (options) => {
           candidateTopK,
           hasQueryTransform,
           hasReranker,
+          maxResultsPerSource: retrieval.maxResultsPerSource ?? null,
           mode: retrieval.mode,
           runLexical,
           runVector,
@@ -4486,7 +4509,18 @@ var createRAGCollection = (options) => {
       },
       stage: "rerank"
     });
-    const limited = reranked.slice(0, topK);
+    const diversified = applyRAGSourceDiversity(reranked, retrieval.maxResultsPerSource);
+    if (typeof retrieval.maxResultsPerSource === "number") {
+      steps.push({
+        count: diversified.length,
+        label: "Balanced candidates across sources",
+        metadata: {
+          maxResultsPerSource: retrieval.maxResultsPerSource
+        },
+        stage: "source_balance"
+      });
+    }
+    const limited = diversified.slice(0, topK);
     if (typeof input.scoreThreshold !== "number") {
       steps.push({
         count: limited.length,
@@ -4501,13 +4535,14 @@ var createRAGCollection = (options) => {
         trace: {
           candidateTopK,
           lexicalTopK,
+          maxResultsPerSource: retrieval.maxResultsPerSource,
           mode: retrieval.mode,
           query: input.query,
           resultCounts: {
             final: limited.length,
             fused: results.length,
             lexical: lexicalResults.length,
-            reranked: reranked.length,
+            reranked: diversified.length,
             vector: vectorResults.length
           },
           runLexical,
@@ -4542,13 +4577,14 @@ var createRAGCollection = (options) => {
       trace: {
         candidateTopK,
         lexicalTopK,
+        maxResultsPerSource: retrieval.maxResultsPerSource,
         mode: retrieval.mode,
         query: input.query,
         resultCounts: {
           final: filtered.length,
           fused: results.length,
           lexical: lexicalResults.length,
-          reranked: reranked.length,
+          reranked: diversified.length,
           vector: vectorResults.length
         },
         runLexical,
@@ -12505,5 +12541,5 @@ export {
   aiChat
 };
 
-//# debugId=7DAB512A9E25D04564756E2164756E21
+//# debugId=FD250EE2F99EB57964756E2164756E21
 //# sourceMappingURL=index.js.map