49 lines
1.6 KiB
C#
49 lines
1.6 KiB
C#
using BotSharp.Abstraction.Agents.Enums;
|
|
using BotSharp.Abstraction.Agents.Settings;
|
|
|
|
namespace BotSharp.Plugin.SqlDriver.Hooks;
|
|
|
|
public class SqlDriverAgentHook : AgentHookBase, IAgentHook
|
|
{
|
|
private const string SQL_TABLE_DEFINITION_FN = "sql_table_definition";
|
|
private const string VERIFY_DICTIONARY_TERM_FN = "verify_dictionary_term";
|
|
private const string SQL_SELECT_FN = "sql_select";
|
|
|
|
public override string SelfId => BuiltInAgentId.Planner;
|
|
|
|
public SqlDriverAgentHook(IServiceProvider services, AgentSettings settings)
|
|
: base(services, settings)
|
|
{
|
|
}
|
|
|
|
public override void OnAgentLoaded(Agent agent)
|
|
{
|
|
var dbType = SqlDriverHelper.GetDatabaseType(_services);
|
|
agent.TemplateDict["db_type"] = dbType;
|
|
|
|
var utilityLoads = new List<AgentUtility>
|
|
{
|
|
new AgentUtility
|
|
{
|
|
Name = UtilityName.SqlTableDefinition,
|
|
Functions = [new(SQL_TABLE_DEFINITION_FN)],
|
|
Templates = [new($"{SQL_TABLE_DEFINITION_FN}.fn")]
|
|
},
|
|
new AgentUtility
|
|
{
|
|
Name = UtilityName.SqlDictionaryLookup,
|
|
Functions = [new(VERIFY_DICTIONARY_TERM_FN)],
|
|
Templates = [new($"{VERIFY_DICTIONARY_TERM_FN}.fn")]
|
|
},
|
|
new AgentUtility
|
|
{
|
|
Name = UtilityName.SqlExecutor,
|
|
Functions = new List<UtilityFunction> { new(SQL_SELECT_FN), new(SQL_TABLE_DEFINITION_FN) },
|
|
Templates = new List<UtilityTemplate> { new($"sql_executor.fn") }
|
|
}
|
|
};
|
|
|
|
base.OnAgentLoaded(agent);
|
|
}
|
|
}
|