Просмотр исходного кода

Update logging_service: configurable log dir

StrayWarrior 1 день назад
Родитель
Сommit
2555e986da
2 измененных файлов с 5 добавлено и 3 удалено
  1. 1 0
      pqai_agent/configs/prod.yaml
  2. 4 3
      pqai_agent/logging_service.py

+ 1 - 0
pqai_agent/configs/prod.yaml

@@ -58,6 +58,7 @@ system:
   human_intervention_alert_url: https://open.feishu.cn/open-apis/bot/v2/hook/c316b559-1c6a-4c4e-97c9-50b44e4c2a9d
   max_reply_workers: 5
   push_task_workers: 5
+  log_dir: /var/log/agent_service
 
 agent_behavior:
   message_aggregation_sec: 20

+ 4 - 3
pqai_agent/logging_service.py

@@ -36,15 +36,16 @@ def setup_root_logger(level=logging.DEBUG, logfile_name='service.log'):
     root_logger = logging.getLogger()
     root_logger.handlers.clear()
     root_logger.addHandler(console_handler)
-    if configs.get_env() == 'prod':
+
+    log_dir = configs.get().get('system', {}).get('log_dir', '')
+    if log_dir:
         file_handler = RotatingFileHandler(
-            f'/var/log/agent_service/{logfile_name}',
+            f'{log_dir}/{logfile_name}',
             maxBytes=64 * 1024 * 1024,
             backupCount=5,
             encoding='utf-8'
         )
         file_handler.setLevel(logging.DEBUG)
-        file_handler.setFormatter(formatter)
         root_logger.addHandler(file_handler)
 
     agent_logger = logging.getLogger('agent')