updated (#16)

Signed-off-by: [email protected] <[email protected]>

Author: robertgshaw2-redhat

vllm/v1/core/kv_cache_manager.py

@@ -291,6 +291,7 @@ def allocate_slots(
             # cache a block after is has finished recving.
             self.num_cached_block[request.request_id] = len(
                 new_computed_blocks)
+            return new_blocks
 
         self.cache_blocks(
             request=request,