21 lines
893 B
Python
21 lines
893 B
Python
from sqlalchemy import Boolean, Float, Integer, String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.models.base import Base, TimestampMixin
|
|
|
|
|
|
class ModelConfig(Base, TimestampMixin):
|
|
__tablename__ = "model_configs"
|
|
|
|
id: Mapped[str] = mapped_column(String(32), primary_key=True)
|
|
name: Mapped[str] = mapped_column(String(64))
|
|
provider: Mapped[str] = mapped_column(String(32))
|
|
api_base: Mapped[str] = mapped_column(String(512))
|
|
api_key: Mapped[str] = mapped_column(String(256))
|
|
model_name: Mapped[str] = mapped_column(String(128))
|
|
weight: Mapped[int] = mapped_column(Integer, default=1)
|
|
max_tokens: Mapped[int] = mapped_column(Integer, default=4096)
|
|
temperature: Mapped[float] = mapped_column(Float, default=0.7)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
priority: Mapped[int] = mapped_column(Integer, default=0)
|