Set correct application log level based on config (#32)

* Set correct log level based on config * Moved log configuration to workflow_server
mlrun · Oct 14, 2024 · 8bb5e82 · 8bb5e82
1 parent 9a61420
commit 8bb5e82
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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)