From e61aaae0e24eb0dbe25384f310d599f624a08126 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 19 Jan 2025 08:44:22 +0800
Subject: refactor: don't throw model/functions related error in info mode
 (#1101)

---
 src/config/mod.rs | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

(limited to 'src/config/mod.rs')

diff --git a/src/config/mod.rs b/src/config/mod.rs
index d6e710a..e2f9048 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -237,7 +237,7 @@ impl Default for Config {
 pub type GlobalConfig = Arc<RwLock<Config>>;
 
 impl Config {
-    pub fn init(working_mode: WorkingMode) -> Result<Self> {
+    pub fn init(working_mode: WorkingMode, info_flag: bool) -> Result<Self> {
         let config_path = Self::config_file();
         let mut config = if !config_path.exists() {
             match env::var(get_env_name("platform")) {
@@ -254,19 +254,26 @@ impl Config {
         };
 
         config.working_mode = working_mode;
+        config.info_flag = info_flag;
 
-        config.load_envs();
+        let setup = |config: &mut Self| -> Result<()> {
+            config.load_envs();
 
-        if let Some(wrap) = config.wrap.clone() {
-            config.set_wrap(&wrap)?;
-        }
-
-        config.load_functions()?;
+            if let Some(wrap) = config.wrap.clone() {
+                config.set_wrap(&wrap)?;
+            }
 
-        config.setup_model()?;
-        config.setup_document_loaders();
-        config.setup_user_agent();
+            config.load_functions()?;
 
+            config.setup_model()?;
+            config.setup_document_loaders();
+            config.setup_user_agent();
+            Ok(())
+        };
+        let ret = setup(&mut config);
+        if !info_flag {
+            ret?;
+        }
         Ok(config)
     }
 
-- 
cgit v1.2.3