From 4b6dd0f91620e326ddcf46864b46aa1549ac7c7e Mon Sep 17 00:00:00 2001
From: ClosetheWorld <develop@closetheworld.net>
Date: Sat, 12 Apr 2025 22:43:35 +0900
Subject: [PATCH] Add AdditionalHeader handling to ReceiveMessageAsync

---
 .../Protocol/Transport/SseClientSessionTransport.cs       | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/ModelContextProtocol/Protocol/Transport/SseClientSessionTransport.cs b/src/ModelContextProtocol/Protocol/Transport/SseClientSessionTransport.cs
index 5348995e7..f0395e8fd 100644
--- a/src/ModelContextProtocol/Protocol/Transport/SseClientSessionTransport.cs
+++ b/src/ModelContextProtocol/Protocol/Transport/SseClientSessionTransport.cs
@@ -191,6 +191,14 @@ private async Task ReceiveMessagesAsync(CancellationToken cancellationToken)
                 using var request = new HttpRequestMessage(HttpMethod.Get, _sseEndpoint);
                 request.Headers.Accept.Add(new MediaTypeWithQualityHeaderValue("text/event-stream"));
 
+                if (_options.AdditionalHeaders != null)
+                {
+                    foreach (var header in _options.AdditionalHeaders)
+                    {
+                        request.Headers.Add(header.Key, header.Value);
+                    }
+                }
+
                 using var response = await _httpClient.SendAsync(
                     request,
                     HttpCompletionOption.ResponseHeadersRead,