support non url task block (#3983)

This commit is contained in:
LawyZheng
2025-11-13 14:28:45 +08:00
committed by GitHub
parent bf4808f336
commit 40cc6c7b47

View File

@@ -47,7 +47,6 @@ from skyvern.exceptions import (
FailedToTakeScreenshot, FailedToTakeScreenshot,
InvalidTaskStatusTransition, InvalidTaskStatusTransition,
InvalidWorkflowTaskURLState, InvalidWorkflowTaskURLState,
MissingBrowserState,
MissingBrowserStatePage, MissingBrowserStatePage,
NoTOTPVerificationCodeFound, NoTOTPVerificationCodeFound,
ScrapingFailed, ScrapingFailed,
@@ -174,9 +173,7 @@ class ForgeAgent:
browser_state = app.BROWSER_MANAGER.get_for_workflow_run( browser_state = app.BROWSER_MANAGER.get_for_workflow_run(
workflow_run_id=workflow_run.workflow_run_id, parent_workflow_run_id=workflow_run.parent_workflow_run_id workflow_run_id=workflow_run.workflow_run_id, parent_workflow_run_id=workflow_run.parent_workflow_run_id
) )
if browser_state is None: if browser_state is not None:
raise MissingBrowserState(workflow_run_id=workflow_run.workflow_run_id)
working_page = await browser_state.get_working_page() working_page = await browser_state.get_working_page()
if not working_page: if not working_page:
LOG.error( LOG.error(
@@ -189,6 +186,9 @@ class ForgeAgent:
raise InvalidWorkflowTaskURLState(workflow_run.workflow_run_id) raise InvalidWorkflowTaskURLState(workflow_run.workflow_run_id)
task_url = working_page.url task_url = working_page.url
else:
LOG.info("No browser state found for workflow run, setting task url to empty string")
task_url = ""
task = await app.DATABASE.create_task( task = await app.DATABASE.create_task(
url=task_url, url=task_url,