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/localai.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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

diff --git a/src/client/localai.rs b/src/client/localai.rs
index ecb0e63..93853dc 100644
--- a/src/client/localai.rs
+++ b/src/client/localai.rs
@@ -41,14 +41,14 @@ impl LocalAIClient {
         ),
     ];
 
-    pub fn list_models(local_config: &LocalAIConfig, client_index: usize) -> Vec<Model> {
+    pub fn list_models(local_config: &LocalAIConfig) -> Vec<Model> {
         let client_name = Self::name(local_config);
 
         local_config
             .models
             .iter()
             .map(|v| {
-                Model::new(client_index, client_name, &v.name)
+                Model::new(client_name, &v.name)
                     .set_max_tokens(v.max_tokens)
                     .set_tokens_count_factors(OPENAI_TOKENS_COUNT_FACTORS)
             })
-- 
cgit v1.2.3