64 lines
2 KiB
C#
64 lines
2 KiB
C#
using BotSharp.Core.MCP.Helpers;
|
|
using BotSharp.Core.MCP.Managers;
|
|
using BotSharp.Core.MCP.Settings;
|
|
using ModelContextProtocol.Client;
|
|
|
|
namespace BotSharp.Core.MCP.Hooks;
|
|
|
|
public class McpToolAgentHook : AgentHookBase
|
|
{
|
|
public override string SelfId => string.Empty;
|
|
|
|
public McpToolAgentHook(IServiceProvider services, AgentSettings settings)
|
|
: base(services, settings)
|
|
{
|
|
}
|
|
|
|
public override void OnAgentMcpToolLoaded(Agent agent)
|
|
{
|
|
if (agent.Type == AgentType.Routing)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var conv = _services.GetRequiredService<IConversationService>();
|
|
var isConvMode = conv.IsConversationMode();
|
|
if (!isConvMode) return;
|
|
|
|
agent.SecondaryFunctions ??= [];
|
|
|
|
var functions = GetMcpContent(agent).Result;
|
|
agent.SecondaryFunctions = agent.SecondaryFunctions.Concat(functions).DistinctBy(x => x.Name, StringComparer.OrdinalIgnoreCase).ToList();
|
|
}
|
|
|
|
private async Task<IEnumerable<FunctionDef>> GetMcpContent(Agent agent)
|
|
{
|
|
var functionDefs = new List<FunctionDef>();
|
|
|
|
var settings = _services.GetRequiredService<McpSettings>();
|
|
if (settings?.Enabled != true)
|
|
{
|
|
return functionDefs;
|
|
}
|
|
|
|
var mcpClientManager = _services.GetRequiredService<McpClientManager>();
|
|
var mcps = agent.McpTools.Where(x => !x.Disabled);
|
|
foreach (var item in mcps)
|
|
{
|
|
var mcpClient = await mcpClientManager.GetMcpClientAsync(item.ServerId);
|
|
if (mcpClient != null)
|
|
{
|
|
var tools = await mcpClient.ListToolsAsync();
|
|
var toolnames = item.Functions.Select(x => x.Name).ToList();
|
|
foreach (var tool in tools.Where(x => toolnames.Contains(x.Name, StringComparer.OrdinalIgnoreCase)))
|
|
{
|
|
var funDef = AiFunctionHelper.MapToFunctionDef(tool);
|
|
functionDefs.Add(funDef);
|
|
}
|
|
}
|
|
}
|
|
|
|
return functionDefs;
|
|
}
|
|
}
|