logging services

src/text_generation/entrypoints/http_api_controller.py

@@ -3,11 +3,11 @@ import traceback
 
 from src.text_generation.adapters.llm.llm import Phi3LanguageModel
 from src.text_generation.adapters.llm.llm_rag import Phi3LanguageModelWithRag
-from src.text_generation.service.logging_service import LoggingService
+from src.text_generation.services.logging.file_logging_service import FileLoggingService
 
 class HttpApiController:
     def __init__(self):
-        self.logger = LoggingService(filename='text_generation.controller.log').logger
+        self.logger = FileLoggingService(filename='text_generation.controller.log').logger
         self.routes = {}
         # Register routes
         self.register_routes()

src/text_generation/entrypoints/server.py

@@ -1,11 +1,11 @@
 from src.text_generation.entrypoints.http_api_controller import HttpApiController
-from src.text_generation.service.logging_service import LoggingService
+from src.text_generation.services.logging.file_logging_service import FileLoggingService
 from wsgiref.simple_server import make_server
 
 
 class RestApiServer:
     def __init__(self):
-        self.logger = LoggingService(filename='text_generation.server.log').logger
+        self.logger = FileLoggingService(filename='text_generation.server.log')
 
     def listen(self):
         try:

src/text_generation/service/logging_service.py

@@ -1,10 +0,0 @@
-import logging
-
-class LoggingService:
-    # TODO use base class and make this file logging service
-
-    def __init__(self, filename):
-        logger = logging.getLogger()
-        logger.setLevel(logging.DEBUG)
-        logger.addHandler(logging.FileHandler(filename))
-        self.logger = logger

src/text_generation/services/logging/abstract_logging_service.py

@@ -0,0 +1,10 @@
+import abc
+import logging
+
+
+class AbstractLoggingService(abc.ABC):
+    def __init__(self, handler: logging.Handler):
+        logger = logging.getLogger()
+        logger.setLevel(logging.DEBUG)
+        logger.addHandler(handler)
+        self.logger = logger

src/text_generation/services/logging/file_logging_service.py

@@ -0,0 +1,8 @@
+import logging
+from src.text_generation.services.logging.abstract_logging_service import AbstractLoggingService
+
+
+class FileLoggingService(AbstractLoggingService):
+    def __init__(self, filename):
+        super().__init__(handler=logging.FileHandler(filename))
+        self.logger = super().logger