From 9a8b302432a3f9bfa1e467dde027fc92dacce3e2 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 7 Nov 2023 10:56:28 +0800
Subject: refactor: remove Model.client_index, match client by name (#218)

---
 src/client/ernie.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/client/ernie.rs')

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
-- 
cgit v1.2.3