diff options
Diffstat (limited to 'src/client')
| -rw-r--r-- | src/client/common.rs | 4 | ||||
| -rw-r--r-- | src/client/model.rs | 8 |
2 files changed, 6 insertions, 6 deletions
diff --git a/src/client/common.rs b/src/client/common.rs index bf2f336..265ab0a 100644 --- a/src/client/common.rs +++ b/src/client/common.rs @@ -154,8 +154,8 @@ macro_rules! register_client { list_models(config).into_iter().filter(|v| v.model_type() == "embedding").collect() } - pub fn list_rerank_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { - list_models(config).into_iter().filter(|v| v.model_type() == "rerank").collect() + pub fn list_reranker_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { + list_models(config).into_iter().filter(|v| v.model_type() == "reranker").collect() } }; } 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}'"), } } |