فهرست منبع

feat(jupyter): check response status for destroy kernel

orenzhang 2 ماه پیش
والد
کامیت
3a7e5a5f31
1فایلهای تغییر یافته به همراه2 افزوده شده و 1 حذف شده
  1. 2 1
      backend/open_webui/utils/code_interpreter.py

+ 2 - 1
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()