Skip to content

Commit b9a0c96

Browse files
refactor: rename _update_headers_for_request to _prepare_request_headers
This better reflects that the method prepares headers for outgoing HTTP requests, not just updating them with context. The method adds both session ID and protocol version headers as needed.
1 parent 6832394 commit b9a0c96

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

src/mcp/client/streamable_http.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -105,9 +105,7 @@ def __init__(
105105
**self.headers,
106106
}
107107

108-
def _update_headers_with_session(
109-
self, base_headers: dict[str, str]
110-
) -> dict[str, str]:
108+
def _prepare_request_headers(self, base_headers: dict[str, str]) -> dict[str, str]:
111109
"""Update headers with session ID and protocol version if available."""
112110
headers = base_headers.copy()
113111
if self.session_id:
@@ -197,7 +195,7 @@ async def handle_get_stream(
197195
if not self.session_id:
198196
return
199197

200-
headers = self._update_headers_with_session(self.request_headers)
198+
headers = self._prepare_request_headers(self.request_headers)
201199

202200
async with aconnect_sse(
203201
client,
@@ -217,7 +215,7 @@ async def handle_get_stream(
217215

218216
async def _handle_resumption_request(self, ctx: RequestContext) -> None:
219217
"""Handle a resumption request using GET with SSE."""
220-
headers = self._update_headers_with_session(ctx.headers)
218+
headers = self._prepare_request_headers(ctx.headers)
221219
if ctx.metadata and ctx.metadata.resumption_token:
222220
headers[LAST_EVENT_ID] = ctx.metadata.resumption_token
223221
else:
@@ -250,7 +248,7 @@ async def _handle_resumption_request(self, ctx: RequestContext) -> None:
250248

251249
async def _handle_post_request(self, ctx: RequestContext) -> None:
252250
"""Handle a POST request with response processing."""
253-
headers = self._update_headers_with_session(ctx.headers)
251+
headers = self._prepare_request_headers(ctx.headers)
254252
message = ctx.session_message.message
255253
is_initialization = self._is_initialization_request(message)
256254

@@ -426,7 +424,7 @@ async def terminate_session(self, client: httpx.AsyncClient) -> None:
426424
return
427425

428426
try:
429-
headers = self._update_headers_with_session(self.request_headers)
427+
headers = self._prepare_request_headers(self.request_headers)
430428
response = await client.delete(self.url, headers=headers)
431429

432430
if response.status_code == 405:

0 commit comments

Comments
 (0)