qzl
6641eba9df
replace monolithic migration with ordered scripts, include profiles/sessions in migration, and verify full downgrade/upgrade cycle for clean Supabase bootstrap
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
from enum import Enum
|
|
|
|
from sqlalchemy import String
|
|
from sqlalchemy.dialects.postgresql import JSONB, UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from core.db.base import Base, SoftDeleteMixin, TimestampMixin
|
|
|
|
|
|
class UserAgentStatus(str, Enum):
|
|
ACTIVE = "active"
|
|
PAUSED = "paused"
|
|
MIGRATING = "migrating"
|
|
|
|
|
|
class AgentType(str, Enum):
|
|
INTENT_RECOGNITION = "INTENT_RECOGNITION"
|
|
TASK_EXECUTION = "TASK_EXECUTION"
|
|
RESULT_REPORTING = "RESULT_REPORTING"
|
|
|
|
|
|
class UserAgent(TimestampMixin, SoftDeleteMixin, Base):
|
|
__tablename__: str = "user_agents"
|
|
__table_args__ = {"extend_existing": True}
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), primary_key=True, default=uuid.uuid4
|
|
)
|
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
nullable=False,
|
|
unique=True,
|
|
)
|
|
llm_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
nullable=False,
|
|
)
|
|
agent_type: Mapped[AgentType] = mapped_column(
|
|
String(20),
|
|
nullable=False,
|
|
)
|
|
config: Mapped[dict] = mapped_column(
|
|
JSONB,
|
|
nullable=False,
|
|
server_default="{}",
|
|
)
|
|
status: Mapped[UserAgentStatus] = mapped_column(
|
|
String(20),
|
|
nullable=False,
|
|
default=UserAgentStatus.ACTIVE,
|
|
)
|
|
created_by: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
nullable=True,
|
|
)
|
|
updated_by: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
nullable=True,
|
|
)
|