diff --git a/python/sglang/srt/mem_cache/storage/mooncake_store/mooncake_store.py b/python/sglang/srt/mem_cache/storage/mooncake_store/mooncake_store.py index a82a2a413..bef26257b 100644 --- a/python/sglang/srt/mem_cache/storage/mooncake_store/mooncake_store.py +++ b/python/sglang/srt/mem_cache/storage/mooncake_store/mooncake_store.py @@ -159,6 +159,7 @@ class MooncakeStore(HiCacheStorage): def batch_set( self, keys: List[str], + values: Optional[List[torch.Tensor]] = None, target_location: Optional[List[int]] = None, target_sizes: Optional[List[int]] = None, ) -> bool: