fix top K

2026-06-28 21:00:58 +03:00 · 2024-02-01 13:14:02 +00:00
parent c98ef7a8b1
commit 5c6b5b233c
1 changed files with 1 additions and 1 deletions
@@ -47,7 +47,7 @@ export class CohereRerank extends BaseDocumentCompressor {
                doc.metadata.relevance_score = result.relevance_score
                finalResults.push(doc)
            })
-            return finalResults
+            return finalResults.splice(0, this.k)
        } catch (error) {
            return documents
        }