diff --git a/src/configuration.py b/src/configuration.py index 4a29a163..7c5b3403 100644 --- a/src/configuration.py +++ b/src/configuration.py @@ -1,8 +1,11 @@ import yaml +import logging from typing import Any, Optional from models.config import Configuration, LLamaStackConfiguration +logger = logging.getLogger(__name__) + class AppConfig: _instance = None @@ -21,7 +24,7 @@ def load_configuration(self, filename: str) -> None: """Load configuration from YAML file.""" with open(filename, encoding="utf-8") as fin: config_dict = yaml.safe_load(fin) - print(config_dict) + logger.info("Loaded configuration: %s", config_dict) self._configuration = Configuration(**config_dict) @property diff --git a/src/lightspeed-stack.py b/src/lightspeed-stack.py index 2f869748..22330527 100644 --- a/src/lightspeed-stack.py +++ b/src/lightspeed-stack.py @@ -1,13 +1,27 @@ """Lightspeed stack.""" +import logging from runners.uvicorn import start_uvicorn from models.config import Configuration from configuration import configuration +from rich.logging import RichHandler + +FORMAT = "%(message)s" +logging.basicConfig( + level="NOTSET", format=FORMAT, datefmt="[%X]", handlers=[RichHandler()] +) # set level=20 or logging.INFO to turn off debug +logger = logging.getLogger("rich") + +logging.basicConfig(level=logging.INFO) +logger = logging.getLogger(__name__) if __name__ == "__main__": - print("Lightspeed stack") + logger.info("Lightspeed stack startup") configuration.load_configuration("lightspeed-stack.yaml") - print(configuration) - print(configuration.llama_stack_configuration) + logger.info("Configuration: %s", configuration.configuration) + logger.info( + "Llama stack configuration: %s", configuration.llama_stack_configuration + ) start_uvicorn() + logger.info("Lightspeed stack finished") diff --git a/src/runners/uvicorn.py b/src/runners/uvicorn.py index fde90d84..d24fbbfc 100644 --- a/src/runners/uvicorn.py +++ b/src/runners/uvicorn.py @@ -14,10 +14,12 @@ def start_uvicorn() -> None: host = "localhost" port = 8080 workers = 1 + log_level = logging.INFO uvicorn.run( "app.main:app", host=host, port=port, workers=workers, + log_level=log_level, )