Merge pull request #1656 from FlowiseAI/bugfix/Rerank-Top-K

Bugfix/Cohere top K
2026-06-28 15:00:57 +03:00 · 2024-02-01 16:47:17 +00:00
parent 7e475f1288 5c6b5b233c
commit 48677c33cc
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
        }