From 2bc9607b00fa5f31ced08f78fe08fe71b923b152 Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 25 Jun 2024 07:50:19 +0800 Subject: refactor: rename model type `rerank` to `reranker` (#646) --- src/config/input.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/config/input.rs') diff --git a/src/config/input.rs b/src/config/input.rs index 3a258b9..09dfa8f 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -177,11 +177,11 @@ impl Input { config.rag_min_score_keyword_search, ) }; - let rerank = match self.config.read().rag_rerank_model.clone() { - Some(rerank_model_id) => { + let rerank = match self.config.read().rag_reranker_model.clone() { + Some(reranker_model_id) => { let min_score = self.config.read().rag_min_score_rerank; let rerank_model = - Model::retrieve_rerank(&self.config.read(), &rerank_model_id)?; + Model::retrieve_reranker(&self.config.read(), &reranker_model_id)?; let rerank_client = init_client(&self.config, Some(rerank_model))?; Some((rerank_client, min_score)) } -- cgit v1.2.3