48 lines
1.7 KiB
C#
48 lines
1.7 KiB
C#
using BotSharp.Abstraction.Functions;
|
|
using BotSharp.Abstraction.Functions.Models;
|
|
using BotSharp.Abstraction.Models;
|
|
using BotSharp.Abstraction.Routing;
|
|
using BotSharp.Abstraction.Routing.Models;
|
|
using BotSharp.Abstraction.Routing.Settings;
|
|
|
|
namespace BotSharp.Core.Routing.Handlers;
|
|
|
|
public class RouteToAgentRoutingHandler : RoutingHandlerBase, IRoutingHandler
|
|
{
|
|
public string Name => "route_to_agent";
|
|
|
|
public string Description => "Route request to appropriate agent.";
|
|
|
|
public List<NameDesc> Parameters => new List<NameDesc>
|
|
{
|
|
new NameDesc("args", "useful parameters of next action agent")
|
|
};
|
|
|
|
public bool IsReasoning => false;
|
|
|
|
public RouteToAgentRoutingHandler(IServiceProvider services, ILogger<RouteToAgentRoutingHandler> logger, RoutingSettings settings)
|
|
: base(services, logger, settings)
|
|
{
|
|
}
|
|
|
|
public async Task<RoleDialogModel> Handle(IRoutingService routing, FunctionCallFromLlm inst)
|
|
{
|
|
var function = _services.GetServices<IFunctionCallback>().FirstOrDefault(x => x.Name == inst.Function);
|
|
var message = new RoleDialogModel(AgentRole.Function, inst.Question)
|
|
{
|
|
FunctionName = inst.Function,
|
|
FunctionArgs = JsonSerializer.Serialize(inst),
|
|
CurrentAgentId = routing.Dialogs.Last().CurrentAgentId
|
|
};
|
|
|
|
var ret = await function.Execute(message);
|
|
|
|
var context = _services.GetRequiredService<RoutingContext>();
|
|
var result = await routing.InvokeAgent(context.CurrentAgentId);
|
|
// Keep last message data for debug
|
|
result.ExecutionData = result.ExecutionData ?? message.ExecutionData;
|
|
result.FunctionName = result.FunctionName ?? message.FunctionName;
|
|
return result;
|
|
}
|
|
}
|