diff options
| author | sigoden <sigoden@gmail.com> | 2023-11-03 06:52:57 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-03 06:52:57 +0800 |
| commit | f9c40e52dabda7b037805c0635b84ccb6d75f5a8 (patch) | |
| tree | 73fda138c432ed18c5f95b7be571f8298f6284c8 /src/client/common.rs | |
| parent | dce6877f5de297803a737387fef7f672593904cf (diff) | |
| download | aichat-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.rs | 18 |
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() |
