From 027b6b77f8cb66423ecc310c381ec5aa61c9df84 Mon Sep 17 00:00:00 2001 From: LawyZheng Date: Wed, 6 Aug 2025 13:39:36 +0800 Subject: [PATCH] fix creating org token encryption issue (#3111) --- skyvern/forge/sdk/services/org_auth_token_service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/skyvern/forge/sdk/services/org_auth_token_service.py b/skyvern/forge/sdk/services/org_auth_token_service.py index ad05b331..ffe23580 100644 --- a/skyvern/forge/sdk/services/org_auth_token_service.py +++ b/skyvern/forge/sdk/services/org_auth_token_service.py @@ -2,7 +2,7 @@ from datetime import timedelta import structlog -from skyvern.forge.app import DATABASE +from skyvern.forge import app from skyvern.forge.sdk.core import security from skyvern.forge.sdk.schemas.organizations import OrganizationAuthToken, OrganizationAuthTokenType @@ -20,7 +20,7 @@ async def create_org_api_token(org_id: str) -> OrganizationAuthToken: The API token created for the specified org_id. """ # get the organization - organization = await DATABASE.get_organization(org_id) + organization = await app.DATABASE.get_organization(org_id) if not organization: raise Exception(f"Organization id {org_id} not found") @@ -30,7 +30,7 @@ async def create_org_api_token(org_id: str) -> OrganizationAuthToken: expires_delta=API_KEY_LIFETIME, ) # generate OrganizationAutoToken - org_auth_token = await DATABASE.create_org_auth_token( + org_auth_token = await app.DATABASE.create_org_auth_token( organization_id=org_id, token=api_key, token_type=OrganizationAuthTokenType.api,