1.0.0初始化源代码

backend/app/models/ai_model.py

@@ -0,0 +1,43 @@
+"""
+AI 模型配置模型
+"""
+from sqlalchemy import Column, Integer, String, DateTime, Boolean, Text, Float
+from sqlalchemy.sql import func
+
+from .base import Base
+
+
+class AIModelConfig(Base):
+    """AI 模型配置表"""
+    __tablename__ = "ai_model_configs"
+    
+    id = Column(Integer, primary_key=True, autoincrement=True, comment="配置 ID")
+    
+    # 模型基本信息
+    model_name = Column(String(100), unique=True, nullable=False, index=True, comment="模型名称（如 gpt-3.5-turbo）")
+    provider = Column(String(50), nullable=False, comment="提供商（openai/zhipu/baidu/aliyun）")
+    display_name = Column(String(200), comment="显示名称")
+    
+    # API 配置
+    api_url = Column(String(500), nullable=False, comment="API 地址")
+    api_key = Column(String(500), nullable=False, comment="API Key（加密存储）")
+    api_version = Column(String(50), comment="API 版本")
+    
+    # 模型参数
+    temperature = Column(Float, default=0.7, comment="温度（0-1）")
+    max_tokens = Column(Integer, default=1000, comment="最大 Token 数")
+    top_p = Column(Float, default=1.0, comment="Top P 参数")
+    
+    # 状态控制
+    is_default = Column(Boolean, default=False, comment="是否为默认模型")
+    is_active = Column(Boolean, default=True, comment="是否启用")
+    
+    # 描述信息
+    description = Column(Text, comment="模型描述")
+    
+    # 时间戳
+    created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
+    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")
+    
+    def __repr__(self):
+        return f"<AIModelConfig(id={self.id}, name='{self.model_name}', provider='{self.provider}')>"