Jeremiah Lowin commited on
Commit
407df75
·
unverified ·
2 Parent(s): 9a8ac2fafc5b27

Merge pull request #357 from jlowin/middleware

Browse files
Files changed (1) hide show
  1. src/fastmcp/server/server.py +3 -1
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 = RequestContextMiddleware(self.sse_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