@absolutejs/absolute 0.19.0-beta.550 → 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,
@@ -6370,10 +6406,14 @@ var buildRAGGroundingProviderOverviewPresentation = (input) => {
     summary: winnerEntry ? `passing ${formatEvaluationPassingRate(winnerEntry.response.passingRate)} \xB7 citation f1 ${winnerEntry.response.summary.averageCitationF1.toFixed(3)} \xB7 resolved ${formatEvaluationPassingRate(winnerEntry.response.summary.averageResolvedCitationRate)}` : "Stored workflow evaluation"
   };
 };
-var buildRAGQualityOverviewPresentation = (input) => ({
+var buildRAGRetrievalOverviewPresentation = (comparison) => ({
+  rows: buildRAGRetrievalComparisonOverviewPresentation(comparison).rows
+});
+var buildRAGRerankerOverviewPresentation = (comparison) => ({
+  rows: buildRAGRerankerComparisonOverviewPresentation(comparison).rows
+});
+var buildRAGGroundingOverviewPresentation = (input) => ({
   rows: [
-    ...buildRAGRetrievalComparisonOverviewPresentation(input.retrievalComparison).rows,
-    ...buildRAGRerankerComparisonOverviewPresentation(input.rerankerComparison).rows,
     {
       label: "Grounding",
       value: formatGroundingHistorySummaryValue(input.groundingEvaluation)
@@ -6386,6 +6426,16 @@ var buildRAGQualityOverviewPresentation = (input) => ({
     ]
   ]
 });
+var buildRAGQualityOverviewPresentation = (input) => ({
+  rows: [
+    ...buildRAGRetrievalOverviewPresentation(input.retrievalComparison).rows,
+    ...buildRAGRerankerOverviewPresentation(input.rerankerComparison).rows,
+    ...buildRAGGroundingOverviewPresentation({
+      groundingEvaluation: input.groundingEvaluation,
+      groundingProviderOverview: input.groundingProviderOverview
+    }).rows
+  ]
+});
 var buildRAGGroundingProviderCaseComparisonPresentations = (comparisons) => comparisons.map((comparison) => {
   const resolveLabel = (key) => comparison.entries.find((entry) => entry.providerKey === key)?.label ?? key ?? "n/a";
   return {
@@ -12444,8 +12494,10 @@ export {
   buildRAGSourceSummaries,
   buildRAGSourceGroups,
   buildRAGRetrievalTracePresentation,
+  buildRAGRetrievalOverviewPresentation,
   buildRAGRetrievalComparisonPresentations,
   buildRAGRetrievalComparisonOverviewPresentation,
+  buildRAGRerankerOverviewPresentation,
   buildRAGRerankerComparisonPresentations,
   buildRAGRerankerComparisonOverviewPresentation,
   buildRAGReadinessPresentation,
@@ -12455,6 +12507,7 @@ export {
   buildRAGGroundingProviderPresentations,
   buildRAGGroundingProviderOverviewPresentation,
   buildRAGGroundingProviderCaseComparisonPresentations,
+  buildRAGGroundingOverviewPresentation,
   buildRAGGroundedAnswer,
   buildRAGEvaluationRunDiff,
   buildRAGEvaluationResponse,
@@ -12488,5 +12541,5 @@ export {
   aiChat
 };
 
-//# debugId=87BF178FDB66370264756E2164756E21
+//# debugId=FD250EE2F99EB57964756E2164756E21
 //# sourceMappingURL=index.js.map