diff --git a/skyvern/forge/sdk/workflow/models/block.py b/skyvern/forge/sdk/workflow/models/block.py index 83b64db6fb..2b514dd203 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -49,7 +49,7 @@ from skyvern.forge.sdk.core.validators import prepend_scheme_and_validate_url from skyvern.forge.sdk.db.enums import TaskType from skyvern.forge.sdk.schemas.observers import ObserverTaskStatus -from skyvern.forge.sdk.schemas.tasks import ProxyLocation, Task, TaskOutput, TaskStatus +from skyvern.forge.sdk.schemas.tasks import Task, TaskOutput, TaskStatus from skyvern.forge.sdk.workflow.context_manager import BlockMetadata, WorkflowRunContext from skyvern.forge.sdk.workflow.exceptions import ( FailedToFormatJinjaStyleParameter, @@ -2106,12 +2106,15 @@ async def execute( organization = await app.DATABASE.get_organization(organization_id) if not organization: raise ValueError(f"Organization not found {organization_id}") + workflow_run = await app.DATABASE.get_workflow_run(workflow_run_id, organization_id) + if not workflow_run: + raise ValueError(f"WorkflowRun not found {workflow_run_id} when running TaskV2Block") observer_task = await observer_service.initialize_observer_task( organization, user_prompt=self.prompt, user_url=self.url, parent_workflow_run_id=workflow_run_id, - proxy_location=ProxyLocation.NONE, + proxy_location=workflow_run.proxy_location, ) await app.DATABASE.update_observer_cruise( observer_task.observer_cruise_id, status=ObserverTaskStatus.queued, organization_id=organization_id