From 3f14615fc623e06e19a17f7b6ce8b34d57948663 Mon Sep 17 00:00:00 2001 From: Mo Zhou Date: Wed, 12 Jun 2024 10:17:42 +0800 Subject: [PATCH] fix: handle load_session error --- enterprise_gateway/services/kernels/remotemanager.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/enterprise_gateway/services/kernels/remotemanager.py b/enterprise_gateway/services/kernels/remotemanager.py index f906d68f..9cc3235f 100644 --- a/enterprise_gateway/services/kernels/remotemanager.py +++ b/enterprise_gateway/services/kernels/remotemanager.py @@ -197,7 +197,11 @@ def check_kernel_id(self, kernel_id: str) -> None: def _refresh_kernel(self, kernel_id: str) -> bool: if self.parent.availability_mode == EnterpriseGatewayConfigMixin.AVAILABILITY_REPLICATION: - self.parent.kernel_session_manager.load_session(kernel_id) + try: + self.parent.kernel_session_manager.load_session(kernel_id) + except Exception as e: + self.log.error(f"Failed to load session, kernel_id:{kernel_id}", e) + return False return self.parent.kernel_session_manager.start_session(kernel_id) # else we should throw 404 when not using an availability mode of 'replication' return False