modelcontextprotocol/csharp-sdk
Watch
1
0
Fork
You've already forked csharp-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Configurable messages endpoint for SSE transport (#109)

Browse source
This commit is contained in:
Eduard Stanculet 2025-03-27 03:05:49 +02:00 • committed by GitHub
commit d1217c57af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/ModelContextProtocol/Protocol/Transport/SseResponseStreamTransport.cs
View file

@ -1,4 +1,5 @@
using System.Buffers;
using System.Text;
using System.Buffers;
using System.Net.ServerSentEvents;
using System.Text.Json;
using System.Threading.Channels;
@ -11,7 +12,8 @@ namespace ModelContextProtocol.Protocol.Transport;
/// Implements the MCP SSE server transport protocol using the SSE response <see cref="Stream"/>.
/// </summary>
/// <param name="sseResponseStream">The stream to write the SSE response body to.</param>
public sealed class SseResponseStreamTransport(Stream sseResponseStream) : ITransport
/// <param name="messageEndpoint">The endpoint to send JSON-RPC messages to. Defaults to "/message".</param>
public sealed class SseResponseStreamTransport(Stream sseResponseStream, string messageEndpoint = "/message") : ITransport
{
private readonly Channel<IJsonRpcMessage> _incomingChannel = CreateSingleItemChannel<IJsonRpcMessage>();
private readonly Channel<SseItem<IJsonRpcMessage?>> _outgoingSseChannel = CreateSingleItemChannel<SseItem<IJsonRpcMessage?>>();
@ -34,7 +36,7 @@ public sealed class SseResponseStreamTransport(Stream sseResponseStream) : ITran
{
if (item.EventType == "endpoint")
{
writer.Write("/message"u8);
writer.Write(Encoding.UTF8.GetBytes(messageEndpoint));
return;
}