From 819f50d9b16cb2afef5e3865de64bd0698910b27 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 8 Nov 2023 08:24:27 +0800
Subject: fix: unexpected additional newline in REPL (#231)

---
 src/config/mod.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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

diff --git a/src/config/mod.rs b/src/config/mod.rs
index 5219727..9c86e4c 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -866,8 +866,12 @@ fn set_bool(target: &mut bool, value: &str) {
 fn setup_logger() -> Result<()> {
     use simplelog::{LevelFilter, WriteLogger};
     let file = std::fs::File::create(Config::local_path("debug.log")?)?;
+    let log_filter = match std::env::var("AICHAT_LOG_FILTER") {
+        Ok(v) => v,
+        Err(_) => "aichat".into(),
+    };
     let config = simplelog::ConfigBuilder::new()
-        .add_filter_allow_str("aichat")
+        .add_filter_allow(log_filter)
         .set_thread_level(LevelFilter::Off)
         .set_time_level(LevelFilter::Off)
         .build();
-- 
cgit v1.2.3