Bitwarden Security Upgrade (#900)

2024-10-02 15:16:08 -07:00
parent 4f6feae03e
commit 36135a613b
9 changed files with 124 additions and 6 deletions
--- a/skyvern/forge/sdk/executor/async_executor.py
+++ b/skyvern/forge/sdk/executor/async_executor.py
@@ -99,8 +99,14 @@ class BackgroundTaskExecutor(AsyncExecutor):
            "Executing workflow using background task executor",
            workflow_run_id=workflow_run_id,
        )
+
+        organization = await app.DATABASE.get_organization(organization_id)
+        if organization is None:
+            raise OrganizationNotFound(organization_id)
+
        background_tasks.add_task(
            app.WORKFLOW_SERVICE.execute_workflow,
            workflow_run_id=workflow_run_id,
            api_key=api_key,
+            organization=organization,
        )