From f3bdabc3cdfce9d430a32174ac8858b907af4137 Mon Sep 17 00:00:00 2001 From: craisin Date: Thu, 1 Jan 2026 18:06:30 -0800 Subject: [PATCH] add log level setting --- composer/agent.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/composer/agent.py b/composer/agent.py index f1116f4..0b04f9b 100644 --- a/composer/agent.py +++ b/composer/agent.py @@ -14,7 +14,16 @@ formatter = logging.Formatter("[%(levelname)s] - %(message)s") handler.setFormatter(formatter) logger = logging.getLogger("composer-agent-logger") logger.addHandler(handler) -logger.setLevel(logging.DEBUG) #TODO: add something to change log levles +log_levels = { + "DEBUG": logging.DEBUG, + "INFO": logging.INFO, + "WARN": logging.WARN, + "ERROR": logging.ERROR +} +if getenv("LOG_LEVEL") in log_levels: + logger.setLevel(log_levels[getenv("LOG_LEVEL")]) +else: + logger.setLevel(logging.INFO) environ['GIT_TERMINAL_PROMPT'] = '0'