[BugFix] Failed to print DEBUG level log (#307)

Signed-off-by: dengyunyang <584797741@qq.com>

vllm_omni/entrypoints/omni_stage.py

@@ -893,7 +893,7 @@ async def _stage_worker_async(
         _rx_bytes_by_rid[rid] = int(_rx_metrics.get("rx_transfer_bytes", 0))
 
         sampling_params = task["sampling_params"]
-        _logging.getLogger(__name__).debug("[Stage-%s] Received batch size=1, request_ids=%d", stage_id, rid)
+        _logging.getLogger(__name__).debug("[Stage-%s] Received batch size=1, request_ids=%s", stage_id, rid)
         print("--------------------------------", flush=True)
         print(f"[Stage-{stage_id}] Received batch size=1, request_ids={rid}", flush=True)
         print("--------------------------------", flush=True)

vllm_omni/logger.py

@@ -0,0 +1,22 @@
+import logging
+
+from vllm.logger import init_logger
+
+
+def _configure_vllm_omni_root_logger():
+    """
+    Configure the root logger for vllm_omni to propagate to vllm's root logger.
+    """
+    vllm_root = logging.getLogger("vllm")
+    vllm_omni_root = logging.getLogger("vllm_omni")
+    vllm_omni_root.handlers = []
+
+    vllm_omni_root.parent = vllm_root
+
+    vllm_omni_root.propagate = True
+
+    vllm_omni_root.setLevel(logging.NOTSET)
+
+
+_configure_vllm_omni_root_logger()
+init_logger(__name__)

vllm_omni/patch.py

@@ -9,6 +9,7 @@ from vllm.v1.engine import EngineCoreOutputs as _OriginalEngineCoreOutputs
 from vllm.v1.engine import EngineCoreRequest as _OriginalEngineCoreRequest
 from vllm.v1.request import Request as _OriginalRequest
 
+import vllm_omni.logger  # noqa: F401
 from vllm_omni.engine import OmniEngineCoreOutput, OmniEngineCoreOutputs, OmniEngineCoreRequest
 from vllm_omni.inputs.data import OmniTokensPrompt
 from vllm_omni.model_executor.layers.mrope import MRotaryEmbedding