diff --git a/logger_config.k8s.yaml b/logger_config.k8s.yaml index 6d5c7298..36fe3704 100644 --- a/logger_config.k8s.yaml +++ b/logger_config.k8s.yaml @@ -37,7 +37,7 @@ loggers: - stderr ccb: - level: WARNING + level: INFO handlers: - stderr propagate: false diff --git a/logger_config.yaml b/logger_config.yaml index 67e1949d..4efa2e11 100644 --- a/logger_config.yaml +++ b/logger_config.yaml @@ -29,7 +29,7 @@ formatters: handlers: stderr: class: logging.StreamHandler - level: WARNING + level: DEBUG formatter: default stream: ext://sys.stderr @@ -50,7 +50,7 @@ loggers: - file_json ccb: - level: DEBUG + level: INFO handlers: - stderr - file_json diff --git a/logger_config_em.yaml b/logger_config_em.yaml index dfb430ea..fc3849ad 100644 --- a/logger_config_em.yaml +++ b/logger_config_em.yaml @@ -29,7 +29,7 @@ formatters: handlers: stderr: class: logging.StreamHandler - level: WARNING + level: DEBUG formatter: default stream: ext://sys.stderr @@ -50,7 +50,7 @@ loggers: - file_json emserver: - level: DEBUG + level: INFO handlers: - stderr - file_json diff --git a/main.py b/main.py index 8fd5c6a5..f082787b 100755 --- a/main.py +++ b/main.py @@ -21,38 +21,15 @@ LOGGER_CONFIG_NAME = 'logger_config.yaml' LOGGER_K8S_CONFIG_NAME = 'logger_config.k8s.yaml' -def _setup_log_levels(debug: bool): - ''' - Set log levels for the modules at once for a cleaner usage later. - ''' - if not debug: - # warning is the default level - return - - LOGGERS = ( - 'ccb', - 'ccb.chain', - 'ccb.doc_loader', - 'ccb.injest', - 'ccb.models', - 'ccb.vectordb', - 'ccb.controller', - 'ccb.dyn_loader', - 'ccb.ocs_utils', - 'ccb.utils', - ) - - for name in LOGGERS: - logger = logging.getLogger(name) - logger.setLevel(logging.DEBUG) - if __name__ == '__main__': k8s_env = is_k8s_env() logging_config = get_logging_config(LOGGER_K8S_CONFIG_NAME if k8s_env else LOGGER_CONFIG_NAME) setup_logging(logging_config) app_config: TConfig = app.extra['CONFIG'] - _setup_log_levels(app_config.debug) + if app_config.debug: + logger = logging.getLogger('ccb') + logger.setLevel(logging.DEBUG) # do forks from a clean process that doesn't have any threads or locks mp.set_start_method('forkserver') diff --git a/main_em.py b/main_em.py index addcfd60..0162d9ea 100755 --- a/main_em.py +++ b/main_em.py @@ -144,7 +144,7 @@ def _wait_main_app_enabled() -> None: 'model_files', em_conf.llama['model'], ) - logger.debug(f'Trying model path: {em_conf.llama["model"]}') + logger.info(f'Trying model path: {em_conf.llama["model"]}') # if the model file is still not found, raise an error if not os.path.isfile(em_conf.llama['model']):