File tree Expand file tree Collapse file tree 2 files changed +9
-4
lines changed
lib/bindings/python/rust/llm/block_manager/vllm/connector Expand file tree Collapse file tree 2 files changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -14,8 +14,8 @@ use crate::llm::block_manager::{
1414 distributed:: KvbmLeader as PyKvbmLeader , vllm:: connector:: leader:: slot:: VllmConnectorSlot ,
1515 vllm:: KvbmRequest , VllmBlockManager ,
1616} ;
17- use dynamo_runtime:: metrics:: prometheus_names:: kvbm_connector_namespace;
1817use crate :: DistributedRuntime as PyDistributedRuntime ;
18+ use dynamo_runtime:: metrics:: prometheus_names:: kvbm_connector_namespace;
1919
2020use dynamo_llm:: block_manager:: {
2121 block:: {
@@ -103,7 +103,9 @@ impl KvConnectorLeader {
103103 // if we need a drt, get it from here
104104 let drt = drt. inner ( ) . clone ( ) ;
105105
106- let ns = drt. namespace ( kvbm_connector_namespace:: KVBM_CONNECTOR_LEADER ) . unwrap ( ) ;
106+ let ns = drt
107+ . namespace ( kvbm_connector_namespace:: KVBM_CONNECTOR_LEADER )
108+ . unwrap ( ) ;
107109
108110 let kvbm_metrics = KvbmMetrics :: new ( & ns) ;
109111
Original file line number Diff line number Diff line change @@ -12,11 +12,11 @@ use std::sync::{Arc, OnceLock};
1212
1313use super :: * ;
1414use crate :: llm:: block_manager:: distributed:: get_barrier_id;
15- use dynamo_runtime:: metrics:: prometheus_names:: kvbm_connector_namespace;
1615use crate :: {
1716 llm:: block_manager:: distributed:: VllmTensor , to_pyerr,
1817 DistributedRuntime as PyDistributedRuntime ,
1918} ;
19+ use dynamo_runtime:: metrics:: prometheus_names:: kvbm_connector_namespace;
2020
2121use anyhow;
2222use dynamo_llm:: block_manager:: distributed:: { KvbmWorker , KvbmWorkerConfig } ;
@@ -92,7 +92,10 @@ impl KvConnectorWorker {
9292 ) ?
9393 . detach ( ) ;
9494
95- let kvbm_metrics = KvbmMetrics :: new ( & drt. namespace ( kvbm_connector_namespace:: KVBM_CONNECTOR_WORKER ) . unwrap ( ) ) ;
95+ let kvbm_metrics = KvbmMetrics :: new (
96+ & drt. namespace ( kvbm_connector_namespace:: KVBM_CONNECTOR_WORKER )
97+ . unwrap ( ) ,
98+ ) ;
9699
97100 tracing:: info!(
98101 "KvConnectorWorker initialized with worker_id: {}" ,
You can’t perform that action at this time.
0 commit comments