From c2e0913e17fc9cac0e0324c999e5cbf88231ed9f Mon Sep 17 00:00:00 2001
From: Cheng Wan <54331508+ch-wan@users.noreply.github.com>
Date: Mon, 29 Dec 2025 13:13:24 -0800
Subject: [PATCH] Fix extend_input_len calculation in decode.py (#16103)

---
 python/sglang/srt/disaggregation/decode.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sglang/srt/disaggregation/decode.py b/python/sglang/srt/disaggregation/decode.py
index b58f713a1..f4ef07e41 100644
--- a/python/sglang/srt/disaggregation/decode.py
+++ b/python/sglang/srt/disaggregation/decode.py
@@ -690,7 +690,7 @@ class DecodePreallocQueue:
 
         # populate metadata
         req.fill_ids = req.origin_input_ids + req.output_ids
-        req.extend_input_len = len(req.origin_input_ids)
+        req.extend_input_len = len(req.fill_ids)
 
         return kv_loc