StrayWarrior 3 недель назад
Родитель
Сommit
b07a19226a
2 измененных файлов с 6 добавлено и 1 удалено
  1. 1 0
      agent_service.py
  2. 5 1
      configs/__init__.py

+ 1 - 0
agent_service.py

@@ -207,6 +207,7 @@ class AgentService:
 if __name__ == "__main__":
     config = configs.get()
     logging_service.setup_root_logger()
+    logging.warning("current env: {}".format(configs.get_env()))
     scheduler_logger = logging.getLogger('apscheduler')
     scheduler_logger.setLevel(logging.WARNING)
 

+ 5 - 1
configs/__init__.py

@@ -11,4 +11,8 @@ def get():
     env = os.environ.get('AI_AGENT_ENV', 'dev')
     if env not in ('dev', 'pre', 'prod'):
         raise ValueError(f"Invalid environment: {env}. Expected one of ('dev', 'pre', 'prod').")
-    return yaml.safe_load(open(f'{dirname}/{env}.yaml').read())
+    return yaml.safe_load(open(f'{dirname}/{env}.yaml').read())
+
+def get_env():
+    env = os.environ.get('AI_AGENT_ENV', 'dev')
+    return env