From 5c6b5b233c28784161048c3ddc5c32368f2a330a Mon Sep 17 00:00:00 2001
From: Henry <hzj94@hotmail.com>
Date: Thu, 1 Feb 2024 13:14:02 +0000
Subject: [PATCH] fix top K

---
 .../nodes/retrievers/CohereRerankRetriever/CohereRerank.ts      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/components/nodes/retrievers/CohereRerankRetriever/CohereRerank.ts b/packages/components/nodes/retrievers/CohereRerankRetriever/CohereRerank.ts
index f74b8365..f12d1026 100644
--- 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
         }