From 3a00fb283e4f503901ce198ead65aebc46ae1875 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 29 Apr 2024 05:22:23 +0000
Subject: refactor: user config models replace client builtin models

---
 src/client/common.rs | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

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

diff --git a/src/client/common.rs b/src/client/common.rs
index aa243a7..842741c 100644
--- a/src/client/common.rs
+++ b/src/client/common.rs
@@ -89,15 +89,19 @@ macro_rules! register_client {
 
                 pub fn list_models(local_config: &$config) -> Vec<Model> {
                     let client_name = Self::name(local_config);
-                    for model in $crate::client::CLIENT_MODELS.iter() {
-                        match model {
-                            $crate::client::ClientModel::$config { models } => {
-                                return Model::from_config(client_name, &local_config.models, models);
+                    if local_config.models.is_empty() {
+                        for model in $crate::client::CLIENT_MODELS.iter() {
+                            match model {
+                                $crate::client::ClientModel::$config { models } => {
+                                    return Model::from_config(client_name, models);
+                                }
+                                _ => {}
                             }
-                            _ => {}
                         }
+                        vec![]
+                    } else {
+                        Model::from_config(client_name, &local_config.models)
                     }
-                    vec![]
                 }
 
                 pub fn name(config: &$config) -> &str {
-- 
cgit v1.2.3