BotSharp/src/Infrastructure/BotSharp.Core/Agents/Services/AgentRouter.cs

using BotSharp.Abstraction.Agents.Models;

namespace BotSharp.Core.Agents.Services;

public class AgentRouter : IAgentRouting
{
    private readonly IServiceProvider _services;
    private readonly ILogger _logger;
    private readonly AgentSettings _settings;

    public AgentRouter(IServiceProvider services, 
        ILogger<AgentRouter> logger,
        AgentSettings settings)
    {
        _services = services;
        _logger = logger;
        _settings = settings;
    }

    public async Task<Agent> LoadCurrentAgent()
    {
        // Load current agent from state
        var state = _services.GetRequiredService<IConversationStateService>();
        var currentAgentId = state.GetState("agentId");
        if (string.IsNullOrEmpty(currentAgentId))
        {
            currentAgentId = _settings.RouterId;
        }
        var agentService = _services.GetRequiredService<IAgentService>();
        var agent = await agentService.LoadAgent(currentAgentId);

        // Set agent and trigger state changed
        state.SetState("agentId", currentAgentId);

        return agent;
    }
}