diff options
Diffstat (limited to 'src/rag')
| -rw-r--r-- | src/rag/mod.rs | 1 |
1 files changed, 1 insertions, 0 deletions
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 |