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: