🎨 cleanup

Signed-off-by: Joe Runde <[email protected]>

Author: joerunde

src/vllm_tgis_adapter/tgis_utils/logs.py

@@ -17,6 +17,7 @@
 
     from vllm import PromptType, RequestOutput, SamplingParams
     from vllm.engine.protocol import EngineClient
+    from vllm.sequence import RequestMetrics
 
 logger = init_logger(__name__)
 
@@ -97,7 +98,7 @@ async def generate_with_logging(*args, **kwargs) -> AsyncGenerator[RequestOutput
         if last:
             # Log the response
             with suppress(BaseException):
-                _new_log_response(
+                _log_response(
                     request_id=request_id,
                     correlation_id=correlation_id,
                     response=last,
@@ -108,10 +109,6 @@ async def generate_with_logging(*args, **kwargs) -> AsyncGenerator[RequestOutput
     engine.generate = generate_with_logging
 
 
-if TYPE_CHECKING:
-    from vllm.sequence import RequestMetrics
-
-
 def _log_error(request_id: str, correlation_id: str, exception_str: str) -> None:
     logger.error(
         "Request failed: request_id=%s correlation_id=%s error=%s",
@@ -143,7 +140,7 @@ def _log_request(
     )
 
 
-def _new_log_response(
+def _log_response(
     request_id: str,
     correlation_id: str,
     response: RequestOutput,