[Bugfix] Fix slice operation when chunk size mismatch (sgl-project#6697)

ShangmingCai · xwu-intel · commit 65585095d370 · 2025-06-17T06:06:22.000+03:00
Signed-off-by: Shangming Cai &lt;caishangming@linux.alibaba.com&gt;
diff --git a/python/sglang/srt/disaggregation/mooncake/conn.py b/python/sglang/srt/disaggregation/mooncake/conn.py
@@ -354,7 +354,7 @@ def transfer_worker(
                             kv_chunk.prefill_kv_indices
                         ):
                             kv_chunk.prefill_kv_indices = kv_chunk.prefill_kv_indices[
-                                len(chunked_dst_kv_indice)
+                                : len(chunked_dst_kv_indice)
                             ]
                             logger.warning(
                                 f"len(chunked_dst_kv_indice) = {len(chunked_dst_kv_indice)}, len(kv_chunk.prefill_kv_indices) = {len(kv_chunk.prefill_kv_indices)}"

Original file line number	Diff line number	Diff line change
`@@ -354,7 +354,7 @@ def transfer_worker(`
`354`	`354`	`kv_chunk.prefill_kv_indices`
`355`	`355`	`):`
`356`	`356`	`kv_chunk.prefill_kv_indices = kv_chunk.prefill_kv_indices[`
`357`		`- len(chunked_dst_kv_indice)`
	`357`	`+ : len(chunked_dst_kv_indice)`
`358`	`358`	`]`
`359`	`359`	`logger.warning(`
`360`	`360`	`f"len(chunked_dst_kv_indice) = {len(chunked_dst_kv_indice)}, len(kv_chunk.prefill_kv_indices) = {len(kv_chunk.prefill_kv_indices)}"`