45 lines
1.4 KiB
C#
45 lines
1.4 KiB
C#
using BotSharp.Abstraction.Functions;
|
|
using BotSharp.Abstraction.Repositories.Filters;
|
|
using BotSharp.Abstraction.Routing.Models;
|
|
using BotSharp.Abstraction.Routing;
|
|
|
|
namespace BotSharp.Core.Routing.Functions;
|
|
|
|
public class FallbackToRouterFn : IFunctionCallback
|
|
{
|
|
public string Name => "fallback_to_router";
|
|
private readonly IServiceProvider _services;
|
|
public FallbackToRouterFn(IServiceProvider services)
|
|
{
|
|
_services = services;
|
|
}
|
|
|
|
public async Task<bool> Execute(RoleDialogModel message)
|
|
{
|
|
var args = JsonSerializer.Deserialize<RoutingArgs>(message.FunctionArgs);
|
|
var agentService = _services.GetRequiredService<IAgentService>();
|
|
var agents = await agentService.GetAgents(new AgentFilter
|
|
{
|
|
AgentName = args.AgentName
|
|
});
|
|
var targetAgent = agents.Items.FirstOrDefault();
|
|
if (targetAgent == null)
|
|
{
|
|
message.Content = $"Can't find routing agent {args.AgentName}";
|
|
return false;
|
|
}
|
|
|
|
var routing = _services.GetRequiredService<RoutingContext>();
|
|
routing.Replace(targetAgent.Id);
|
|
|
|
var router = _services.GetRequiredService<IRoutingService>();
|
|
message.CurrentAgentId = targetAgent.Id;
|
|
var response = await router.InstructLoop(message);
|
|
|
|
message.Content = response.Content;
|
|
message.StopCompletion = true;
|
|
|
|
return true;
|
|
}
|
|
}
|