diff --git a/src/vllm_tgis_adapter/grpc/grpc_server.py b/src/vllm_tgis_adapter/grpc/grpc_server.py index 15960677..17787889 100644 --- a/src/vllm_tgis_adapter/grpc/grpc_server.py +++ b/src/vllm_tgis_adapter/grpc/grpc_server.py @@ -195,8 +195,8 @@ async def post_init(self) -> None: assert self.tokenizer is not None # Swap in the special TGIS stats logger - assert hasattr(self.engine.engine, "stat_logger") - assert self.engine.engine.stat_logger + assert hasattr(self.engine.engine, "stat_loggers") + assert self.engine.engine.stat_loggers vllm_stat_logger = self.engine.engine.stat_loggers["prometheus"] tgis_stats_logger = TGISStatLogger(