8d4f51da50- Add missing filling of ResourceCollection in McpServerOptionsSetup - Remove AIFunctionMcpServerXx.ToString overrides; for resources, it was changing behavior undesirably, and for the others it was duplicate code. - Fix notification method used for resource list changes. - Fix type in WithResources parameter name.
83 lines
3.6 KiB
C#
83 lines
3.6 KiB
C#
using ModelContextProtocol.Server;
|
|
using Microsoft.Extensions.Options;
|
|
using ModelContextProtocol.Utils;
|
|
|
|
namespace Microsoft.Extensions.DependencyInjection;
|
|
|
|
/// <summary>
|
|
/// Configures the McpServerOptions using addition services from DI.
|
|
/// </summary>
|
|
/// <param name="serverHandlers">The server handlers configuration options.</param>
|
|
/// <param name="serverTools">Tools individually registered.</param>
|
|
/// <param name="serverPrompts">Prompts individually registered.</param>
|
|
/// <param name="serverResources">Resources individually registered.</param>
|
|
internal sealed class McpServerOptionsSetup(
|
|
IOptions<McpServerHandlers> serverHandlers,
|
|
IEnumerable<McpServerTool> serverTools,
|
|
IEnumerable<McpServerPrompt> serverPrompts,
|
|
IEnumerable<McpServerResource> serverResources) : IConfigureOptions<McpServerOptions>
|
|
{
|
|
/// <summary>
|
|
/// Configures the given McpServerOptions instance by setting server information
|
|
/// and applying custom server handlers and tools.
|
|
/// </summary>
|
|
/// <param name="options">The options instance to be configured.</param>
|
|
public void Configure(McpServerOptions options)
|
|
{
|
|
Throw.IfNull(options);
|
|
|
|
// Collect all of the provided tools into a tools collection. If the options already has
|
|
// a collection, add to it, otherwise create a new one. We want to maintain the identity
|
|
// of an existing collection in case someone has provided their own derived type, wants
|
|
// change notifications, etc.
|
|
McpServerPrimitiveCollection<McpServerTool> toolCollection = options.Capabilities?.Tools?.ToolCollection ?? [];
|
|
foreach (var tool in serverTools)
|
|
{
|
|
toolCollection.TryAdd(tool);
|
|
}
|
|
|
|
if (!toolCollection.IsEmpty)
|
|
{
|
|
options.Capabilities ??= new();
|
|
options.Capabilities.Tools ??= new();
|
|
options.Capabilities.Tools.ToolCollection = toolCollection;
|
|
}
|
|
|
|
// Collect all of the provided prompts into a prompts collection. If the options already has
|
|
// a collection, add to it, otherwise create a new one. We want to maintain the identity
|
|
// of an existing collection in case someone has provided their own derived type, wants
|
|
// change notifications, etc.
|
|
McpServerPrimitiveCollection<McpServerPrompt> promptCollection = options.Capabilities?.Prompts?.PromptCollection ?? [];
|
|
foreach (var prompt in serverPrompts)
|
|
{
|
|
promptCollection.TryAdd(prompt);
|
|
}
|
|
|
|
if (!promptCollection.IsEmpty)
|
|
{
|
|
options.Capabilities ??= new();
|
|
options.Capabilities.Prompts ??= new();
|
|
options.Capabilities.Prompts.PromptCollection = promptCollection;
|
|
}
|
|
|
|
// Collect all of the provided resources into a resources collection. If the options already has
|
|
// a collection, add to it, otherwise create a new one. We want to maintain the identity
|
|
// of an existing collection in case someone has provided their own derived type, wants
|
|
// change notifications, etc.
|
|
McpServerPrimitiveCollection<McpServerResource> resourceCollection = options.Capabilities?.Resources?.ResourceCollection ?? [];
|
|
foreach (var resource in serverResources)
|
|
{
|
|
resourceCollection.TryAdd(resource);
|
|
}
|
|
|
|
if (!resourceCollection.IsEmpty)
|
|
{
|
|
options.Capabilities ??= new();
|
|
options.Capabilities.Resources ??= new();
|
|
options.Capabilities.Resources.ResourceCollection = resourceCollection;
|
|
}
|
|
|
|
// Apply custom server handlers.
|
|
serverHandlers.Value.OverwriteWithSetHandlers(options);
|
|
}
|
|
}
|