ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-18 19:44:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #475 from Maxi91f/feat/middleware-on-run

Browse source 
feat: Add middleware option in server.run
This commit is contained in:
nate nowack 2025-05-15 19:38:59 -05:00 • committed by GitHub
commit f20bc7076c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/fastmcp/server/server.py
View file

@ -189,15 +189,13 @@ class FastMCP(Generic[LifespanResultT]):
"""
if transport is None:
transport = "stdio"
if transport not in ["stdio", "streamable-http", "sse"]:
if transport not in {"stdio", "streamable-http", "sse"}:
raise ValueError(f"Unknown transport: {transport}")
if transport == "stdio":
await self.run_stdio_async(**transport_kwargs)
elif transport == "streamable-http":
await self.run_http_async(transport="streamable-http", **transport_kwargs)
elif transport == "sse":
await self.run_http_async(transport="sse", **transport_kwargs)
elif transport in {"streamable-http", "sse"}:
await self.run_http_async(transport=transport, **transport_kwargs)
else:
raise ValueError(f"Unknown transport: {transport}")
@ -744,6 +742,7 @@ class FastMCP(Generic[LifespanResultT]):
log_level: str | None = None,
path: str | None = None,
uvicorn_config: dict | None = None,
middleware: list[Middleware] | None = None,
) -> None:
"""Run the server using HTTP transport.
@ -760,7 +759,7 @@ class FastMCP(Generic[LifespanResultT]):
# lifespan is required for streamable http
uvicorn_config["lifespan"] = "on"
app = self.http_app(path=path, transport=transport)
app = self.http_app(path=path, transport=transport, middleware=middleware)
config = uvicorn.Config(
app,