summaryrefslogtreecommitdiffstats
path: root/src/config/input.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-25 07:50:19 +0800
committerGitHub <noreply@github.com>2024-06-25 07:50:19 +0800
commit2bc9607b00fa5f31ced08f78fe08fe71b923b152 (patch)
treea641079ed22de950e4c8f6e49ee6fcb713d5421e /src/config/input.rs
parent2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (diff)
downloadaichat-2bc9607b00fa5f31ced08f78fe08fe71b923b152.tar.gz
refactor: rename model type `rerank` to `reranker` (#646)
Diffstat (limited to 'src/config/input.rs')
-rw-r--r--src/config/input.rs6
1 files changed, 3 insertions, 3 deletions
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))
}