From d229beeaf878ec43e2adb6ff840993afb97ee58d Mon Sep 17 00:00:00 2001
From: unclecode <unclecode@kidocode.com>
Date: Sat, 7 Mar 2026 06:15:54 +0000
Subject: [PATCH] fix: add wait_for_images option to screenshot endpoint
 (#1792)

From PR #1792 by @Br1an67
---
 deploy/docker/schemas.py | 1 +
 deploy/docker/server.py  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/deploy/docker/schemas.py b/deploy/docker/schemas.py
index 21d47fc4..ef75ce8b 100644
--- a/deploy/docker/schemas.py
+++ b/deploy/docker/schemas.py
@@ -73,6 +73,7 @@ class HTMLRequest(BaseModel):
 class ScreenshotRequest(BaseModel):
     url: str
     screenshot_wait_for: Optional[float] = 2
+    wait_for_images: Optional[bool] = False
     output_path: Optional[str] = None
 
 class PDFRequest(BaseModel):
diff --git a/deploy/docker/server.py b/deploy/docker/server.py
index 210f421b..7f22c06b 100644
--- a/deploy/docker/server.py
+++ b/deploy/docker/server.py
@@ -404,7 +404,7 @@ async def generate_screenshot(
     validate_url_scheme(body.url)
     crawler = None
     try:
-        cfg = CrawlerRunConfig(screenshot=True, screenshot_wait_for=body.screenshot_wait_for)
+        cfg = CrawlerRunConfig(screenshot=True, screenshot_wait_for=body.screenshot_wait_for, wait_for_images=body.wait_for_images)
         crawler = await get_crawler(get_default_browser_config())
         results = await crawler.arun(url=body.url, config=cfg)
         if not results[0].success: