PMFlow/backend/app/models/model_version.py

21 lines
848 B
Python

"""模型版本表"""
from .base import Base
from .types import JSONB
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy import BigInteger, Integer, VARCHAR, DECIMAL, Boolean
from typing import Optional
class ModelVersion(Base):
__tablename__ = "model_versions"
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
name: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
provider: Mapped[str] = mapped_column(VARCHAR(100), nullable=False)
version: Mapped[str] = mapped_column(VARCHAR(100), nullable=False)
capability_boundaries: Mapped[dict] = mapped_column(JSONB, nullable=False)
cost_per_1k_tokens: Mapped[Optional[float]] = mapped_column(
DECIMAL(10, 6), nullable=True
)
is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)