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
4 changed files with 45 additions and 18 deletions

View File

@@ -132,13 +132,11 @@ class BrowserManager:
) -> BrowserState:
parent_workflow_run_id = workflow_run.parent_workflow_run_id
workflow_run_id = workflow_run.workflow_run_id
browser_state = self.get_for_workflow_run(workflow_run_id=workflow_run_id)
if parent_workflow_run_id:
browser_state = self.get_for_workflow_run(workflow_run_id=parent_workflow_run_id)
if browser_state:
self.pages[workflow_run_id] = browser_state
if browser_state is not None:
browser_state = self.get_for_workflow_run(
workflow_run_id=workflow_run_id, parent_workflow_run_id=parent_workflow_run_id
)
if browser_state:
self.pages[workflow_run_id] = browser_state
return browser_state
if browser_session_id:
@@ -193,9 +191,15 @@ class BrowserManager:
)
return browser_state
def get_for_workflow_run(self, workflow_run_id: str) -> BrowserState | None:
def get_for_workflow_run(
self, workflow_run_id: str, parent_workflow_run_id: str | None = None
) -> BrowserState | None:
if workflow_run_id in self.pages:
return self.pages[workflow_run_id]
if parent_workflow_run_id and parent_workflow_run_id in self.pages:
return self.pages[parent_workflow_run_id]
return None
def set_video_artifact_for_task(self, task: Task, artifacts: list[VideoArtifact]) -> None: