From 054e998ff49a1f3abbc0a474d9a8d23c0ed4b17b Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 21 Jun 2024 19:45:55 +0800
Subject: feat: serve embeddings api (#624)

---
 src/logger.rs | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

(limited to 'src/logger.rs')

diff --git a/src/logger.rs b/src/logger.rs
index f7ef2f5..25d6382 100644
--- a/src/logger.rs
+++ b/src/logger.rs
@@ -6,8 +6,9 @@ use simplelog::{format_description, Config as LogConfig, ConfigBuilder};
 
 #[cfg(debug_assertions)]
 pub fn setup_logger(working_mode: WorkingMode) -> Result<()> {
-    let config = build_config();
-    if working_mode == WorkingMode::Serve {
+    let is_serve = working_mode.is_serve();
+    let config = build_config(is_serve);
+    if is_serve {
         simplelog::SimpleLogger::init(LevelFilter::Debug, config)?;
     } else {
         let file = std::fs::File::create(crate::config::Config::local_path("debug.log")?)?;
@@ -18,17 +19,21 @@ pub fn setup_logger(working_mode: WorkingMode) -> Result<()> {
 
 #[cfg(not(debug_assertions))]
 pub fn setup_logger(working_mode: WorkingMode) -> Result<()> {
-    let config = build_config();
-    if working_mode == WorkingMode::Serve {
+    if working_mode.is_serve() {
+        let config = build_config(true);
         simplelog::SimpleLogger::init(log::LevelFilter::Info, config)?;
     }
     Ok(())
 }
 
-fn build_config() -> LogConfig {
-    let log_filter = match std::env::var("AICHAT_LOG_FILTER") {
-        Ok(v) => v,
-        Err(_) => "aichat".into(),
+fn build_config(is_serve: bool) -> LogConfig {
+    let log_filter = if is_serve {
+        "aichat::serve".into()
+    } else {
+        match std::env::var("AICHAT_LOG_FILTER") {
+            Ok(v) => v,
+            Err(_) => "aichat".into(),
+        }
     };
     ConfigBuilder::new()
         .add_filter_allow(log_filter)
-- 
cgit v1.2.3