69 lines
2.2 KiB
C#
69 lines
2.2 KiB
C#
|
using BotSharp.Abstraction.Agents;
|
||
|
|
using BotSharp.Abstraction.Agents.Enums;
|
||
|
|
using BotSharp.Abstraction.Agents.Models;
|
||
|
|
using BotSharp.Abstraction.Agents.Settings;
|
||
|
|
using BotSharp.Abstraction.Conversations;
|
||
|
|
using BotSharp.Abstraction.Functions.Models;
|
||
|
|
using BotSharp.Core.Mcp;
|
||
|
|
using BotSharp.Core.MCP;
|
||
|
|
using Microsoft.Extensions.DependencyInjection;
|
||
|
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Linq;
|
||
|
|
|
||
|
|
namespace BotSharp.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);
|
||
|
|
|
||
|
|
foreach (var fn in functions)
|
||
|
|
{
|
||
|
|
if (!agent.SecondaryFunctions.Any(x => x.Name.Equals(fn.Name, StringComparison.OrdinalIgnoreCase)))
|
||
|
|
{
|
||
|
|
agent.SecondaryFunctions.Add(fn);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private IEnumerable<FunctionDef> GetMCPContent(Agent agent)
|
||
|
|
{
|
||
|
|
List<FunctionDef> functionDefs = new List<FunctionDef>();
|
||
|
|
var mcpClientManager = _services.GetRequiredService<MCPClientManager>();
|
||
|
|
var mcps = agent.McpTools;
|
||
|
|
foreach (var item in mcps)
|
||
|
|
{
|
||
|
|
var ua = mcpClientManager.Factory.GetClientAsync(item.ServerId).Result;
|
||
|
|
if (ua != null)
|
||
|
|
{
|
||
|
|
var tools = ua.ListToolsAsync().Result;
|
||
|
|
var funcnames = item.Functions.Select(x => x.Name).ToList();
|
||
|
|
foreach (var tool in tools.Tools.Where(x => funcnames.Contains(x.Name, StringComparer.OrdinalIgnoreCase)))
|
||
|
|
{
|
||
|
|
var funDef = AIFunctionUtilities.MapToFunctionDef(tool);
|
||
|
|
functionDefs.Add(funDef);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return functionDefs;
|
||
|
|
}
|
||
|
|
}
|