zwt13703
17 lines
1007 B
Python
17 lines
1007 B
Python
from sqlalchemy import Boolean, 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")
|
|
supports_streaming = Column(Boolean, default=False, comment="是否支持流式传输")
|
|
enable_reasoning = Column(Boolean, default=False, comment="是否开启思考模式")
|
|
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())
|