aiagent/backend/app/models/model_config.py

"""
模型配置模型
"""
from sqlalchemy import Column, String, DateTime, ForeignKey, Index, func
from sqlalchemy.dialects.mysql import CHAR
from sqlalchemy.orm import relationship
from app.core.database import Base
import uuid


class ModelConfig(Base):
    """模型配置表"""
    __tablename__ = "model_configs"

    id = Column(CHAR(36), primary_key=True, default=lambda: str(uuid.uuid4()), comment="配置ID")
    name = Column(String(100), nullable=False, comment="配置名称")
    provider = Column(String(50), nullable=False, comment="提供商: openai/claude/local")
    model_name = Column(String(100), nullable=False, comment="模型名称")
    api_key = Column(String(500), nullable=False, comment="API密钥（加密存储）")
    base_url = Column(String(255), comment="API地址")
    user_id = Column(CHAR(36), ForeignKey("users.id"), comment="所属用户ID")
    workspace_id = Column(CHAR(36), ForeignKey("workspaces.id"), nullable=True, comment="所属工作区ID")
    created_at = Column(DateTime, default=func.now(), comment="创建时间")
    updated_at = Column(DateTime, default=func.now(), onupdate=func.now(), comment="更新时间")

    # 关系
    user = relationship("User", backref="model_configs")

    __table_args__ = (
        Index("ix_model_configs_workspace", "workspace_id"),
    )

    def __repr__(self):
        return f"<ModelConfig(id={self.id}, name={self.name}, provider={self.provider})>"