From f2378e172548f3357fca6e561803bfe3a013bcb3 Mon Sep 17 00:00:00 2001 From: sigoden Date: Fri, 21 Jun 2024 21:26:18 +0800 Subject: refactor: rename model.mode to model.type (#625) --- src/client/common.rs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'src/client/common.rs') diff --git a/src/client/common.rs b/src/client/common.rs index 6c86acd..de2fa50 100644 --- a/src/client/common.rs +++ b/src/client/common.rs @@ -145,15 +145,15 @@ macro_rules! register_client { } pub fn list_chat_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { - list_models(config).into_iter().filter(|v| v.mode() == "chat").collect() + list_models(config).into_iter().filter(|v| v.model_type() == "chat").collect() } pub fn list_embedding_models(config: &$crate::config::Config) -> Vec<&'static $crate::client::Model> { - list_models(config).into_iter().filter(|v| v.mode() == "embedding").collect() + 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.mode() == "rerank").collect() + list_models(config).into_iter().filter(|v| v.model_type() == "rerank").collect() } }; } -- cgit v1.2.3