xiamuceer
52 lines
2.2 KiB
Python
52 lines
2.2 KiB
Python
"""MCP插件配置数据模型"""
|
||
from sqlalchemy import Column, String, Text, Boolean, Integer, DateTime, Index, JSON
|
||
from sqlalchemy.sql import func
|
||
from app.database import Base
|
||
import uuid
|
||
|
||
|
||
class MCPPlugin(Base):
|
||
"""MCP插件配置表"""
|
||
__tablename__ = "mcp_plugins"
|
||
|
||
id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
||
user_id = Column(String(50), nullable=False, index=True, comment="用户ID")
|
||
|
||
# 插件基本信息
|
||
plugin_name = Column(String(100), nullable=False, comment="插件名称（唯一标识）")
|
||
display_name = Column(String(200), nullable=False, comment="显示名称")
|
||
description = Column(Text, comment="插件描述")
|
||
plugin_type = Column(String(50), default="http", comment="插件类型：http/stdio")
|
||
|
||
# 连接配置
|
||
server_url = Column(String(500), comment="服务器URL（HTTP类型）")
|
||
command = Column(String(500), comment="启动命令（stdio类型）")
|
||
args = Column(JSON, comment="命令参数（stdio类型）")
|
||
env = Column(JSON, comment="环境变量")
|
||
headers = Column(JSON, comment="HTTP请求头")
|
||
|
||
# 插件配置
|
||
config = Column(JSON, comment="插件特定配置（JSON）")
|
||
tools = Column(JSON, comment="提供的工具列表")
|
||
|
||
# 状态管理
|
||
enabled = Column(Boolean, default=True, comment="是否启用")
|
||
status = Column(String(50), default="inactive", comment="状态：active/inactive/error")
|
||
last_error = Column(Text, comment="最后错误信息")
|
||
last_test_at = Column(DateTime, comment="最后测试时间")
|
||
|
||
# 排序和分组
|
||
category = Column(String(100), default="general", comment="分类")
|
||
sort_order = Column(Integer, default=0, comment="排序顺序")
|
||
|
||
# 时间戳
|
||
created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
|
||
updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")
|
||
|
||
__table_args__ = (
|
||
Index('idx_user_plugin', 'user_id', 'plugin_name', unique=True),
|
||
Index('idx_user_enabled', 'user_id', 'enabled'),
|
||
)
|
||
|
||
def __repr__(self):
|
||
return f"<MCPPlugin(id={self.id}, name={self.plugin_name}, enabled={self.enabled})>" |