fix task v2 block goto url issue (#1899)

This commit is contained in:
Shuchang Zheng 2025-03-06 18:27:19 -08:00 committed by GitHub
parent e720177735
commit fd934dcfe6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 45 additions and 18 deletions

View file

@ -68,6 +68,7 @@ from skyvern.forge.sdk.workflow.models.parameter import (
WorkflowParameter,
)
from skyvern.webeye.browser_factory import BrowserState
from skyvern.webeye.utils.page import SkyvernFrame
LOG = structlog.get_logger()
@ -2144,6 +2145,15 @@ class TaskV2Block(Block):
from skyvern.forge.sdk.services import task_v2_service
from skyvern.forge.sdk.workflow.models.workflow import WorkflowRunStatus
if not self.url:
browser_state = app.BROWSER_MANAGER.get_for_workflow_run(workflow_run_id)
if browser_state:
page = await browser_state.get_working_page()
if page:
current_url = await SkyvernFrame.get_url(frame=page)
if current_url != "about:blank":
self.url = current_url
if not organization_id:
raise ValueError("Running TaskV2Block requires organization_id")