🐛 fix health check error not catched

cscs181 · Dec 4, 2023 · b7bbd64 · b7bbd64
1 parent e6acb91
commit b7bbd64
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/plugins/health_check/__init__.py b/src/plugins/health_check/__init__.py
@@ -2,7 +2,7 @@
 @Author         : yanyongyu
 @Date           : 2022-10-10 06:57:31
 @LastEditors    : yanyongyu
-@LastEditTime   : 2023-12-02 14:54:09
+@LastEditTime   : 2023-12-04 16:42:20
 @Description    : Health check plugin
 @GitHub         : https://github.com/yanyongyu
 """
@@ -45,7 +45,8 @@ async def health_check():
 
     # check playwright connection
     try:
-        assert get_browser().is_connected()
+        if not get_browser().is_connected():
+            raise RuntimeError("Playwright browser is not connected.")
     except Exception as e:
         logger.opt(exception=e).error("Playwright connection health check failed.")
         return JSONResponse(