| from sqlalchemy import text, Integer, Text, func |
| from sqlalchemy.orm import Mapped, mapped_column |
|
|
| from app.models.base import Base |
|
|
|
|
| class ModelAlias(Base): |
| __tablename__ = "model_aliases" |
|
|
| id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True) |
| alias: Mapped[str] = mapped_column(Text, unique=True, nullable=False) |
| provider: Mapped[str] = mapped_column(Text, nullable=False) |
| provider_model: Mapped[str] = mapped_column(Text, nullable=False) |
| description: Mapped[str] = mapped_column(Text, default="") |
| max_tokens_override: Mapped[int] = mapped_column(Integer, default=0) |
| prompt_template_id: Mapped[int | None] = mapped_column(Integer, nullable=True) |
| auto_select: Mapped[int] = mapped_column(Integer, default=0) |
| fallback_provider: Mapped[str | None] = mapped_column(Text, nullable=True) |
| fallback_model: Mapped[str | None] = mapped_column(Text, nullable=True) |
| is_active: Mapped[int] = mapped_column(Integer, default=1) |
| created_at: Mapped[str] = mapped_column(Text, server_default=text("CURRENT_TIMESTAMP")) |
|
|
