enable http headers in sse in the context

Author: abdullahkady

src/mcp/server/lowlevel/server.py

@@ -526,6 +526,11 @@ async def _handle_request(
             logger.debug(f"Dispatching request of type {type(req).__name__}")
 
             token = None
+            headers = {}
+            try:
+                headers = message.request.root.headers  # type: ignore
+            except Exception:
+                pass
             try:
                 # Set our global state that can be retrieved via
                 # app.get_request_context()
@@ -535,6 +540,7 @@ async def _handle_request(
                         message.request_meta,
                         session,
                         lifespan_context,
+                        headers,
                     )
                 )
                 response = await handler(req)

src/mcp/server/sse.py

@@ -160,7 +160,8 @@ async def handle_post_message(
         logger.debug(f"Received JSON: {json}")
 
         try:
-            message = types.JSONRPCMessage.model_validate(json)
+            message_with_headers = {**json, "headers": dict(request.headers)}
+            message = types.JSONRPCMessage.model_validate(message_with_headers)
             logger.debug(f"Validated client message: {message}")
         except ValidationError as err:
             logger.error(f"Failed to parse message: {err}")

src/mcp/shared/context.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Generic, TypeVar
 
 from mcp.shared.session import BaseSession
@@ -14,3 +14,4 @@ class RequestContext(Generic[SessionT, LifespanContextT]):
     meta: RequestParams.Meta | None
     session: SessionT
     lifespan_context: LifespanContextT
+    headers: dict[str, str] = field(default_factory=dict)

src/mcp/types.py

@@ -118,6 +118,7 @@ class JSONRPCRequest(Request):
     jsonrpc: Literal["2.0"]
     id: RequestId
     params: dict[str, Any] | None = None
+    headers: dict[str, str] | None = None
 
 
 class JSONRPCNotification(Notification):