BotSharp/src/Plugins/BotSharp.Plugin.Planner/SqlGeneration/Functions/SqlReviewFn.cs

using BotSharp.Plugin.Planner.SqlGeneration.Models;

namespace BotSharp.Plugin.Planner.SqlGeneration.Functions;

public class SqlReviewFn : IFunctionCallback
{
    public string Name => "sql_review";
    public string Indication => "Currently reviewing SQL statement";

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

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

    public async Task<bool> Execute(RoleDialogModel message)
    {
        var args = JsonSerializer.Deserialize<SqlReviewArgs>(message.FunctionArgs);

        if (!message.Content.StartsWith("```sql"))
        {
            message.Content = $"```sql\r\n{args.SqlStatement}\r\n```";
        }

        if (args != null && !args.IsSqlTemplate && args.ContainsSqlStatements)
        {
            await HookEmitter.Emit<IPlanningHook>(_services, async hook =>
                await hook.OnSourceCodeGenerated(nameof(SqlGenerationPlanner), message, "sql"),
                message.CurrentAgentId
            );
        }
        return true;
    }
}