enhance error handling for browser initialization and cleanup processes

2025-03-31 14:50:11 +02:00
parent ab715342b1
commit 6db86a176b
1 changed files with 6 additions and 1 deletions
@@ -128,6 +128,8 @@ async def visit_url(url: str, x_api_key: Optional[str] = Header(None), backgroun

        # Get browser instance
        browser_instance = await get_browser()
+        if browser_instance is None:
+            raise HTTPException(status_code=500, detail="Failed to initialize browser")

        # Create new page
        page = await browser_instance.newPage()
@@ -156,7 +158,10 @@ async def visit_url(url: str, x_api_key: Optional[str] = Header(None), backgroun
        logger.error(f"Error processing request: {str(e)}")
        # Ensure cleanup happens even on error
        if page:
+            try:
                await cleanup_resources(page, False)
+            except Exception as cleanup_error:
+                logger.error(f"Error during error cleanup: {str(cleanup_error)}")
        raise HTTPException(status_code=500, detail=str(e))

@app.on_event("shutdown")