🚧 added tracing log in reranking

curieo-org · Jul 2, 2024 · bb85335 · bb85335
1 parent 098c386
commit bb85335
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/server/src/rag/post_process.rs b/server/src/rag/post_process.rs
@@ -8,6 +8,7 @@ use regex::Regex;
 use std::cmp::Ordering;
 use tokio::sync::mpsc::Sender;
 
+#[tracing::instrument(level = "info", ret)]
 pub fn rerank_search_results(
     query_embeddings: &Embeddings,
     results_embeddings: &Vec<Embeddings>,