46 lines
2.2 KiB
Python
46 lines
2.2 KiB
Python
|
"""AI configuration models."""
|
||
|
|
from sqlalchemy import Boolean, Float, ForeignKey, Integer, JSON, String, Text
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
|
|
||
|
|
from app.models.base import Base, TimestampMixin, UUIDMixin
|
||
|
|
|
||
|
|
|
||
|
|
class AIProviderConfig(Base, UUIDMixin, TimestampMixin):
|
||
|
|
"""AI provider configuration (OpenAI, Anthropic, etc.)."""
|
||
|
|
|
||
|
|
__tablename__ = "ai_provider_configs"
|
||
|
|
|
||
|
|
name: Mapped[str] = mapped_column(String(128), nullable=False)
|
||
|
|
provider: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
|
||
|
|
base_url: Mapped[str | None] = mapped_column(String(512), nullable=True)
|
||
|
|
api_key_encrypted: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
|
default_model: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
||
|
|
timeout: Mapped[int] = mapped_column(Integer, default=60, nullable=False)
|
||
|
|
max_retries: Mapped[int] = mapped_column(Integer, default=3, nullable=False)
|
||
|
|
rate_limit_rpm: Mapped[int] = mapped_column(Integer, default=60, nullable=False)
|
||
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||
|
|
|
||
|
|
|
||
|
|
class AITaskConfig(Base, UUIDMixin, TimestampMixin):
|
||
|
|
"""AI task configuration (which model/skill for which task)."""
|
||
|
|
|
||
|
|
__tablename__ = "ai_task_configs"
|
||
|
|
|
||
|
|
task_type: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
|
||
|
|
name: Mapped[str] = mapped_column(String(128), nullable=False)
|
||
|
|
provider_config_id: Mapped[str | None] = mapped_column(
|
||
|
|
ForeignKey("ai_provider_configs.id", ondelete="SET NULL"), nullable=True
|
||
|
|
)
|
||
|
|
model: Mapped[str] = mapped_column(String(128), nullable=False)
|
||
|
|
skill_id: Mapped[str | None] = mapped_column(
|
||
|
|
ForeignKey("skills.id", ondelete="SET NULL"), nullable=True
|
||
|
|
)
|
||
|
|
temperature: Mapped[float] = mapped_column(Float, default=0.3, nullable=False)
|
||
|
|
max_tokens: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||
|
|
top_p: Mapped[float] = mapped_column(Float, default=1.0, nullable=False)
|
||
|
|
system_prompt_override: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||
|
|
fallback_config_id: Mapped[str | None] = mapped_column(
|
||
|
|
ForeignKey("ai_task_configs.id", ondelete="SET NULL"), nullable=True
|
||
|
|
)
|
||
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|