"""ORM models.

Importing this package registers every model on ``Base.metadata`` so that
``create_all`` (tests) and Alembic autogenerate (migrations) see them all.
"""

from legate.db.models.agent import Agent
from legate.db.models.api_key import ApiKey
from legate.db.models.approval import Approval
from legate.db.models.audit import AuditLog
from legate.db.models.chunk import Chunk
from legate.db.models.connector import Connector
from legate.db.models.credential import Credential
from legate.db.models.document import Document
from legate.db.models.knowledge_base import KnowledgeBase
from legate.db.models.membership import Membership
from legate.db.models.run import Run
from legate.db.models.step import Step
from legate.db.models.usage import UsageRecord
from legate.db.models.user import User
from legate.db.models.workflow import Workflow
from legate.db.models.workspace import Workspace

__all__ = [
    "Agent",
    "ApiKey",
    "Approval",
    "AuditLog",
    "Chunk",
    "Connector",
    "Credential",
    "Document",
    "KnowledgeBase",
    "Membership",
    "Run",
    "Step",
    "UsageRecord",
    "User",
    "Workflow",
    "Workspace",
]
