diff options
Diffstat (limited to 'src/client/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs index c871450..5e32a81 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -64,11 +64,11 @@ impl ErnieClient { ("secret_key", "Secret Key:", true, PromptKind::String), ]; - pub fn list_models(local_config: &ErnieConfig, client_index: usize) -> Vec<Model> { + pub fn list_models(local_config: &ErnieConfig) -> Vec<Model> { let client_name = Self::name(local_config); MODELS .into_iter() - .map(|(name, _)| Model::new(client_index, client_name, name)) + .map(|(name, _)| Model::new(client_name, name)) .collect() } @@ -79,7 +79,7 @@ impl ErnieClient { let (_, chat_endpoint) = MODELS .iter() .find(|(v, _)| v == &model) - .ok_or_else(|| anyhow!("Miss Model '{}' in {}", model, self.model.client_name))?; + .ok_or_else(|| anyhow!("Miss Model '{}'", self.model.id()))?; let url = format!("{API_BASE}{chat_endpoint}?access_token={}", unsafe { &ACCESS_TOKEN |
