Merge pull request #26 from tisnik/proper-logging

Proper logging in the service

Author: tisnik

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

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")

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,
     )