fix creating org token encryption issue (#3111)
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user