add timeout for page.content() (#541)

This commit is contained in:
LawyZheng
2024-07-03 17:17:24 +08:00
committed by GitHub
parent 22566f01ce
commit b791cc1ae5
3 changed files with 20 additions and 4 deletions

View File

@@ -52,7 +52,7 @@ from skyvern.webeye.actions.handler import ActionHandler
from skyvern.webeye.actions.models import AgentStepOutput, DetailedAgentStepOutput
from skyvern.webeye.actions.responses import ActionResult
from skyvern.webeye.browser_factory import BrowserState
from skyvern.webeye.scraper.scraper import ElementTreeFormat, ScrapedPage, scrape_website
from skyvern.webeye.scraper.scraper import ElementTreeFormat, ScrapedPage, get_page_content, scrape_website
LOG = structlog.get_logger()
@@ -786,7 +786,7 @@ class ForgeAgent:
)
try:
html = await browser_state.page.content()
html = await get_page_content(browser_state.page)
await app.ARTIFACT_MANAGER.create_artifact(
step=step,
artifact_type=ArtifactType.HTML_ACTION,