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/client/model.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/client/model.rs') 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 { - match Self::find(&list_rerank_models(config), model_id) { + pub fn retrieve_reranker(config: &Config, model_id: &str) -> Result { + 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}'"), } } -- cgit v1.2.3