Merge pull request #1 from jackjiang-sms/feature/AII-400

AII-400

Author: jackjiang-sms

src/Infrastructure/BotSharp.Abstraction/Agents/AgentHookBase.cs

@@ -56,4 +56,9 @@ public virtual void OnAgentLoaded(Agent agent)
     public virtual void OnAgentUtilityLoaded(Agent agent)
     {
     }
+
+    public virtual void OnAgentLoadFilter(Agent agent)
+    {
+
+    }
 }

src/Infrastructure/BotSharp.Abstraction/Agents/IAgentHook.cs

@@ -33,4 +33,6 @@ public interface IAgentHook
     /// <param name="agent"></param>
     /// <returns></returns>
     void OnAgentLoaded(Agent agent);
+
+    void OnAgentLoadFilter(Agent agent);
 }

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

@@ -5,16 +5,35 @@ namespace BotSharp.Core.Agents.Services;
 
 public partial class AgentService
 {
-    public static ConcurrentDictionary<string, Dictionary<string,string>> AgentParameterTypes = new();
+    public static ConcurrentDictionary<string, Dictionary<string, string>> AgentParameterTypes = new();
 
-    [MemoryCache(10 * 60, perInstanceCache: true)]
     public async Task<Agent> LoadAgent(string id)
     {
         if (string.IsNullOrEmpty(id) || id == Guid.Empty.ToString())
         {
             return null;
         }
 
+        Agent agent = await GetLoadAgent(id);
+        OnAgentLoadFilter(agent);
+        return agent;
+    }
+
+    private void OnAgentLoadFilter(Agent? agent)
+    {
+        if (agent != null && agent.Type == AgentType.Routing)
+        {
+            var hooks = _services.GetServices<IAgentHook>();
+            foreach (var hook in hooks)
+            {
+                hook.OnAgentLoadFilter(agent);
+            }
+        }
+    }
+
+    [MemoryCache(10 * 60, perInstanceCache: true)]
+    private async Task<Agent> GetLoadAgent(string id)
+    {
         var hooks = _services.GetServices<IAgentHook>();
 
         // Before agent is loaded.
@@ -106,14 +125,14 @@ private void AddOrUpdateParameters(Agent agent)
     {
         var agentId = agent.Id ?? agent.Name;
         if (AgentParameterTypes.ContainsKey(agentId)) return;
-        
+
         AddOrUpdateRoutesParameters(agentId, agent.RoutingRules);
         AddOrUpdateFunctionsParameters(agentId, agent.Functions);
     }
 
     private void AddOrUpdateRoutesParameters(string agentId, List<RoutingRule> routingRules)
     {
-        if(!AgentParameterTypes.TryGetValue(agentId, out var parameterTypes))
+        if (!AgentParameterTypes.TryGetValue(agentId, out var parameterTypes))
         {
             parameterTypes = new();
         }