vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add organization_id filter for get_workflow and get_workflow_run (#1422)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-12-22 17:49:33 -08:00
committed by GitHub
parent b256bace6a
commit 2e37542218
7 changed files with 44 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -311,7 +311,10 @@ async def cancel_workflow_run(
current_org: Organization = Depends(org_auth_service.get_current_org),
x_api_key: Annotated[str | None, Header()] = None,
) -> None:
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=current_org.organization_id,
)
if not workflow_run:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,

5
skyvern/forge/sdk/routes/streaming.py
View File

@@ -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",