From 3a7e5a5f3170b69987813d1767084680f758cb71 Mon Sep 17 00:00:00 2001 From: orenzhang Date: Tue, 4 Mar 2025 11:59:39 +0800 Subject: [PATCH] feat(jupyter): check response status for destroy kernel --- backend/open_webui/utils/code_interpreter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/utils/code_interpreter.py b/backend/open_webui/utils/code_interpreter.py index 883ff3713..b9f5a0079 100644 --- a/backend/open_webui/utils/code_interpreter.py +++ b/backend/open_webui/utils/code_interpreter.py @@ -54,7 +54,8 @@ class JupyterCodeExecuter: async def __aexit__(self, exc_type, exc_val, exc_tb): if self.kernel_id: try: - await self.session.delete(f"/api/kernels/{self.kernel_id}", params=self.params) + async with self.session.delete(f"/api/kernels/{self.kernel_id}", params=self.params) as response: + response.raise_for_status() except Exception as err: logger.exception("close kernel failed, %s", err) await self.session.close()