agent-platform/services/model-gateway-service/app/db/models.py

from core_db import AuditMixin, Base, EntityMixin
from sqlalchemy import Float, Integer, String, Text
from sqlalchemy import JSON
from sqlalchemy.orm import Mapped, mapped_column


class ModelDefinition(EntityMixin, AuditMixin, Base):
    __tablename__ = "model_definition"

    code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    name: Mapped[str] = mapped_column(String(128))
    provider_id: Mapped[str | None] = mapped_column(String(36), nullable=True, index=True)
    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)


class ModelProviderDefinition(EntityMixin, AuditMixin, Base):
    __tablename__ = "model_provider_definition"

    name: Mapped[str] = mapped_column(String(128))
    provider_type: Mapped[str] = mapped_column(String(32), default="openai_compatible", index=True)
    base_url: Mapped[str] = mapped_column(String(512))
    api_key: Mapped[str | None] = mapped_column(Text, nullable=True)
    models_json: Mapped[list[dict]] = mapped_column(JSON, default=list)
    default_model: Mapped[str | None] = mapped_column(String(128), nullable=True)
    extra_config_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)