diff --git a/private_gpt/components/llm/custom/sagemaker.py b/private_gpt/components/llm/custom/sagemaker.py index 4acb30902..070f8ed67 100644 --- a/private_gpt/components/llm/custom/sagemaker.py +++ b/private_gpt/components/llm/custom/sagemaker.py @@ -27,7 +27,7 @@ CompletionResponseGen, ) -log = logging.getLogger(__name__) +logger = logging.getLogger(__name__) class LineIterator: @@ -84,7 +84,7 @@ def __next__(self) -> Any: continue raise if "PayloadPart" not in chunk: - log.warning("Unknown event type=%s", chunk) + logger.warning("Unknown event type=%s", chunk) continue self.buffer.seek(0, io.SEEK_END) self.buffer.write(chunk["PayloadPart"]["Bytes"]) diff --git a/private_gpt/settings/settings_loader.py b/private_gpt/settings/settings_loader.py index 4de5ac7df..99c2ca407 100644 --- a/private_gpt/settings/settings_loader.py +++ b/private_gpt/settings/settings_loader.py @@ -10,7 +10,7 @@ from private_gpt.constants import PROJECT_ROOT_PATH from private_gpt.settings.yaml import load_yaml_with_envvars -log = logging.getLogger(__name__) +logger = logging.getLogger(__name__) _settings_folder = os.environ.get("PGPT_SETTINGS_FOLDER", PROJECT_ROOT_PATH) @@ -44,7 +44,7 @@ def load_profile(profile: str) -> dict[str, Any]: def load_active_profiles() -> dict[str, Any]: """Load active profiles and merge them.""" - log.info("Starting application with profiles=%s", active_profiles) + logger.info("Starting application with profiles=%s", active_profiles) loaded_profiles = [load_profile(profile) for profile in active_profiles] merged: dict[str, Any] = functools.reduce(deep_update, loaded_profiles, {}) return merged