support non url task block (#3983)
This commit is contained in:
@@ -47,7 +47,6 @@ from skyvern.exceptions import (
|
||||
FailedToTakeScreenshot,
|
||||
InvalidTaskStatusTransition,
|
||||
InvalidWorkflowTaskURLState,
|
||||
MissingBrowserState,
|
||||
MissingBrowserStatePage,
|
||||
NoTOTPVerificationCodeFound,
|
||||
ScrapingFailed,
|
||||
@@ -174,9 +173,7 @@ class ForgeAgent:
|
||||
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
|
||||
)
|
||||
if browser_state is None:
|
||||
raise MissingBrowserState(workflow_run_id=workflow_run.workflow_run_id)
|
||||
|
||||
if browser_state is not None:
|
||||
working_page = await browser_state.get_working_page()
|
||||
if not working_page:
|
||||
LOG.error(
|
||||
@@ -189,6 +186,9 @@ class ForgeAgent:
|
||||
raise InvalidWorkflowTaskURLState(workflow_run.workflow_run_id)
|
||||
|
||||
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(
|
||||
url=task_url,
|
||||
|
||||
Reference in New Issue
Block a user