Mark samples as AOT compatible (#184)

* Revert changes to remove APIs

* Apply suggestions from code review

Co-authored-by: Stephen Toub <[email protected]>

---------

Co-authored-by: Stephen Toub <[email protected]>

Author: agocke

samples/AspNetCoreSseServer/AspNetCoreSseServer.csproj

@@ -4,6 +4,7 @@
     <TargetFramework>net9.0</TargetFramework>
     <Nullable>enable</Nullable>
     <ImplicitUsings>enable</ImplicitUsings>
+    <PublishAot>true</PublishAot>
   </PropertyGroup>
 
   <ItemGroup>

samples/AspNetCoreSseServer/Program.cs

@@ -1,5 +1,10 @@
+using TestServerWithHosting.Tools;
+
 var builder = WebApplication.CreateBuilder(args);
-builder.Services.AddMcpServer().WithToolsFromAssembly();
+builder.Services.AddMcpServer()
+    .WithTools<EchoTool>()
+    .WithTools<SampleLlmTool>();
+
 var app = builder.Build();
 
 app.MapMcp();

samples/AspNetCoreSseServer/Tools/EchoTool.cs

@@ -4,7 +4,7 @@
 namespace TestServerWithHosting.Tools;
 
 [McpServerToolType]
-public static class EchoTool
+public sealed class EchoTool
 {
     [McpServerTool, Description("Echoes the input back to the client.")]
     public static string Echo(string message)

samples/AspNetCoreSseServer/Tools/SampleLlmTool.cs

@@ -8,7 +8,7 @@ namespace TestServerWithHosting.Tools;
 /// This tool uses dependency injection and async method
 /// </summary>
 [McpServerToolType]
-public static class SampleLlmTool
+public sealed class SampleLlmTool
 {
     [McpServerTool(Name = "sampleLLM"), Description("Samples from an LLM using MCP's sampling feature")]
     public static async Task<string> SampleLLM(

samples/ChatWithTools/ChatWithTools.csproj

@@ -5,6 +5,10 @@
     <TargetFramework>net8.0</TargetFramework>
     <ImplicitUsings>enable</ImplicitUsings>
     <Nullable>enable</Nullable>
+    <!--
+    Anthropic SDK isn't AOT compatible yet
+    <PublishAot>true</PublishAot>
+    -->
   </PropertyGroup>
 
   <ItemGroup>

samples/QuickstartClient/QuickstartClient.csproj

@@ -6,6 +6,10 @@
         <ImplicitUsings>enable</ImplicitUsings>
         <Nullable>enable</Nullable>
         <UserSecretsId>a4e20a70-5009-4b81-b5b6-780b6d43e78e</UserSecretsId>
+        <!--
+        Anthropic SDK isn't AOT compatible yet
+        <PublishAot>true</PublishAot>
+        -->
     </PropertyGroup>
 
     <ItemGroup>

samples/QuickstartWeatherServer/Program.cs

@@ -1,13 +1,14 @@
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Logging;
+using QuickstartWeatherServer.Tools;
 using System.Net.Http.Headers;
 
 var builder = Host.CreateApplicationBuilder(args);
 
 builder.Services.AddMcpServer()
     .WithStdioServerTransport()
-    .WithToolsFromAssembly();
+    .WithTools<WeatherTools>();
 
 builder.Logging.AddConsole(options =>
 {

samples/QuickstartWeatherServer/QuickstartWeatherServer.csproj

@@ -5,6 +5,7 @@
         <TargetFramework>net8.0</TargetFramework>
         <ImplicitUsings>enable</ImplicitUsings>
         <Nullable>enable</Nullable>
+        <PublishAot>true</PublishAot>
     </PropertyGroup>
 
     <ItemGroup>

samples/QuickstartWeatherServer/Tools/HttpClientExt.cs

@@ -0,0 +1,13 @@
+using System.Text.Json;
+
+namespace ModelContextProtocol;
+
+internal static class HttpClientExt
+{
+    public static async Task<JsonDocument> ReadJsonDocumentAsync(this HttpClient client, string requestUri)
+    {
+        using var response = await client.GetAsync(requestUri);
+        response.EnsureSuccessStatusCode();
+        return await JsonDocument.ParseAsync(await response.Content.ReadAsStreamAsync());
+    }
+}

samples/QuickstartWeatherServer/Tools/WeatherTools.cs

@@ -1,3 +1,4 @@
+using ModelContextProtocol;
 using ModelContextProtocol.Server;
 using System.ComponentModel;
 using System.Net.Http.Json;
@@ -6,14 +7,15 @@
 namespace QuickstartWeatherServer.Tools;
 
 [McpServerToolType]
-public static class WeatherTools
+public sealed class WeatherTools
 {
     [McpServerTool, Description("Get weather alerts for a US state.")]
     public static async Task<string> GetAlerts(
         HttpClient client,
         [Description("The US state to get alerts for.")] string state)
     {
-        var jsonElement = await client.GetFromJsonAsync<JsonElement>($"/alerts/active/area/{state}");
+        using var jsonDocument = await client.ReadJsonDocumentAsync($"/alerts/active/area/{state}");
+        var jsonElement = jsonDocument.RootElement;
         var alerts = jsonElement.GetProperty("features").EnumerateArray();
 
         if (!alerts.Any())
@@ -40,7 +42,8 @@ public static async Task<string> GetForecast(
         [Description("Latitude of the location.")] double latitude,
         [Description("Longitude of the location.")] double longitude)
     {
-        var jsonElement = await client.GetFromJsonAsync<JsonElement>($"/points/{latitude},{longitude}");
+        using var jsonDocument = await client.ReadJsonDocumentAsync($"/points/{latitude},{longitude}");
+        var jsonElement = jsonDocument.RootElement;
         var periods = jsonElement.GetProperty("properties").GetProperty("periods").EnumerateArray();
 
         return string.Join("\n---\n", periods.Select(period => $"""