Update logging setup

agent_service.py

@@ -13,6 +13,7 @@ from apscheduler.schedulers.background import BackgroundScheduler
 
 import chat_service
 import global_flags
+import logging_service
 from chat_service import CozeChat, ChatServiceType
 from dialogue_manager import DialogueManager, DialogueState
 from user_manager import UserManager, LocalUserManager
@@ -187,16 +188,7 @@ class AgentService:
         return response
 
 if __name__ == "__main__":
-    logging.getLogger().setLevel(logging.DEBUG)
-    console_handler = logging.StreamHandler()
-    console_handler.setLevel(logging.DEBUG)
-    formatter = ColoredFormatter(
-        '%(asctime)s - %(funcName)s[%(lineno)d] - %(levelname)s - %(message)s'
-    )
-    console_handler.setFormatter(formatter)
-    root_logger = logging.getLogger()
-    root_logger.handlers.clear()
-    root_logger.addHandler(console_handler)
+    logging_service.setup_root_logger()
     scheduler_logger = logging.getLogger('apscheduler')
     scheduler_logger.setLevel(logging.WARNING)
 

logging_service.py

@@ -22,4 +22,14 @@ class ColoredFormatter(logging.Formatter):
             message = f"{COLORS[record.levelname]}{message}{COLORS['RESET']}"
         return message
 
-
+def setup_root_logger():
+    logging.getLogger().setLevel(logging.DEBUG)
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(logging.DEBUG)
+    formatter = ColoredFormatter(
+        '%(asctime)s - %(funcName)s[%(lineno)d] - %(levelname)s - %(message)s'
+    )
+    console_handler.setFormatter(formatter)
+    root_logger = logging.getLogger()
+    root_logger.handlers.clear()
+    root_logger.addHandler(console_handler)