from sqlalchemy import Column, Integer, String, Text, DateTime, func
from database import Base

class AiModel(Base):
    __tablename__ = "ai_models"
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(255), nullable=False, comment="模型名称")
    provider = Column(String(100), default="", comment="供应厂商")
    api_format = Column(String(20), default="openai", comment="anthropic/openai")
    api_endpoint = Column(String(500), default="", comment="API接口地址")
    api_key_encrypted = Column(Text, default="", comment="加密后的API Key")
    status = Column(String(20), default="enabled", comment="enabled/disabled")
    created_at = Column(DateTime, server_default=func.now())
    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())