diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-25 07:39:35 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-25 07:39:35 +0800 |
| commit | 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (patch) | |
| tree | 5e72e72d592025c00a46cebd0eb7012ccd765702 /src/rag/mod.rs | |
| parent | ed71901611247d41daed8112a5106b42eb12395b (diff) | |
| download | aichat-2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f.tar.gz | |
feat: support rag-dedicated clients (jina and voyageai) (#645)
Diffstat (limited to 'src/rag/mod.rs')
| -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 |