new lines for get_workflow_run parameter - code style update

Skyvern-AI · Dec 23, 2024 · 1fcf7bc · 1fcf7bc
1 parent e10d9e3
commit 1fcf7bc
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 5 deletions.
diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py
@@ -249,7  249,8 @@ async def execute_step(
         workflow_run: WorkflowRun | None = None
         if task.workflow_run_id:
             workflow_run = await app.DATABASE.get_workflow_run(
-                workflow_run_id=task.workflow_run_id, organization_id=organization.organization_id
                 workflow_run_id=task.workflow_run_id,
                 organization_id=organization.organization_id,
             )
             if workflow_run and workflow_run.status == WorkflowRunStatus.canceled:
                 LOG.info(

diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -312,7  312,8 @@ async def cancel_workflow_run(
     x_api_key: Annotated[str | None, Header()] = None,
 ) -> None:
     workflow_run = await app.DATABASE.get_workflow_run(
-        workflow_run_id=workflow_run_id, organization_id=current_org.organization_id
         workflow_run_id=workflow_run_id,
         organization_id=current_org.organization_id,
     )
     if not workflow_run:
         raise HTTPException(

diff --git a/skyvern/forge/sdk/routes/streaming.py b/skyvern/forge/sdk/routes/streaming.py
@@ -172,7  172,8 @@ async def workflow_run_streaming(
                 return
 
             workflow_run = await app.DATABASE.get_workflow_run(
-                workflow_run_id=workflow_run_id, organization_id=organization_id
                 workflow_run_id=workflow_run_id,
                 organization_id=organization_id,
             )
             if not workflow_run or workflow_run.organization_id != organization_id:
                 LOG.info(

diff --git a/skyvern/forge/sdk/services/observer_service.py b/skyvern/forge/sdk/services/observer_service.py
@@ -484,7  484,8 @@ async def handle_block_result(
             )
     # refresh workflow run model
     return await app.WORKFLOW_SERVICE.get_workflow_run(
-        workflow_run_id=workflow_run_id, organization_id=workflow.organization_id
         workflow_run_id=workflow_run_id,
         organization_id=workflow.organization_id,
     )
 
 

diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
@@ -360,7  360,8 @@ async def execute_workflow(
                 return workflow_run
 
         refreshed_workflow_run = await app.DATABASE.get_workflow_run(
-            workflow_run_id=workflow_run.workflow_run_id, organization_id=organization_id
             workflow_run_id=workflow_run.workflow_run_id,
             organization_id=organization_id,
         )
         if refreshed_workflow_run and refreshed_workflow_run.status not in (
             WorkflowRunStatus.canceled,