github/v-firecrawl
github/v-firecrawl
0
Fork 0
Code Actions

Add timeout to class and provide default.

Browse Source
This commit is contained in:
Matt Joyce 2024-06-01 13:10:14 +10:00
parent 03ba4a9fa7
commit 2a39b5382b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/playwright-service/main.py
View File

@ -15,6 +15,7 @@ app = FastAPI()
class UrlModel(BaseModel): class UrlModel(BaseModel):
url: str url: str
wait: int = None wait: int = None
timeout: int = 15000
browser: Browser = None browser: Browser = None
@ -50,7 +51,7 @@ async def root(body: UrlModel):
await page.goto( await page.goto(
body.url, body.url,
wait_until="load", wait_until="load",
timeout=body.timeout if body.timeout else 15000, timeout=body.timeout,
) )
# Wait != timeout. Wait is the time to wait after the page is loaded - useful in some cases were "load" / "networkidle" is not enough # Wait != timeout. Wait is the time to wait after the page is loaded - useful in some cases were "load" / "networkidle" is not enough
if body.wait: if body.wait: