e4959d584f
主要更新： - 更新代理商端文档，明确项目由品牌方分配流程 - 新增Brief配置详情页(已配置)设计稿 - 完善工作台紧急待办中品牌新任务功能 - 整理Pencil设计文件中代理商端页面顺序 - 新增后端FastAPI框架及核心API - 新增前端Next.js页面和组件库 - 添加.gitignore排除构建和缓存文件 Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
"""
|
|
AI 配置模型
|
|
"""
|
|
from typing import TYPE_CHECKING, Optional
|
|
from datetime import datetime
|
|
from sqlalchemy import String, Text, Float, Integer, ForeignKey, DateTime
|
|
from app.models.types import JSONType
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.models.base import Base, TimestampMixin
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.tenant import Tenant
|
|
|
|
|
|
class AIConfig(Base, TimestampMixin):
|
|
"""AI 服务配置表"""
|
|
__tablename__ = "ai_configs"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
tenant_id: Mapped[str] = mapped_column(
|
|
String(64),
|
|
ForeignKey("tenants.id", ondelete="CASCADE"),
|
|
unique=True,
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
|
|
# 提供商配置
|
|
provider: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
base_url: Mapped[str] = mapped_column(String(500), nullable=False)
|
|
api_key_encrypted: Mapped[str] = mapped_column(Text, nullable=False)
|
|
|
|
# 模型配置 (JSON)
|
|
# {"text": "gpt-4o", "vision": "gpt-4o", "audio": "whisper-1"}
|
|
models: Mapped[dict] = mapped_column(JSONType, nullable=False)
|
|
|
|
# 参数配置
|
|
temperature: Mapped[float] = mapped_column(Float, default=0.7, nullable=False)
|
|
max_tokens: Mapped[int] = mapped_column(Integer, default=2000, nullable=False)
|
|
|
|
# 可用模型缓存 (JSON)
|
|
available_models: Mapped[Optional[dict]] = mapped_column(JSONType, nullable=True)
|
|
|
|
# 测试结果
|
|
last_test_at: Mapped[Optional[datetime]] = mapped_column(
|
|
DateTime(timezone=True),
|
|
nullable=True,
|
|
)
|
|
last_test_result: Mapped[Optional[dict]] = mapped_column(JSONType, nullable=True)
|
|
|
|
# 配置状态
|
|
is_configured: Mapped[bool] = mapped_column(default=False, nullable=False)
|
|
|
|
# 关联
|
|
tenant: Mapped["Tenant"] = relationship("Tenant", back_populates="ai_config")
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<AIConfig(tenant_id={self.tenant_id}, provider={self.provider})>"
|