summaryrefslogtreecommitdiffstats
path: root/src/client/common.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-11-03 06:52:57 +0800
committerGitHub <noreply@github.com>2023-11-03 06:52:57 +0800
commitf9c40e52dabda7b037805c0635b84ccb6d75f5a8 (patch)
tree73fda138c432ed18c5f95b7be571f8298f6284c8 /src/client/common.rs
parentdce6877f5de297803a737387fef7f672593904cf (diff)
downloadaichat-f9c40e52dabda7b037805c0635b84ccb6d75f5a8.tar.gz
refactor: improve code quanity (#203)
- update field name of ModelInfo - rename ModelInfo to Model
Diffstat (limited to 'src/client/common.rs')
-rw-r--r--src/client/common.rs18
1 files changed, 9 insertions, 9 deletions
diff --git a/src/client/common.rs b/src/client/common.rs
index 0dc637b..464e46a 100644
--- a/src/client/common.rs
+++ b/src/client/common.rs
@@ -46,16 +46,16 @@ macro_rules! register_client {
pub struct $client {
global_config: $crate::config::GlobalConfig,
config: $config,
- model_info: $crate::client::ModelInfo,
+ model: $crate::client::Model,
}
impl $client {
pub const NAME: &str = $name;
pub fn init(global_config: $crate::config::GlobalConfig) -> Option<Box<dyn Client>> {
- let model_info = global_config.read().model_info.clone();
+ let model = global_config.read().model.clone();
let config = {
- if let ClientConfig::$config_key(c) = &global_config.read().clients[model_info.index] {
+ if let ClientConfig::$config_key(c) = &global_config.read().clients[model.client_index] {
c.clone()
} else {
return None;
@@ -64,7 +64,7 @@ macro_rules! register_client {
Some(Box::new(Self {
global_config,
config,
- model_info,
+ model,
}))
}
@@ -79,11 +79,11 @@ macro_rules! register_client {
None
$(.or_else(|| $client::init(config.clone())))+
.ok_or_else(|| {
- let model_info = config.read().model_info.clone();
+ let model = config.read().model.clone();
anyhow::anyhow!(
- "Unknown client {} at config.clients[{}]",
- &model_info.client,
- &model_info.index
+ "Unknown client '{}' at config.clients[{}]",
+ &model.client_name,
+ &model.client_index
)
})
}
@@ -101,7 +101,7 @@ macro_rules! register_client {
anyhow::bail!("Unknown client {}", client)
}
- pub fn list_models(config: &$crate::config::Config) -> Vec<$crate::client::ModelInfo> {
+ pub fn list_models(config: &$crate::config::Config) -> Vec<$crate::client::Model> {
config
.clients
.iter()