using BotSharp.Abstraction.Functions;
using BotSharp.Abstraction.Instructs.Models;

namespace BotSharp.Core.Instructs.Functions;

public class ExecuteTemplateFn : IFunctionCallback
{
    public string Name => "util-instruct-execute_template";

    private readonly IServiceProvider _services;
    private readonly ILogger<ExecuteTemplateFn> _logger;

    public ExecuteTemplateFn(
        IServiceProvider services,
        ILogger<ExecuteTemplateFn> logger)
    {
        _services = services;
        _logger = logger;
    }

    public async Task<bool> Execute(RoleDialogModel message)
    {
        var args = JsonSerializer.Deserialize<ExecuteTemplateArgs>(message.FunctionArgs);
        if (string.IsNullOrEmpty(args.TemplateName))
        {
            message.Content = $"Empty template name.";
            return false;
        }

        var agentService = _services.GetRequiredService<IAgentService>();
        var agent = await agentService.GetAgent(message.CurrentAgentId);
        var template = agent.Templates.FirstOrDefault(x => x.Name.IsEqualTo(args.TemplateName));

        if (template == null)
        {
            message.Content = $"Cannot find template ({args.TemplateName}) in agent {agent.Name}";
            return false;
        }

        var prompt = agentService.RenderedTemplate(agent, args.TemplateName);
        var response = await GetAiResponse(agent, prompt);
        message.Content = response;
        return true;
    }

    private async Task<string> GetAiResponse(Agent agent, string text)
    {
        try
        {
            var completion = CompletionProvider.GetChatCompletion(_services, provider: agent.LlmConfig?.Provider, model: agent.LlmConfig?.Model);
            var response = await completion.GetChatCompletions(new Agent()
            {
                Id = agent.Id
            },
            new List<RoleDialogModel>
            {
                new(AgentRole.User, text)
            });
            return response.Content;
        }
        catch (Exception ex)
        {
            var error = $"Error when getting agent {agent.Name} instruction response.";
            _logger.LogWarning($"{error} {ex.Message}\r\n{ex.InnerException}");
            return error;
        }

    }
}