From 887a8a9b6484b24279296c582b5837e5d67bfead Mon Sep 17 00:00:00 2001 From: Jonathan Dobson Date: Tue, 11 Nov 2025 12:44:17 -0500 Subject: [PATCH] fix get_workflow_for_workflow_run query; filter on wr org, not wf org (#3963) --- skyvern/forge/sdk/db/client.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 3176a022..a36b4232 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -1527,14 +1527,15 @@ class AgentDB: if exclude_deleted: get_workflow_query = get_workflow_query.filter(WorkflowModel.deleted_at.is_(None)) - if organization_id: - get_workflow_query = get_workflow_query.filter_by(organization_id=organization_id) get_workflow_query = get_workflow_query.join( WorkflowRunModel, WorkflowRunModel.workflow_id == WorkflowModel.workflow_id, ) + if organization_id: + get_workflow_query = get_workflow_query.filter(WorkflowRunModel.organization_id == organization_id) + get_workflow_query = get_workflow_query.filter(WorkflowRunModel.workflow_run_id == workflow_run_id) async with self.Session() as session: if workflow := (await session.scalars(get_workflow_query)).first():