Skip to content

Commit 2b8c260

Browse files
authored
Fixed AttributeError caused by missing 'req_ids' in older vllm (LMCache#1001)
Signed-off-by: popsiclexu <[email protected]>
1 parent 44285c0 commit 2b8c260

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

lmcache/integration/vllm/vllm_v1_adapter.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -847,6 +847,27 @@ def build_connector_meta(
847847
meta.add_request(req_meta)
848848

849849
cached_reqs = scheduler_output.scheduled_cached_reqs
850+
851+
# NOTE: For backward compatibility with vllm version < 0.9.2,
852+
# In the latest vllm version, the type of scheduled_cached_reqs has
853+
# changed from list to object `CachedRequestData`
854+
if isinstance(cached_reqs, list):
855+
for i, req in enumerate(cached_reqs):
856+
request_tracker = self._request_trackers[req.req_id]
857+
request_tracker.update(req.new_token_ids, req.new_block_ids)
858+
859+
req_meta = ReqMeta.from_request_tracker(
860+
request_tracker,
861+
self._block_size,
862+
self._lmcache_chunk_size,
863+
load_spec=None,
864+
skip_save=force_skip_save,
865+
discard_partial_chunks=self._discard_partial_chunks,
866+
)
867+
if req_meta is not None:
868+
meta.add_request(req_meta)
869+
return meta
870+
850871
for i, req_id in enumerate(cached_reqs.req_ids):
851872
request_tracker = self._request_trackers[req_id]
852873
num_new_tokens = scheduler_output.num_scheduled_tokens[req_id]

0 commit comments

Comments
 (0)