OpenWebUI/mcpo
1
0
Fork 0
mirror of https://github.com/open-webui/mcpo synced 2025-06-26 18:26:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added ability to connect to SSE MCP server

Browse Source 
- use format below to connect to SSE MCP server
    uv run mcpo -- sse http://localhost:8001/sse

- use format below for config.json to connect to SSE MCP server
    {
      "mcpServers": {
        "mcp_sse": {
          "url": "http://127.0.0.1:8001/sse"
        },
        "mcp_stdio": {
          "command": "python",
          "args": [
            "mcp_stdio.py"
          ]
        }
      }
    }
This commit is contained in:
Azam Janah 2025-04-12 08:15:45 +08:00
parent 96047bc297
commit 6fbc97e506
1 changed files with 31 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/mcpo/main.py
View File

@ -11,7 +11,7 @@ from starlette.routing import Mount
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client
from mcp.client.sse import sse_client
from mcpo.utils.main import get_model_fields, get_tool_handler
from mcpo.utils.auth import get_verify_api_key
@ -69,9 +69,12 @@ async def lifespan(app: FastAPI):
args = getattr(app.state, "args", [])
env = getattr(app.state, "env", {})
mcptype = "sse" if command == "sse" else "stdio"
args = args if isinstance(args, list) else [args]
api_dependency = getattr(app.state, "api_dependency", None)
if not command:
if (mcptype == "stdio" and not command) or (mcptype == "sse" and not args[0]):
async with AsyncExitStack() as stack:
for route in app.routes:
if isinstance(route, Mount) and isinstance(route.app, FastAPI):
@ -81,17 +84,24 @@ async def lifespan(app: FastAPI):
yield
else:
server_params = StdioServerParameters(
command=command,
args=args,
env={**env},
)
if mcptype == "stdio":
server_params = StdioServerParameters(
command=command,
args=args,
env={**env},
)
async with stdio_client(server_params) as (reader, writer):
async with ClientSession(reader, writer) as session:
app.state.session = session
await create_dynamic_endpoints(app, api_dependency=api_dependency)
yield
async with stdio_client(server_params) as (reader, writer):
async with ClientSession(reader, writer) as session:
app.state.session = session
await create_dynamic_endpoints(app, api_dependency=api_dependency)
yield
if mcptype == "sse":
async with sse_client(url=args[0]) as (reader, writer):
async with ClientSession(reader, writer) as session:
app.state.session = session
await create_dynamic_endpoints(app, api_dependency=api_dependency)
yield
async def run(
@ -166,9 +176,15 @@ async def run(
allow_headers=["*"],
)
sub_app.state.command = server_cfg["command"]
sub_app.state.args = server_cfg.get("args", [])
sub_app.state.env = {**os.environ, **server_cfg.get("env", {})}
if server_cfg.get("command"):
# stdio
sub_app.state.command = server_cfg["command"]
sub_app.state.args = server_cfg.get("args", [])
sub_app.state.env = {**os.environ, **server_cfg.get("env", {})}
if server_cfg.get("url"):
# SSE
sub_app.state.command = "sse"
sub_app.state.args = server_cfg["url"]
sub_app.state.api_dependency = api_dependency