MCPGateway/src/ManagedCode.MCPGateway/Configuration/McpGatewayOptions.cs at main · managedcode/MCPGateway · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
using Microsoft.Extensions.AI;
using ModelContextProtocol.Client;

namespace ManagedCode.MCPGateway;

public sealed class McpGatewayOptions
{
    private readonly McpGatewayRegistrationCollection _sourceRegistrations = new();

    public McpGatewaySearchStrategy SearchStrategy { get; set; } = McpGatewaySearchStrategy.Auto;

    public McpGatewaySearchQueryNormalization SearchQueryNormalization { get; set; } =
        McpGatewaySearchQueryNormalization.TranslateToEnglishWhenAvailable;

    public int DefaultSearchLimit { get; set; } = 5;

    public int MaxSearchResults { get; set; } = 15;

    public int MaxDescriptorLength { get; set; } = 4096;

    internal IReadOnlyList<McpGatewayToolSourceRegistration> SourceRegistrations => _sourceRegistrations.Snapshot();

    public McpGatewayOptions AddTool(string sourceId, AITool tool, string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddTool(sourceId, tool, displayName));

    public McpGatewayOptions AddTool(AITool tool, string sourceId = McpGatewayDefaults.DefaultSourceId, string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddTool(tool, sourceId, displayName));

    public McpGatewayOptions AddTools(string sourceId, IEnumerable<AITool> tools, string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddTools(sourceId, tools, displayName));

    public McpGatewayOptions AddTools(IEnumerable<AITool> tools, string sourceId = McpGatewayDefaults.DefaultSourceId, string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddTools(tools, sourceId, displayName));

    public McpGatewayOptions AddHttpServer(
        string sourceId,
        Uri endpoint,
        IReadOnlyDictionary<string, string>? headers = null,
        string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddHttpServer(sourceId, endpoint, headers, displayName));

    public McpGatewayOptions AddStdioServer(
        string sourceId,
        string command,
        IReadOnlyList<string>? arguments = null,
        string? workingDirectory = null,
        IReadOnlyDictionary<string, string?>? environmentVariables = null,
        string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddStdioServer(sourceId, command, arguments, workingDirectory, environmentVariables, displayName));

    public McpGatewayOptions AddMcpClient(
        string sourceId,
        McpClient client,
        bool disposeClient = false,
        string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddMcpClient(sourceId, client, disposeClient, displayName));

    public McpGatewayOptions AddMcpClientFactory(
        string sourceId,
        Func<CancellationToken, ValueTask<McpClient>> clientFactory,
        bool disposeClient = true,
        string? displayName = null)
        => ConfigureRegistrations(registrations => registrations.AddMcpClientFactory(sourceId, clientFactory, disposeClient, displayName));

    private McpGatewayOptions ConfigureRegistrations(Action<McpGatewayRegistrationCollection> configure)
    {
        configure(_sourceRegistrations);
        return this;
    }
}