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

Bugfix/Cohere top K
menloparklab · Feb 1, 2024 · 48677c3 · 48677c3
2 parents 7e475f1 + 5c6b5b2
commit 48677c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/components/nodes/retrievers/CohereRerankRetriever/CohereRerank.ts b/packages/components/nodes/retrievers/CohereRerankRetriever/CohereRerank.ts
@@ -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
         }