MCPGateway/src/ManagedCode.MCPGateway/Registration/McpGatewayChatOptionsExtensions.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
using Microsoft.Extensions.AI;
using Microsoft.Extensions.DependencyInjection;

namespace ManagedCode.MCPGateway;

public static class McpGatewayChatOptionsExtensions
{
    public static ChatOptions AddMcpGatewayTools(
        this ChatOptions options,
        McpGatewayToolSet toolSet,
        string searchToolName = McpGatewayToolSet.DefaultSearchToolName,
        string invokeToolName = McpGatewayToolSet.DefaultInvokeToolName)
    {
        ArgumentNullException.ThrowIfNull(options);
        ArgumentNullException.ThrowIfNull(toolSet);

        options.Tools = toolSet.AddTools(options.Tools ?? new List<AITool>(), searchToolName, invokeToolName);
        return options;
    }

    public static ChatOptions AddMcpGatewayTools(
        this ChatOptions options,
        IServiceProvider serviceProvider,
        string searchToolName = McpGatewayToolSet.DefaultSearchToolName,
        string invokeToolName = McpGatewayToolSet.DefaultInvokeToolName)
    {
        ArgumentNullException.ThrowIfNull(options);
        ArgumentNullException.ThrowIfNull(serviceProvider);

        return options.AddMcpGatewayTools(
            serviceProvider.GetRequiredService<McpGatewayToolSet>(),
            searchToolName,
            invokeToolName);
    }
}