fix(threshold): fixes threshold length calculation (askorama#356)

h4ad-forks · Apr 25, 2023 · beb5ef2 · beb5ef2
1 parent 305d15b
commit beb5ef2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/packages/orama/src/components/algorithms.ts b/packages/orama/src/components/algorithms.ts
@@ -44,7 +44,7 @@ export function prioritizeTokenScores(arrays: TokenScore[][], boost: number, thr
   // (fuzzy match with a minimum threshold)
   const thresholdLength = Math.ceil((threshold * 100 * results.length) / 100)
 
-  return results.slice(0, thresholdLength)
+  return results.slice(0, results.length + thresholdLength)
 }
 
 export function BM25(

diff --git a/packages/orama/tests/threshold.test.ts b/packages/orama/tests/threshold.test.ts
@@ -143,6 +143,6 @@ t.test('should return all the exact matches + X% of the partial matches', async
     threshold: 0.7,
   })
 
-  t.same(r1.count, 3)
+  t.same(r1.count, 4)
   t.same(r2.count, 3)
 })