feat: add production metric for retracted requests due to insufficient kvcache (#7030)
Signed-off-by: Zhao Chen <zhaochen.zju@gmail.com>
This commit is contained in:
@@ -145,6 +145,7 @@ class SchedulerStats:
|
||||
num_prefill_infight_queue_reqs: int = 0
|
||||
num_decode_prealloc_queue_reqs: int = 0
|
||||
num_decode_transfer_queue_reqs: int = 0
|
||||
total_retracted_reqs: int = 0
|
||||
|
||||
|
||||
class SchedulerMetricsCollector:
|
||||
@@ -219,6 +220,13 @@ class SchedulerMetricsCollector:
|
||||
multiprocess_mode="mostrecent",
|
||||
)
|
||||
|
||||
self.total_retracted_reqs = Gauge(
|
||||
name="sglang:total_retracted_reqs",
|
||||
documentation="The total number of retracted requests due to kvcache full.",
|
||||
labelnames=labels.keys(),
|
||||
multiprocess_mode="mostrecent",
|
||||
)
|
||||
|
||||
# Disaggregation queue metrics
|
||||
self.num_prefill_prealloc_queue_reqs = Gauge(
|
||||
name="sglang:num_prefill_prealloc_queue_reqs",
|
||||
@@ -279,6 +287,7 @@ class SchedulerMetricsCollector:
|
||||
self._log_gauge(self.num_grammar_queue_reqs, stats.num_grammar_queue_reqs)
|
||||
self._log_gauge(self.cache_hit_rate, stats.cache_hit_rate)
|
||||
self._log_gauge(self.spec_accept_length, stats.spec_accept_length)
|
||||
self._log_gauge(self.total_retracted_reqs, stats.total_retracted_reqs)
|
||||
|
||||
# Disaggregation metrics
|
||||
self._log_gauge(
|
||||
|
||||
Reference in New Issue
Block a user