refactor: embeddings/rerank fn accept ref data (#878)

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);