| 1234567891011121314151617181920212223 |
- from core_db import AuditMixin, Base, EntityMixin, VersionMixin
- from sqlalchemy import Float, Integer, String, Text
- from sqlalchemy.dialects.sqlite import JSON
- from sqlalchemy.orm import Mapped, mapped_column
- class ModelDefinition(EntityMixin, AuditMixin, VersionMixin, Base):
- __tablename__ = "model_definition"
- code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
- name: Mapped[str] = mapped_column(String(128))
- provider_type: Mapped[str] = mapped_column(String(32), default="openai_compatible", index=True)
- provider_base_url: Mapped[str] = mapped_column(String(512))
- provider_api_key: Mapped[str | None] = mapped_column(Text, nullable=True)
- model_name: Mapped[str] = mapped_column(String(128), index=True)
- status: Mapped[str] = mapped_column(String(32), default="active", index=True)
- description: Mapped[str | None] = mapped_column(Text, nullable=True)
- capabilities_json: Mapped[list[str]] = mapped_column(JSON, default=list)
- context_window: Mapped[int | None] = mapped_column(Integer, nullable=True)
- max_output_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
- default_temperature: Mapped[float | None] = mapped_column(Float, nullable=True)
- timeout_seconds: Mapped[float] = mapped_column(Float, default=60.0)
- metadata_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
