BotSharp/src/Infrastructure/BotSharp.Core/Routing/Functions/FallbackToRouterFn.cs

using BotSharp.Abstraction.Functions;
using BotSharp.Abstraction.Routing.Models;

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 conv = _services.GetRequiredService<IConversationService>();
        var dialogs = conv.GetDialogHistory();

        var routing = _services.GetRequiredService<IRoutingService>();
        routing.Context.Replace(targetAgent.Id);
        message.CurrentAgentId = targetAgent.Id;

        var response = await routing.InstructLoop(message, dialogs, null);

        message.Content = response.Content;
        message.StopCompletion = true;

        return true;
    }
}