diff options
Diffstat (limited to 'src/rag')
| -rw-r--r-- | src/rag/mod.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/rag/mod.rs b/src/rag/mod.rs index 62a8ff8..96fb9d5 100644 --- a/src/rag/mod.rs +++ b/src/rag/mod.rs @@ -483,7 +483,7 @@ impl Rag { } } let data = RerankData::new(query.to_string(), documents, top_k); - let list = client.rerank(data).await?; + let list = client.rerank(&data).await?; let ids: Vec<_> = list .into_iter() .take(top_k) @@ -588,7 +588,7 @@ impl Rag { query, }; let chunk_output = embedding_client - .embeddings(chunk_data) + .embeddings(&chunk_data) .await .context("Failed to create embedding")?; output.extend(chunk_output); |