fix(disaggregation): check_status not found case crackdown

2026-03-25 21:21:39 +08:00
parent ef607c35c9
commit 6efa9b4350
1 changed files with 5 additions and 1 deletions
--- 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: