update token_type to string instead of enum (#3087)

This commit is contained in:
Shuchang Zheng
2025-08-01 11:06:34 -07:00
committed by GitHub
parent 4407ff198b
commit 30eb580306
3 changed files with 47 additions and 3 deletions

View File

@@ -3,6 +3,7 @@ from enum import StrEnum
class OrganizationAuthTokenType(StrEnum):
api = "api"
onepassword_service_account = "onepassword_service_account"
class TaskType(StrEnum):

View File

@@ -5,7 +5,6 @@ from sqlalchemy import (
Boolean,
Column,
DateTime,
Enum,
ForeignKey,
Index,
Integer,
@@ -18,7 +17,7 @@ from sqlalchemy import (
from sqlalchemy.ext.asyncio import AsyncAttrs
from sqlalchemy.orm import DeclarativeBase
from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType, TaskType
from skyvern.forge.sdk.db.enums import TaskType
from skyvern.forge.sdk.db.id import (
generate_action_id,
generate_ai_suggestion_id,
@@ -167,7 +166,7 @@ class OrganizationAuthTokenModel(Base):
)
organization_id = Column(String, ForeignKey("organizations.organization_id"), index=True, nullable=False)
token_type = Column(Enum(OrganizationAuthTokenType), nullable=False)
token_type = Column(String, nullable=False)
token = Column(String, index=True, nullable=False)
valid = Column(Boolean, nullable=False, default=True)