Spaces:
Running
Running
Merge pull request #357 from jlowin/middleware
Browse files
src/fastmcp/server/server.py
CHANGED
|
@@ -728,7 +728,7 @@ class FastMCP(Generic[LifespanResultT]):
|
|
| 728 |
# timeout to make it possible to close immediately. see
|
| 729 |
# https://github.com/jlowin/fastmcp/issues/296
|
| 730 |
uvicorn_config.setdefault("timeout_graceful_shutdown", 0)
|
| 731 |
-
app =
|
| 732 |
|
| 733 |
config = uvicorn.Config(
|
| 734 |
app,
|
|
@@ -837,6 +837,8 @@ class FastMCP(Generic[LifespanResultT]):
|
|
| 837 |
# mount these routes last, so they have the lowest route matching precedence
|
| 838 |
routes.extend(self._custom_starlette_routes)
|
| 839 |
|
|
|
|
|
|
|
| 840 |
# Create Starlette app with routes and middleware
|
| 841 |
return Starlette(
|
| 842 |
debug=self.settings.debug, routes=routes, middleware=middleware
|
|
|
|
| 728 |
# timeout to make it possible to close immediately. see
|
| 729 |
# https://github.com/jlowin/fastmcp/issues/296
|
| 730 |
uvicorn_config.setdefault("timeout_graceful_shutdown", 0)
|
| 731 |
+
app = self.sse_app()
|
| 732 |
|
| 733 |
config = uvicorn.Config(
|
| 734 |
app,
|
|
|
|
| 837 |
# mount these routes last, so they have the lowest route matching precedence
|
| 838 |
routes.extend(self._custom_starlette_routes)
|
| 839 |
|
| 840 |
+
middleware.append(Middleware(RequestContextMiddleware))
|
| 841 |
+
|
| 842 |
# Create Starlette app with routes and middleware
|
| 843 |
return Starlette(
|
| 844 |
debug=self.settings.debug, routes=routes, middleware=middleware
|