using BotSharp.Abstraction.Templating;

namespace BotSharp.Core.Routing;

public partial class RoutingService
{
    const int MAXIMUM_RECURSION_DEPTH = 2;
    int CurrentRecursionDepth = 0;
    public async Task<RoleDialogModel> InvokeAgent(string agentId)
    {
        CurrentRecursionDepth++;
        if (CurrentRecursionDepth > MAXIMUM_RECURSION_DEPTH)
        {
            return Dialogs.Last();
        }

        var agentService = _services.GetRequiredService<IAgentService>();
        var agent = await agentService.LoadAgent(agentId);

        var chatCompletion = CompletionProvider.GetChatCompletion(_services);
        RoleDialogModel response = chatCompletion.GetChatCompletions(agent, Dialogs);

        if (response.Role == AgentRole.Function)
        {
            var fn = response;
            // execute function
            // Save states
            SaveStateByArgs(JsonSerializer.Deserialize<JsonDocument>(fn.FunctionArgs));

            var conversationService = _services.GetRequiredService<IConversationService>();
            // Call functions
            await conversationService.CallFunctions(fn);

            if (string.IsNullOrEmpty(fn.Content))
            {
                fn.Content = fn.ExecutionResult;
            }

            Dialogs.Add(fn);

            if (!fn.StopCompletion)
            {
                // Find response template
                var templateService = _services.GetRequiredService<IResponseTemplateService>();
                var quickResponse = await templateService.RenderFunctionResponse(agent.Id, fn);
                if (!string.IsNullOrEmpty(quickResponse))
                {
                    response = new RoleDialogModel(AgentRole.Assistant, quickResponse)
                    {
                        CurrentAgentId = agent.Id
                    };
                }
                else
                {
                    response = await InvokeAgent(fn.CurrentAgentId);
                }
            }
            else
            {
                response = fn;
            }
        }

        return response;
    }
}