60 lines
1.9 KiB
C#
60 lines
1.9 KiB
C#
using BotSharp.Core.MCP.Functions;
|
|
using BotSharp.Core.MCP.Hooks;
|
|
using BotSharp.Core.MCP.Managers;
|
|
using BotSharp.Core.MCP.Services;
|
|
using BotSharp.Core.MCP.Settings;
|
|
using Microsoft.Extensions.Configuration;
|
|
using ModelContextProtocol.Client;
|
|
|
|
namespace BotSharp.Core.MCP;
|
|
|
|
public static class BotSharpMcpExtensions
|
|
{
|
|
/// <summary>
|
|
/// Add mcp
|
|
/// </summary>
|
|
/// <param name="services"></param>
|
|
/// <param name="config"></param>
|
|
/// <returns></returns>
|
|
public static IServiceCollection AddBotSharpMCP(this IServiceCollection services, IConfiguration config)
|
|
{
|
|
services.AddScoped<IMcpService, McpService>();
|
|
var settings = config.GetSection("MCP").Get<McpSettings>();
|
|
services.AddScoped(provider => settings);
|
|
|
|
if (settings != null && settings.Enabled && !settings.McpServerConfigs.IsNullOrEmpty())
|
|
{
|
|
var clientManager = new McpClientManager(settings);
|
|
services.AddSingleton(clientManager);
|
|
|
|
foreach (var server in settings.McpServerConfigs)
|
|
{
|
|
RegisterFunctionCall(services, server, clientManager)
|
|
.ConfigureAwait(false)
|
|
.GetAwaiter()
|
|
.GetResult();
|
|
}
|
|
|
|
// Register hooks
|
|
services.AddScoped<IAgentHook, McpToolAgentHook>();
|
|
}
|
|
return services;
|
|
}
|
|
|
|
private static async Task RegisterFunctionCall(IServiceCollection services, McpServerConfigModel server, McpClientManager clientManager)
|
|
{
|
|
var client = await clientManager.GetMcpClientAsync(server.Id);
|
|
var tools = await client.ListToolsAsync();
|
|
|
|
foreach (var tool in tools)
|
|
{
|
|
services.AddScoped(provider => tool);
|
|
|
|
services.AddScoped<IFunctionCallback>(provider =>
|
|
{
|
|
var funcTool = new McpToolAdapter(provider, tool, clientManager);
|
|
return funcTool;
|
|
});
|
|
}
|
|
}
|
|
} |