From 8bb5e82bc16af1a2894974588bb1a6361007f646 Mon Sep 17 00:00:00 2001 From: Nick Schenone Date: Mon, 14 Oct 2024 09:07:39 -0700 Subject: [PATCH] Set correct application log level based on config (#32) * Set correct log level based on config * Moved log configuration to workflow_server --- genai_factory/src/genai_factory/__main__.py | 2 +- genai_factory/src/genai_factory/workflows/workflow_server.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/genai_factory/src/genai_factory/__main__.py b/genai_factory/src/genai_factory/__main__.py index 608bdc9..9a4900d 100644 --- a/genai_factory/src/genai_factory/__main__.py +++ b/genai_factory/src/genai_factory/__main__.py @@ -20,9 +20,9 @@ import click import dotenv +from genai_factory.api import router from genai_factory import WorkflowServerConfig -from genai_factory.api import router # Load the environment variables: dotenv.load_dotenv(os.environ.get("GENAI_FACTORY_ENV_PATH", "./.env")) diff --git a/genai_factory/src/genai_factory/workflows/workflow_server.py b/genai_factory/src/genai_factory/workflows/workflow_server.py index 23c41f8..4cfe865 100644 --- a/genai_factory/src/genai_factory/workflows/workflow_server.py +++ b/genai_factory/src/genai_factory/workflows/workflow_server.py @@ -49,6 +49,7 @@ def _set_controller_client(self): def set_config(self, config: WorkflowServerConfig): self._config = config + logger.setLevel(config.log_level.upper()) # reinitialize the controller client with the new config self._set_controller_client() self._session_store = SessionStore(self._controller_client)