dynamic register fluid model

Author: Jicheng Lu

src/Infrastructure/BotSharp.Abstraction/Templating/ITemplateRender.cs

@@ -3,4 +3,5 @@ namespace BotSharp.Abstraction.Templating;
 public interface ITemplateRender
 {
     string Render(string template, Dictionary<string, object> dict);
+    void Register(Type type);
 }

src/Infrastructure/BotSharp.Core/Agents/AgentPlugin.cs

@@ -1,6 +1,7 @@
 using BotSharp.Abstraction.MLTasks;
 using BotSharp.Abstraction.Plugins.Models;
 using BotSharp.Abstraction.Settings;
+using BotSharp.Abstraction.Templating;
 using BotSharp.Abstraction.Users.Enums;
 using Microsoft.Extensions.Configuration;
 
@@ -33,6 +34,8 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         services.AddScoped(provider =>
         {
             var settingService = provider.GetRequiredService<ISettingService>();
+            var render = provider.GetRequiredService<ITemplateRender>();
+            render.Register(typeof(AgentSettings));
             return settingService.Bind<AgentSettings>("Agent");
         });
     }

src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.Rendering.cs

@@ -9,13 +9,16 @@ public partial class AgentService
     public string RenderedInstruction(Agent agent)
     {
         var render = _services.GetRequiredService<ITemplateRender>();
-        // update states
         var conv = _services.GetRequiredService<IConversationService>();
+
+        // update states
         foreach (var t in conv.States.GetStates())
         {
             agent.TemplateDict[t.Key] = t.Value;
         }
-        return render.Render(agent.Instruction, agent.TemplateDict);
+
+        var res = render.Render(agent.Instruction, agent.TemplateDict);
+        return res;
     }
 
     public bool RenderFunction(Agent agent, FunctionDef def)
@@ -108,16 +111,18 @@ public bool RenderFunction(Agent agent, FunctionDef def)
 
     public string RenderedTemplate(Agent agent, string templateName)
     {
-        // render liquid template
+        var conv = _services.GetRequiredService<IConversationService>();
         var render = _services.GetRequiredService<ITemplateRender>();
+
         var template = agent.Templates.First(x => x.Name == templateName).Content;
+
         // update states
-        var conv = _services.GetRequiredService<IConversationService>();
         foreach (var t in conv.States.GetStates())
         {
             agent.TemplateDict[t.Key] = t.Value;
         }
 
+        // render liquid template
         var content = render.Render(template, agent.TemplateDict);
 
         HookEmitter.Emit<IContentGeneratingHook>(_services, async hook =>
@@ -126,4 +131,4 @@ await hook.OnRenderingTemplate(agent, templateName, content)
 
         return content;
     }
-}
+}

src/Infrastructure/BotSharp.Core/BotSharpCoreExtensions.cs

@@ -12,6 +12,8 @@
 using StackExchange.Redis;
 using BotSharp.Core.Infrastructures.Events;
 using BotSharp.Core.Roles.Services;
+using BotSharp.Abstraction.Templating;
+using BotSharp.Core.Templating;
 
 namespace BotSharp.Core;
 
@@ -24,6 +26,8 @@ public static IServiceCollection AddBotSharpCore(this IServiceCollection service
         services.AddSingleton(x => interpreterSettings);
 
         services.AddSingleton<DistributedLocker>();
+        // Register template render
+        services.AddSingleton<ITemplateRender, TemplateRender>();
 
         services.AddScoped<ISettingService, SettingService>();
         services.AddScoped<IRoleService, RoleService>();

src/Infrastructure/BotSharp.Core/Conversations/ConversationPlugin.cs

@@ -30,6 +30,8 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         services.AddScoped(provider =>
         {
             var settingService = provider.GetRequiredService<ISettingService>();
+            var render = provider.GetRequiredService<ITemplateRender>();
+            render.Register(typeof(ConversationSetting));
             return settingService.Bind<ConversationSetting>("Conversation");
         });
 
@@ -48,8 +50,6 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         // Rich content messaging
         services.AddScoped<IRichContentService, RichContentService>();
 
-        // Register template render
-        services.AddSingleton<ITemplateRender, TemplateRender>();
         services.AddScoped<IResponseTemplateService, ResponseTemplateService>();
 
         services.AddScoped<IExecutor, InstructExecutor>();

src/Infrastructure/BotSharp.Core/Templating/TemplateRender.cs

@@ -1,10 +1,10 @@
-using BotSharp.Abstraction.Agents.Models;
-using BotSharp.Abstraction.Functions.Models;
 using BotSharp.Abstraction.Models;
 using BotSharp.Abstraction.Routing.Models;
 using BotSharp.Abstraction.Templating;
 using BotSharp.Abstraction.Translation.Models;
 using Fluid;
+using System.Collections;
+using System.Reflection;
 
 namespace BotSharp.Core.Templating;
 
@@ -48,4 +48,47 @@ public string Render(string template, Dictionary<string, object> dict)
             return template;
         }
     }
+
+
+    public void Register(Type type)
+    {
+        if (type == null || IsStringType(type)) return;
+
+        if (IsListType(type))
+        {
+            if (type.IsGenericType)
+            {
+                var genericType = type.GetGenericArguments()[0];
+                Register(genericType);
+            }
+        }
+        else if (IsTrackToNextLevel(type))
+        {
+            _options.MemberAccessStrategy.Register(type);
+            var props = type.GetProperties();
+            foreach (var prop in props)
+            {
+                Register(prop.PropertyType);
+            }
+        }
+    }
+
+
+    #region Private methods
+    private static bool IsStringType(Type type)
+    {
+        return type == typeof(string);
+    }
+
+    private static bool IsListType(Type type)
+    {
+        var interfaces = type.GetTypeInfo().ImplementedInterfaces;
+        return type.IsArray || interfaces.Any(x => x.Name == typeof(IEnumerable).Name);
+    }
+
+    private static bool IsTrackToNextLevel(Type type)
+    {
+        return type.IsClass || type.IsInterface || type.IsAbstract;
+    }
+    #endregion
 }

src/Plugins/BotSharp.Plugin.SqlDriver/Hooks/SqlDriverAgentHook.cs

@@ -0,0 +1,19 @@
+using BotSharp.Abstraction.Agents.Settings;
+
+namespace BotSharp.Plugin.SqlDriver.Hooks;
+
+public class SqlDriverAgentHook : AgentHookBase, IAgentHook
+{
+    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;
+    }
+}

src/Plugins/BotSharp.Plugin.SqlDriver/SqlDriverPlugin.cs

@@ -27,6 +27,7 @@ public void RegisterDI(IServiceCollection services, IConfiguration config)
         services.AddScoped<DbKnowledgeService>();
         services.AddScoped<IPlanningHook, SqlDriverPlanningHook>();
         services.AddScoped<IKnowledgeHook, SqlDriverKnowledgeHook>();
+        services.AddScoped<IAgentHook, SqlDriverAgentHook>();
         services.AddScoped<IConversationHook, SqlDriverConversationHook>();
         services.AddScoped<IAgentUtilityHook, SqlUtilityHook>();
     }