From 6efa9b4350b60b22f1e68c5a0d2c4008ec5649c9 Mon Sep 17 00:00:00 2001 From: laoyao0822 Date: Wed, 25 Mar 2026 21:21:39 +0800 Subject: [PATCH] fix(disaggregation): check_status not found case crackdown --- python/sglang/srt/disaggregation/common/conn.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/disaggregation/common/conn.py b/python/sglang/srt/disaggregation/common/conn.py index 399acde06..20b1424e4 100644 --- a/python/sglang/srt/disaggregation/common/conn.py +++ b/python/sglang/srt/disaggregation/common/conn.py @@ -175,7 +175,11 @@ class CommonKVManager(BaseKVManager): ) def check_status(self, bootstrap_room: int) -> KVPoll: - return self.request_status[bootstrap_room] + # return self.request_status[bootstrap_room] + if bootstrap_room is None: + return KVPoll.Failed + kvpoll = self.request_status.get(bootstrap_room, KVPoll.Failed) + return kvpoll def update_status(self, bootstrap_room: int, status: KVPoll): if bootstrap_room not in self.request_status: