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

Update main.py

Browse Source
This commit is contained in:
Nicolas 2024-05-09 18:00:58 -07:00
parent 08d6c59a78
commit 73687822ad
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,6 +8,7 @@ app = FastAPI()
class UrlModel(BaseModel): class UrlModel(BaseModel):
url: str url: str
wait: int = None
browser: Browser = None browser: Browser = None
@ -29,7 +30,9 @@ async def shutdown_event():
async def root(body: UrlModel): async def root(body: UrlModel):
context = await browser.new_context() context = await browser.new_context()
page = await context.new_page() page = await context.new_page()
await page.goto(body.url) await page.goto(body.url, timeout=15000) # Set max timeout to 15s
if body.wait: # Check if wait parameter is provided in the request body
await page.wait_for_timeout(body.wait) # Convert seconds to milliseconds for playwright
page_content = await page.content() page_content = await page.content()
await context.close() await context.close()
json_compatible_item_data = {"content": page_content} json_compatible_item_data = {"content": page_content}