42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
# main.py
|
||
# 将 Materials CIF MCP 与 System Tools MCP 一起挂载到 Starlette。
|
||
# 关键点：
|
||
# - 在 lifespan 中启动每个 MCP 的 session_manager.run()（参考 SDK README 的 Starlette 挂载示例与 streamable_http_app 用法 [1]）
|
||
# - 通过 Mount 指定各自的子路径（如 /system 与 /materials）
|
||
|
||
import contextlib
|
||
from starlette.applications import Starlette
|
||
from starlette.routing import Mount
|
||
|
||
from system_tools import create_system_mcp
|
||
from materialproject_mcp import create_materials_mcp
|
||
|
||
# 创建 MCP 实例
|
||
system_mcp = create_system_mcp()
|
||
materials_mcp = create_materials_mcp()
|
||
|
||
# 在 Starlette 的 lifespan 中启动 MCP 的 session manager
|
||
@contextlib.asynccontextmanager
|
||
async def lifespan(app: Starlette):
|
||
async with contextlib.AsyncExitStack() as stack:
|
||
await stack.enter_async_context(system_mcp.session_manager.run())
|
||
await stack.enter_async_context(materials_mcp.session_manager.run())
|
||
yield # 服务器运行期间
|
||
# 退出时自动清理
|
||
|
||
# 挂载两个 MCP 的 Streamable HTTP App
|
||
app = Starlette(
|
||
lifespan=lifespan,
|
||
routes=[
|
||
Mount("/system", app=system_mcp.streamable_http_app()),
|
||
Mount("/materials", app=materials_mcp.streamable_http_app()),
|
||
],
|
||
)
|
||
|
||
# 启动命令（终端执行）：
|
||
# uvicorn main:app --host 0.0.0.0 --port 8000
|
||
# 访问：
|
||
# http://localhost:8000/system
|
||
# http://localhost:8000/materials
|
||
#
|
||
# 如果需要浏览器客户端访问（CORS 暴露 Mcp-Session-Id），请参考 README 中的 CORS 配置示例 [1] |