summaryrefslogtreecommitdiffstats
path: root/src/client/model.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/client/model.rs
parent2fbb5271af27eee7ba3ae6d2b3c1aa0818648c8f (diff)
downloadaichat-2bc9607b00fa5f31ced08f78fe08fe71b923b152.tar.gz
refactor: rename model type `rerank` to `reranker` (#646)
Diffstat (limited to 'src/client/model.rs')
-rw-r--r--src/client/model.rs8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/client/model.rs b/src/client/model.rs
index d661d85..e6142d4 100644
--- a/src/client/model.rs
+++ b/src/client/model.rs
@@ -1,5 +1,5 @@
use super::{
- list_chat_models, list_embedding_models, list_rerank_models,
+ list_chat_models, list_embedding_models, list_reranker_models,
message::{Message, MessageContent},
EmbeddingsData,
};
@@ -57,10 +57,10 @@ impl Model {
}
}
- pub fn retrieve_rerank(config: &Config, model_id: &str) -> Result<Self> {
- match Self::find(&list_rerank_models(config), model_id) {
+ pub fn retrieve_reranker(config: &Config, model_id: &str) -> Result<Self> {
+ match Self::find(&list_reranker_models(config), model_id) {
Some(v) => Ok(v),
- None => bail!("Invalid rerank model '{model_id}'"),
+ None => bail!("Invalid reranker model '{model_id}'"),
}
}