From 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 25 Jun 2024 07:39:35 +0800
Subject: feat: support rag-dedicated clients (jina and voyageai) (#645)

---
 src/rag/mod.rs | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/rag/mod.rs')

diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 30ca92f..3fdad02 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -334,6 +334,7 @@ impl Rag {
                 let list = client.rerank(data).await?;
                 let ids = list
                     .into_iter()
+                    .take(top_k)
                     .filter_map(|item| {
                         if item.relevance_score < min_score {
                             None
-- 
cgit v1.2.3