diff --git a/src/ModelContextProtocol.Core/ModelContextProtocol.Core.csproj b/src/ModelContextProtocol.Core/ModelContextProtocol.Core.csproj
index f3eec2350..6c06d2003 100644
--- a/src/ModelContextProtocol.Core/ModelContextProtocol.Core.csproj
+++ b/src/ModelContextProtocol.Core/ModelContextProtocol.Core.csproj
@@ -63,6 +63,7 @@
+
diff --git a/src/ModelContextProtocol.Core/Server/AIFunctionMcpServerTool.cs b/src/ModelContextProtocol.Core/Server/AIFunctionMcpServerTool.cs
index 6d4579f66..d494d711c 100644
--- a/src/ModelContextProtocol.Core/Server/AIFunctionMcpServerTool.cs
+++ b/src/ModelContextProtocol.Core/Server/AIFunctionMcpServerTool.cs
@@ -1,11 +1,18 @@
using Microsoft.Extensions.AI;
+using Json.Schema;
using Microsoft.Extensions.DependencyInjection;
using ModelContextProtocol.Protocol;
+using System.Collections;
+using System.Collections.Concurrent;
using System.ComponentModel;
+using System.Globalization;
using System.Diagnostics;
using System.Reflection;
+using System.Runtime.CompilerServices;
using System.Text.Json;
using System.Text.Json.Nodes;
+using System.Text.Json.Serialization;
+using System.Text.Json.Serialization.Metadata;
using System.Text.RegularExpressions;
namespace ModelContextProtocol.Server;
@@ -15,6 +22,10 @@ internal sealed partial class AIFunctionMcpServerTool : McpServerTool
{
private readonly IReadOnlyList