add cache_config's info to prometheus metrics. (#3100)

This commit is contained in:
Allen.Dou
2024-02-29 14:15:18 +08:00
committed by GitHub
parent a6d471c759
commit 9289e577ec
3 changed files with 14 additions and 1 deletions

View File

@@ -1,5 +1,5 @@
from vllm.logger import init_logger
from prometheus_client import Counter, Gauge, Histogram, REGISTRY, disable_created_metrics
from prometheus_client import Counter, Gauge, Histogram, Info, REGISTRY, disable_created_metrics
import time
import numpy as np
@@ -23,6 +23,10 @@ class Metrics:
if hasattr(collector, "_name") and "vllm" in collector._name:
REGISTRY.unregister(collector)
self.info_cache_config = Info(
name='vllm:cache_config',
documentation='information of cache_config')
# System stats
self.gauge_scheduler_running = Gauge(
name="vllm:num_requests_running",
@@ -128,6 +132,10 @@ class StatLogger:
self.labels = labels
self.metrics = Metrics(labelnames=list(labels.keys()))
def info(self, type: str, obj: object) -> None:
if type == "cache_config":
self.metrics.info_cache_config.info(obj.metrics_info())
def _get_throughput(self, tracked_stats: List[int], now: float) -> float:
return float(np.sum(tracked_stats) / (now - self.last_local_log))