fix package level logger name

Signed-off-by: Travis Johnson <[email protected]>

Author: tjohnson31415

src/vllm_tgis_adapter/__main__.py

@@ -19,14 +19,14 @@
 
 from .grpc import run_grpc_server
 from .http import run_http_server
-from .logging import init_logger
+from .logging import DEFAULT_LOGGER_NAME, init_logger
 from .tgis_utils.args import EnvVarArgumentParser, add_tgis_args, postprocess_tgis_args
 from .utils import check_for_failed_tasks, write_termination_log
 
 if TYPE_CHECKING:
     import argparse
 
-logger = init_logger("vllm-tgis-adapter")
+logger = init_logger(DEFAULT_LOGGER_NAME)
 
 
 async def start_servers(args: argparse.Namespace) -> None:

src/vllm_tgis_adapter/logging.py

@@ -5,16 +5,18 @@
     init_logger,  # noqa: F401
 )
 
+DEFAULT_LOGGER_NAME = __name__.split(".")[0]
+
 config = {**DEFAULT_LOGGING_CONFIG}
 
-config["formatters"]["vllm_tgis_adapter"] = DEFAULT_LOGGING_CONFIG["formatters"]["vllm"]
+config["formatters"][DEFAULT_LOGGER_NAME] = DEFAULT_LOGGING_CONFIG["formatters"]["vllm"]
 
 handler_config = DEFAULT_LOGGING_CONFIG["handlers"]["vllm"]
-handler_config["formatter"] = "vllm_tgis_adapter"
-config["handlers"]["vllm_tgis_adapter"] = handler_config
+handler_config["formatter"] = DEFAULT_LOGGER_NAME
+config["handlers"][DEFAULT_LOGGER_NAME] = handler_config
 
 logger_config = DEFAULT_LOGGING_CONFIG["loggers"]["vllm"]
-logger_config["handlers"] = ["vllm_tgis_adapter"]
-config["loggers"]["vllm_tgis_adapter"] = logger_config
+logger_config["handlers"] = [DEFAULT_LOGGER_NAME]
+config["loggers"][DEFAULT_LOGGER_NAME] = logger_config
 
 logging.config.dictConfig(config)

src/vllm_tgis_adapter/utils.py

@@ -26,9 +26,9 @@ def write_termination_log(msg: str, file: str = "/dev/termination-log") -> None:
         # Ignore any errors writing to the termination logfile.
         # Users can fall back to the stdout logs, and we don't want to pollute
         # those with an error here.
-        from .logging import init_logger
+        from .logging import DEFAULT_LOGGER_NAME, init_logger
 
-        logger = init_logger("vllm-tgis-adapter")
+        logger = init_logger(DEFAULT_LOGGER_NAME)
         logger.exception("Unable to write termination logs to %s", file)