Skip to content

Commit 46b0779

Browse files
authored
[BugFix] Update KV block hash type from BlockHash to ExternalBlockHash in kv_events_subscriber - #26264 (#26265)
Signed-off-by: atalhens <[email protected]>
1 parent de34258 commit 46b0779

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/online_serving/kv_events_subscriber.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import zmq
77
from msgspec.msgpack import Decoder
88

9-
from vllm.v1.core.kv_cache_utils import BlockHash
9+
from vllm.v1.core.kv_cache_utils import ExternalBlockHash
1010

1111

1212
#
@@ -24,16 +24,16 @@ class KVCacheEvent(
2424

2525

2626
class BlockStored(KVCacheEvent):
27-
block_hashes: list[BlockHash]
28-
parent_block_hash: Optional[BlockHash]
27+
block_hashes: list[ExternalBlockHash]
28+
parent_block_hash: Optional[ExternalBlockHash]
2929
token_ids: list[int]
3030
block_size: int
3131
lora_id: Optional[int]
3232
medium: Optional[str]
3333

3434

3535
class BlockRemoved(KVCacheEvent):
36-
block_hashes: list[BlockHash]
36+
block_hashes: list[ExternalBlockHash]
3737
medium: Optional[str]
3838

3939

0 commit comments

Comments
 (0)