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

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,
@@ -5765,8 +5809,11 @@ var summarizeRetrievalTraces = (traces) => {
   }
   const totalCases = traces.length;
   const modeSet = new Set;
+  const sourceBalanceStrategySet = new Set;
   let vectorCases = 0;
   let lexicalCases = 0;
+  let balancedCases = 0;
+  let roundRobinCases = 0;
   let transformedCases = 0;
   let variantCases = 0;
   let finalCountSum = 0;
@@ -5776,12 +5823,19 @@ var summarizeRetrievalTraces = (traces) => {
   let lexicalTopKSum = 0;
   for (const trace of traces) {
     modeSet.add(trace.mode);
+    sourceBalanceStrategySet.add(trace.sourceBalanceStrategy ?? "cap");
     if (trace.runVector) {
       vectorCases += 1;
     }
     if (trace.runLexical) {
       lexicalCases += 1;
     }
+    if (typeof trace.maxResultsPerSource === "number") {
+      balancedCases += 1;
+      if (trace.sourceBalanceStrategy === "round_robin") {
+        roundRobinCases += 1;
+      }
+    }
     if (trace.transformedQuery !== trace.query) {
       transformedCases += 1;
     }
@@ -5799,9 +5853,12 @@ var summarizeRetrievalTraces = (traces) => {
     averageFinalCount: roundTraceAverage(finalCountSum, totalCases),
     averageLexicalCount: roundTraceAverage(lexicalCountSum, totalCases),
     averageLexicalTopK: roundTraceAverage(lexicalTopKSum, totalCases),
+    balancedCases,
     averageVectorCount: roundTraceAverage(vectorCountSum, totalCases),
     lexicalCases,
     modes: Array.from(modeSet),
+    roundRobinCases,
+    sourceBalanceStrategies: Array.from(sourceBalanceStrategySet),
     stageCounts: buildTraceStageCounts(traces),
     totalCases,
     transformedCases,
@@ -6137,7 +6194,7 @@ var buildEvaluationCaseTraceSnapshot = ({
 }) => {
   const stageCounts = currentTrace ? buildTraceStageCounts([currentTrace]) : {};
   const previousStageCounts = previousTrace?.stageCounts ?? {};
-  const traceChange = !previousTrace ? currentTrace ? "new" : "unchanged" : previousTrace.traceMode !== currentTrace?.mode || previousTrace.transformedQuery !== (currentTrace?.transformedQuery || undefined) || previousTrace.variantQueries.join("|") !== (currentTrace?.variantQueries ?? []).join("|") || previousTrace.finalCount !== (currentTrace?.resultCounts.final ?? 0) || previousTrace.vectorCount !== (currentTrace?.resultCounts.vector ?? 0) || previousTrace.lexicalCount !== (currentTrace?.resultCounts.lexical ?? 0) || previousTrace.candidateTopK !== (currentTrace?.candidateTopK ?? 0) || previousTrace.lexicalTopK !== (currentTrace?.lexicalTopK ?? 0) || !areStageCountsEqual(previousStageCounts, stageCounts) ? "changed" : "unchanged";
+  const traceChange = !previousTrace ? currentTrace ? "new" : "unchanged" : previousTrace.traceMode !== currentTrace?.mode || previousTrace.sourceBalanceStrategy !== currentTrace?.sourceBalanceStrategy || previousTrace.transformedQuery !== (currentTrace?.transformedQuery || undefined) || previousTrace.variantQueries.join("|") !== (currentTrace?.variantQueries ?? []).join("|") || previousTrace.finalCount !== (currentTrace?.resultCounts.final ?? 0) || previousTrace.vectorCount !== (currentTrace?.resultCounts.vector ?? 0) || previousTrace.lexicalCount !== (currentTrace?.resultCounts.lexical ?? 0) || previousTrace.candidateTopK !== (currentTrace?.candidateTopK ?? 0) || previousTrace.lexicalTopK !== (currentTrace?.lexicalTopK ?? 0) || !areStageCountsEqual(previousStageCounts, stageCounts) ? "changed" : "unchanged";
   return {
     candidateTopK: currentTrace?.candidateTopK ?? 0,
     caseId: caseResult.caseId,
@@ -6149,6 +6206,7 @@ var buildEvaluationCaseTraceSnapshot = ({
     previousFinalCount: previousTrace?.finalCount,
     previousLexicalCount: previousTrace?.lexicalCount,
     previousLexicalTopK: previousTrace?.lexicalTopK,
+    previousSourceBalanceStrategy: previousTrace?.sourceBalanceStrategy,
     previousStageCounts,
     previousTraceMode: previousTrace?.traceMode,
     previousTransformedQuery: previousTrace?.transformedQuery,
@@ -6157,6 +6215,7 @@ var buildEvaluationCaseTraceSnapshot = ({
     query: caseResult.query,
     stageCounts,
     status: caseResult.status,
+    sourceBalanceStrategy: currentTrace?.sourceBalanceStrategy,
     traceChange,
     traceMode: currentTrace?.mode,
     transformedQuery: currentTrace?.transformedQuery || undefined,
@@ -6201,6 +6260,7 @@ var buildEvaluationCaseTraceSnapshots = ({
         },
         runLexical: currentSnapshot.lexicalCount > 0,
         runVector: currentSnapshot.vectorCount > 0,
+        sourceBalanceStrategy: currentSnapshot.sourceBalanceStrategy ?? "cap",
         steps: [],
         topK: caseResult.topK,
         transformedQuery: currentSnapshot.transformedQuery ?? caseResult.query,
@@ -6231,6 +6291,7 @@ var formatRetrievalComparisonHeadline = (entry) => [
   `latency ${entry.response.summary.averageLatencyMs.toFixed(1)}ms`
 ].join(" \xB7 ");
 var formatTraceModes = (modes) => modes.length > 0 ? modes.join(" / ") : "n/a";
+var formatSourceBalanceStrategies = (strategies) => strategies.length > 0 ? strategies.join(" / ") : "n/a";
 var formatTraceStageSummary = (stageCounts) => {
   const topStages = Object.entries(stageCounts).sort((left, right) => right[1] - left[1]).slice(0, 3);
   return topStages.length > 0 ? topStages.map(([stage, count]) => `${stage} ${count}`).join(" \xB7 ") : "n/a";
@@ -6266,9 +6327,21 @@ var buildRAGComparisonTraceSummaryRows = (entry) => {
   }
   return [
     { label: "Modes", value: formatTraceModes(trace.modes) },
+    {
+      label: "Balance",
+      value: formatSourceBalanceStrategies(trace.sourceBalanceStrategies)
+    },
     { label: "Avg final", value: trace.averageFinalCount.toFixed(1) },
     { label: "Avg vector", value: trace.averageVectorCount.toFixed(1) },
     { label: "Avg lexical", value: trace.averageLexicalCount.toFixed(1) },
+    {
+      label: "Balanced",
+      value: formatTraceRatio(trace.balancedCases, trace.totalCases)
+    },
+    {
+      label: "Round robin",
+      value: formatTraceRatio(trace.roundRobinCases, trace.totalCases)
+    },
     {
       label: "Transforms",
       value: formatTraceRatio(trace.transformedCases, trace.totalCases)
@@ -6316,6 +6389,12 @@ var buildRAGComparisonTraceDiffRows = (entry, leader) => {
       value: `${formatTraceModes(trace.modes)} vs ${formatTraceModes(leaderTrace.modes)}`
     });
   }
+  if (formatSourceBalanceStrategies(trace.sourceBalanceStrategies) !== formatSourceBalanceStrategies(leaderTrace.sourceBalanceStrategies)) {
+    rows.push({
+      label: "Balance vs leader",
+      value: `${formatSourceBalanceStrategies(trace.sourceBalanceStrategies)} vs ${formatSourceBalanceStrategies(leaderTrace.sourceBalanceStrategies)}`
+    });
+  }
   rows.push({
     label: "Final delta",
     value: formatSignedDelta(trace.averageFinalCount - leaderTrace.averageFinalCount, 1)
@@ -6328,6 +6407,12 @@ var buildRAGComparisonTraceDiffRows = (entry, leader) => {
   }, {
     label: "Transform delta",
     value: formatTraceCountDelta(trace.transformedCases - leaderTrace.transformedCases)
+  }, {
+    label: "Balanced delta",
+    value: formatTraceCountDelta(trace.balancedCases - leaderTrace.balancedCases)
+  }, {
+    label: "Round robin delta",
+    value: formatTraceCountDelta(trace.roundRobinCases - leaderTrace.roundRobinCases)
   });
   if (stageDelta) {
     rows.push({ label: "Stage delta", value: stageDelta });
@@ -6491,7 +6576,7 @@ var buildRAGEvaluationHistoryRows = (history) => {
   if (history.latestRun.traceSummary) {
     rows.push({
       label: "Latest trace",
-      value: `${formatTraceModes(history.latestRun.traceSummary.modes)} \xB7 final ${history.latestRun.traceSummary.averageFinalCount.toFixed(1)} \xB7 vector ${history.latestRun.traceSummary.averageVectorCount.toFixed(1)} \xB7 lexical ${history.latestRun.traceSummary.averageLexicalCount.toFixed(1)}`
+      value: `${formatTraceModes(history.latestRun.traceSummary.modes)} \xB7 balance ${formatSourceBalanceStrategies(history.latestRun.traceSummary.sourceBalanceStrategies)} \xB7 final ${history.latestRun.traceSummary.averageFinalCount.toFixed(1)} \xB7 vector ${history.latestRun.traceSummary.averageVectorCount.toFixed(1)} \xB7 lexical ${history.latestRun.traceSummary.averageLexicalCount.toFixed(1)}`
     });
   }
   if (history.previousRun) {
@@ -6527,6 +6612,9 @@ var buildRAGEvaluationHistoryRows = (history) => {
     rows.push({
       label: "Trace mode shift",
       value: history.diff.traceSummaryDelta.modesChanged ? "changed" : "stable"
+    }, {
+      label: "Trace balance shift",
+      value: history.diff.traceSummaryDelta.sourceBalanceStrategiesChanged ? "changed" : "stable"
     }, {
       label: "Trace final delta",
       value: formatSignedDelta(history.diff.traceSummaryDelta.averageFinalCount, 1)
@@ -6536,6 +6624,12 @@ var buildRAGEvaluationHistoryRows = (history) => {
     }, {
       label: "Trace lexical delta",
       value: formatSignedDelta(history.diff.traceSummaryDelta.averageLexicalCount, 1)
+    }, {
+      label: "Trace balanced delta",
+      value: formatTraceCountDelta(history.diff.traceSummaryDelta.balancedCases)
+    }, {
+      label: "Trace round robin delta",
+      value: formatTraceCountDelta(history.diff.traceSummaryDelta.roundRobinCases)
     }, {
       label: "Trace transform delta",
       value: formatTraceCountDelta(history.diff.traceSummaryDelta.transformedCases)
@@ -6565,6 +6659,8 @@ var buildRAGEvaluationCaseTracePresentations = (history) => {
     const label = entry.label ?? entry.caseId;
     const currentMode = entry.traceMode ?? "no-trace";
     const previousMode = entry.previousTraceMode ?? "n/a";
+    const currentBalance = entry.sourceBalanceStrategy ?? "cap";
+    const previousBalance = entry.previousSourceBalanceStrategy ?? "n/a";
     const currentVariants = entry.variantQueries.length > 0 ? entry.variantQueries.join(", ") : "none";
     const previousVariants = entry.previousVariantQueries.length > 0 ? entry.previousVariantQueries.join(", ") : "none";
     const currentStages = Object.keys(entry.stageCounts).length > 0 ? Object.entries(entry.stageCounts).map(([stage, count]) => `${stage} ${count}`).join(", ") : "none";
@@ -6572,12 +6668,16 @@ var buildRAGEvaluationCaseTracePresentations = (history) => {
     return {
       caseId: entry.caseId,
       label,
-      summary: `${entry.traceChange} \xB7 ${previousMode}\u2192${currentMode} \xB7 final ${entry.previousFinalCount ?? 0}\u2192${entry.finalCount}`,
+      summary: `${entry.traceChange} \xB7 ${previousMode}\u2192${currentMode} \xB7 ${previousBalance}\u2192${currentBalance} \xB7 final ${entry.previousFinalCount ?? 0}\u2192${entry.finalCount}`,
       traceChange: entry.traceChange,
       rows: [
         { label: "Query", value: entry.query },
         { label: "Trace change", value: entry.traceChange },
         { label: "Mode", value: `${previousMode}\u2192${currentMode}` },
+        {
+          label: "Balance",
+          value: `${previousBalance}\u2192${currentBalance}`
+        },
         {
           label: "Transformed query",
           value: `${entry.previousTransformedQuery?.trim() || "n/a"}\u2192${entry.transformedQuery?.trim() || "n/a"}`
@@ -6649,12 +6749,15 @@ var buildRAGEvaluationRunDiff = ({
       averageLexicalCount: (current.traceSummary?.averageLexicalCount ?? 0) - (previous?.traceSummary?.averageLexicalCount ?? 0),
       averageLexicalTopK: (current.traceSummary?.averageLexicalTopK ?? 0) - (previous?.traceSummary?.averageLexicalTopK ?? 0),
       averageVectorCount: (current.traceSummary?.averageVectorCount ?? 0) - (previous?.traceSummary?.averageVectorCount ?? 0),
+      balancedCases: (current.traceSummary?.balancedCases ?? 0) - (previous?.traceSummary?.balancedCases ?? 0),
       lexicalCases: (current.traceSummary?.lexicalCases ?? 0) - (previous?.traceSummary?.lexicalCases ?? 0),
       modesChanged: (current.traceSummary?.modes ?? []).join("|") !== (previous?.traceSummary?.modes ?? []).join("|"),
+      roundRobinCases: (current.traceSummary?.roundRobinCases ?? 0) - (previous?.traceSummary?.roundRobinCases ?? 0),
       stageCounts: diffTraceStageCounts({
         current: current.traceSummary?.stageCounts ?? {},
         previous: previous?.traceSummary?.stageCounts ?? {}
       }),
+      sourceBalanceStrategiesChanged: (current.traceSummary?.sourceBalanceStrategies ?? []).join("|") !== (previous?.traceSummary?.sourceBalanceStrategies ?? []).join("|"),
       transformedCases: (current.traceSummary?.transformedCases ?? 0) - (previous?.traceSummary?.transformedCases ?? 0),
       variantCases: (current.traceSummary?.variantCases ?? 0) - (previous?.traceSummary?.variantCases ?? 0),
       vectorCases: (current.traceSummary?.vectorCases ?? 0) - (previous?.traceSummary?.vectorCases ?? 0)
@@ -12541,5 +12644,5 @@ export {
   aiChat
 };
 
-//# debugId=FD250EE2F99EB57964756E2164756E21
+//# debugId=952E4DA5A5D3C13F64756E2164756E21
 //# sourceMappingURL=index.js.map