44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
"""百工 MCP server 模型（skills-design §5）。
|
||
|
||
注册的 MCP server 元数据 + 拉取缓存的工具清单（tools_json），供启动加载与
|
||
转 ToolSpec register。
|
||
|
||
支持两种传输协议：
|
||
- streamable-http：通过 endpoint_url POST /message（标准 MCP Streamable HTTP）
|
||
- stdio：通过子进程 stdin/stdout 通信（本地 MCP 集成）
|
||
"""
|
||
|
||
from datetime import datetime
|
||
|
||
from sqlalchemy import DateTime, String, Text
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
from app.core.ids import new_mcp_server_id
|
||
from app.core.time import utc_now
|
||
from app.db.base import Base
|
||
|
||
|
||
class McpServer(Base):
|
||
"""MCP server 注册记录。"""
|
||
|
||
__tablename__ = "mcp_servers"
|
||
|
||
id: Mapped[str] = mapped_column(String(80), primary_key=True, default=new_mcp_server_id)
|
||
name: Mapped[str] = mapped_column(String(120), nullable=False, unique=True)
|
||
transport: Mapped[str] = mapped_column(String(40), nullable=False, default="streamable-http")
|
||
# streamable-http 用
|
||
endpoint_url: Mapped[str | None] = mapped_column(String(500))
|
||
auth_token: Mapped[str | None] = mapped_column(String(500))
|
||
# stdio 用
|
||
command: Mapped[str | None] = mapped_column(String(500))
|
||
args_json: Mapped[str | None] = mapped_column(Text) # JSON list
|
||
env_json: Mapped[str | None] = mapped_column(Text) # JSON dict
|
||
cwd: Mapped[str | None] = mapped_column(String(500))
|
||
# 通用
|
||
status: Mapped[str] = mapped_column(String(40), nullable=False, default="enabled", index=True)
|
||
tools_json: Mapped[str] = mapped_column(Text, nullable=False, default="[]")
|
||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||
updated_at: Mapped[datetime] = mapped_column(
|
||
DateTime(timezone=True), default=utc_now, onupdate=utc_now
|
||
)
|