add organization_id filter for get_workflow and get_workflow_run (#1422)

2024-12-22 17:49:33 -08:00
parent b256bace6a
commit 2e37542218
7 changed files with 44 additions and 18 deletions
--- a/skyvern/forge/sdk/routes/streaming.py
+++ b/skyvern/forge/sdk/routes/streaming.py
@@ -171,7 +171,10 @@ async def workflow_run_streaming(
                )
                return

-            workflow_run = await app.DATABASE.get_workflow_run(workflow_run_id=workflow_run_id)
+            workflow_run = await app.DATABASE.get_workflow_run(
+                workflow_run_id=workflow_run_id,
+                organization_id=organization_id,
+            )
            if not workflow_run or workflow_run.organization_id != organization_id:
                LOG.info(
                    "WofklowRun Streaming: Workflow not found",