summaryrefslogtreecommitdiffstats
path: root/src/client/common.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-21 21:26:18 +0800
committerGitHub <noreply@github.com>2024-06-21 21:26:18 +0800
commitf2378e172548f3357fca6e561803bfe3a013bcb3 (patch)
treee79a9253ec845d16fc4596f377c5515f45a104e9 /src/client/common.rs
parent054e998ff49a1f3abbc0a474d9a8d23c0ed4b17b (diff)
downloadaichat-f2378e172548f3357fca6e561803bfe3a013bcb3.tar.gz
refactor: rename model.mode to model.type (#625)
Diffstat (limited to 'src/client/common.rs')
-rw-r--r--src/client/common.rs6
1 files changed, 3 insertions, 3 deletions
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()
}
};
}