Merge pull request #1165 from major/mhayden/logging-consistency-followup

RSPEED-2444: Replace remaining raw logging imports with get_logger()

Author: tisnik

src/app/database.py

@@ -1,6 +1,6 @@
 """Database engine management."""
 
-import logging
+from logging import DEBUG
 from pathlib import Path
 from typing import Any, Optional
 
@@ -175,7 +175,7 @@ def initialize_database() -> None:
     global engine, session_local  # pylint: disable=global-statement
 
     # Debug print all SQL statements if our logger is at-least DEBUG level
-    echo = bool(logger.isEnabledFor(logging.DEBUG))
+    echo = bool(logger.isEnabledFor(DEBUG))
 
     create_engine_kwargs = {
         "echo": echo,

src/runners/uvicorn.py

@@ -1,6 +1,7 @@
 """Uvicorn runner."""
 
-import logging
+from logging import INFO
+
 import uvicorn
 
 from log import get_logger
@@ -21,7 +22,7 @@ def start_uvicorn(configuration: ServiceConfiguration) -> None:
     """
     logger.info("Starting Uvicorn")
 
-    log_level = logging.INFO
+    log_level = INFO
 
     # please note:
     # TLS fields can be None, which means we will pass those values as None to uvicorn.run

src/telemetry/configuration_snapshot.py

@@ -10,7 +10,6 @@
 """
 
 import asyncio
-import logging
 from dataclasses import dataclass
 from enum import Enum
 from pathlib import PurePath
@@ -19,9 +18,10 @@
 import yaml
 from pydantic import SecretStr
 
+from log import get_logger
 from models.config import Configuration
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 # Masking output constants
 CONFIGURED: Literal["configured"] = "configured"

src/utils/vector_search.py

@@ -4,7 +4,6 @@
 and processing RAG chunks that is shared between query_v2.py and streaming_query_v2.py.
 """
 
-import logging
 import traceback
 from typing import Any, Optional
 from urllib.parse import urljoin
@@ -15,11 +14,12 @@
 
 import constants
 from configuration import AppConfig
+from log import get_logger
 from models.requests import QueryRequest
 from models.responses import ReferencedDocument
 from utils.types import RAGChunk
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 def _is_solr_enabled(configuration: AppConfig) -> bool: