@absolutejs/absolute 0.19.0-beta.509 → 0.19.0-beta.510

package/dist/ai/index.js

@@ -2717,15 +2717,8 @@ var detectDomains = (tokens) => {
 var uniqueQueryStrings = (values) => Array.from(new Set(values.map((value) => value.trim()).filter((value) => value.length > 0)));
 var hasExplicitTimestamp = (query) => /\b\d{1,2}:\d{2}\b/.test(query) || /\b\d{1,2}:\d{2}\s*(?:to|-)\s*\d{1,2}:\d{2}\b/i.test(query);
 var hasNamedSourcePhrase = (query) => /\bnamed\s+[a-z0-9]/i.test(query) || /\btitled\s+[a-z0-9]/i.test(query);
-var shouldPreservePrimaryQuery = (query, domains) => {
-  if (hasExplicitTimestamp(query) && (domains.includes("audio") || domains.includes("video"))) {
-    return true;
-  }
-  if (hasNamedSourcePhrase(query) && domains.includes("spreadsheet")) {
-    return true;
-  }
-  return false;
-};
+var isExactMediaTimestampQuery = (query, domains) => hasExplicitTimestamp(query) && (domains.includes("audio") || domains.includes("video"));
+var isExactNamedSpreadsheetQuery = (query, domains) => hasNamedSourcePhrase(query) && domains.includes("spreadsheet");
 var createHeuristicRAGQueryTransform = (options = {}) => createRAGQueryTransform({
   defaultModel: options.defaultModel ?? "absolute-heuristic-query-transform",
   providerName: options.providerName ?? "absolute_heuristic",
@@ -2769,10 +2762,11 @@ var createHeuristicRAGQueryTransform = (options = {}) => createRAGQueryTransform
     if (mediaTimestampVariant.length > 0) {
       variants.push(mediaTimestampVariant);
     }
-    const preservePrimaryQuery = shouldPreservePrimaryQuery(query, domains);
+    const exactMediaTimestampQuery = isExactMediaTimestampQuery(query, domains);
+    const preservePrimaryQuery = exactMediaTimestampQuery || isExactNamedSpreadsheetQuery(query, domains);
     return {
       query: preservePrimaryQuery ? query : rewrittenQuery,
-      variants: uniqueQueryStrings(preservePrimaryQuery ? [rewrittenQuery, ...variants] : variants)
+      variants: uniqueQueryStrings(exactMediaTimestampQuery ? [] : preservePrimaryQuery ? [rewrittenQuery, ...variants] : variants)
     };
   }
 });
@@ -4298,6 +4292,7 @@ var prepareRAGDirectoryDocuments = async (input) => prepareRAGDocuments(await lo
 
 // src/ai/rag/collection.ts
 var DEFAULT_TOP_K = 6;
+var VARIANT_RESULT_WEIGHT = 0.92;
 var mergeQueryResults = (results) => {
   const merged = new Map;
   for (const result of results) {
@@ -4313,6 +4308,16 @@ var mergeQueryResults = (results) => {
     return left.chunkId.localeCompare(right.chunkId);
   });
 };
+var weightQueryResults = (results, queryIndex) => {
+  if (queryIndex === 0) {
+    return results;
+  }
+  const weight = Math.pow(VARIANT_RESULT_WEIGHT, queryIndex);
+  return results.map((result) => ({
+    ...result,
+    score: result.score * weight
+  }));
+};
 var shouldRunVectorRetrieval = (mode) => mode === "vector" || mode === "hybrid";
 var shouldRunLexicalRetrieval = (mode, store) => mode === "lexical" || mode === "hybrid" && Boolean(store.queryLexical);
 var createRAGCollection = (options) => {
@@ -4355,7 +4360,7 @@ var createRAGCollection = (options) => {
       signal: input.signal,
       text: input.query
     }, "query") : [];
-    const resultGroups = await Promise.all(searchQueries.map(async (query) => {
+    const resultGroups = await Promise.all(searchQueries.map(async (query, queryIndex) => {
       const [vectorResults2, lexicalResults2] = await Promise.all([
         runVector ? embed({
           model,
@@ -4372,7 +4377,10 @@ var createRAGCollection = (options) => {
           topK: lexicalTopK
         }) ?? Promise.resolve([]) : Promise.resolve([])
       ]);
-      return { lexicalResults: lexicalResults2, vectorResults: vectorResults2 };
+      return {
+        lexicalResults: weightQueryResults(lexicalResults2, queryIndex),
+        vectorResults: weightQueryResults(vectorResults2, queryIndex)
+      };
     }));
     const vectorResults = mergeQueryResults(resultGroups.flatMap((group) => group.vectorResults));
     const lexicalResults = mergeQueryResults(resultGroups.flatMap((group) => group.lexicalResults));
@@ -10277,5 +10285,5 @@ export {
   aiChat
 };
 
-//# debugId=E4C202F91B1AEC6364756E2164756E21
+//# debugId=9FFECCD86029913764756E2164756E21
 //# sourceMappingURL=index.js.map