Cleanup environment variable in tabby.server

tabby/server/__init__.py

@@ -35,6 +35,10 @@ if MODEL_BACKEND == "triton":
 else:
     model_backend = PythonModelService(MODEL_NAME)
 
+LOGS_DIR = os.environ.get("LOGS_DIR", None)
+if LOGS_DIR is not None:
+    events.setup_logging(os.path.join(LOGS_DIR, "tabby-server"))
+
 
 @app.post("/v1/completions")
 async def completions(request: CompletionRequest) -> CompletionResponse:

tabby/server/events.py

@@ -6,8 +6,6 @@ from pydantic import BaseModel
 
 from . import models
 
-logger.configure(handlers=[])
-
 
 def setup_logging(logdir):
     try:
@@ -15,6 +13,8 @@ def setup_logging(logdir):
     except FileNotFoundError:
         pass
 
+    # Remove default handler
+    logger.remove()
     logger.add(
         os.path.join(logdir, "events.{time}.log"),
         rotation="1 hours",
@@ -27,11 +27,6 @@ def setup_logging(logdir):
     )
 
 
-LOGS_DIR = os.environ.get("LOGS_DIR", None)
-if LOGS_DIR is not None:
-    setup_logging(os.path.join(LOGS_DIR, "tabby-server"))
-
-
 def log_completions(
     request: models.CompletionRequest, response: models.CompletionResponse
 ) -> None: