PR feedback

Author: MackinnonBuck

src/ModelContextProtocol.AspNetCore/StreamableHttpHandler.cs

@@ -154,6 +154,11 @@ await WriteJsonRpcErrorAsync(context,
         {
             await using var _ = await session.AcquireReferenceAsync(cancellationToken);
             InitializeSseResponse(context);
+
+            // We should flush headers to indicate a 200 success quickly, because the initialization response
+            // will be sent in response to a different POST request. It might be a while before we send a message
+            // over this response body.
+            await context.Response.StartAsync(cancellationToken);
             await session.Transport.HandleGetRequestAsync(context.Response.Body, cancellationToken);
         }
         catch (OperationCanceledException) when (cancellationToken.IsCancellationRequested)

src/ModelContextProtocol.Core/Server/StreamableHttpServerTransport.cs

@@ -125,11 +125,6 @@ public async Task HandleGetRequestAsync(Stream sseResponseStream, CancellationTo
                 var primingItem = await _storeSseWriter.WriteEventAsync(SseItem.Prime<JsonRpcMessage>(), cancellationToken).ConfigureAwait(false);
                 await _httpSseWriter.WriteAsync(primingItem, cancellationToken).ConfigureAwait(false);
             }
-
-            // We should flush to indicate a 200 success quickly, because the initialization response
-            // will be sent in response to a different POST request. It might be a while before we send a message
-            // over this response body.
-            await sseResponseStream.FlushAsync(cancellationToken).ConfigureAwait(false);
         }
 
         // Wait for the response to be written before returning from the handler.