diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-25 07:50:19 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-25 07:50:19 +0800 |
| commit | 2bc9607b00fa5f31ced08f78fe08fe71b923b152 (patch) | |
| tree | a641079ed22de950e4c8f6e49ee6fcb713d5421e /src/config/input.rs | |
| parent | 2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (diff) | |
| download | aichat-2bc9607b00fa5f31ced08f78fe08fe71b923b152.tar.gz | |
refactor: rename model type `rerank` to `reranker` (#646)
Diffstat (limited to 'src/config/input.rs')
| -rw-r--r-- | src/config/input.rs | 6 |
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)) } |
