Merge pull request #1115 from iceljc/features/add-optional-migrate-agents

add option agents

Author: iceljc

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

@@ -10,7 +10,7 @@ namespace BotSharp.Abstraction.Agents;
 public interface IAgentService
 {
     Task<Agent> CreateAgent(Agent agent);
-    Task<string> RefreshAgents();
+    Task<string> RefreshAgents(IEnumerable<string>? agentIds = null);
     Task<PagedItems<Agent>> GetAgents(AgentFilter filter);
     Task<List<IdName>> GetAgentOptions(List<string>? agentIds = null, bool byName = false);
     Task<IEnumerable<AgentUtility>> GetAgentUtilityOptions();

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

@@ -6,7 +6,7 @@ namespace BotSharp.Core.Agents.Services;
 
 public partial class AgentService
 {
-    public async Task<string> RefreshAgents()
+    public async Task<string> RefreshAgents(IEnumerable<string>? agentIds = null)
     {
         string refreshResult;
         var dbSettings = _services.GetRequiredService<BotSharpDatabaseSettings>();
@@ -28,7 +28,13 @@ public async Task<string> RefreshAgents()
         }
 
         var refreshedAgents = new List<string>();
-        foreach (var dir in Directory.GetDirectories(agentDir))
+        var dirs = Directory.GetDirectories(agentDir);
+        if (!agentIds.IsNullOrEmpty())
+        {
+            dirs = dirs.Where(x => agentIds.Contains(x.Split(Path.DirectorySeparatorChar).Last())).ToArray();
+        }
+
+        foreach (var dir in dirs)
         {
             try
             {
@@ -78,7 +84,7 @@ public async Task<string> RefreshAgents()
         }
         else
         {
-            refreshResult = "No agent gets refreshed!";
+            refreshResult = "No agent gets migrated!";
         }
 
         _logger.LogInformation(refreshResult);

src/Infrastructure/BotSharp.OpenAPI/Controllers/AgentController.cs

@@ -111,9 +111,9 @@ public async Task<AgentViewModel> CreateAgent(AgentCreationModel agent)
 
     [BotSharpAuth]
     [HttpPost("/refresh-agents")]
-    public async Task<string> RefreshAgents()
+    public async Task<string> RefreshAgents([FromBody] AgentMigrationModel request)
     {
-        return await _agentService.RefreshAgents();
+        return await _agentService.RefreshAgents(request?.AgentIds);
     }
 
     [HttpPut("/agent/file/{agentId}")]

src/Infrastructure/BotSharp.OpenAPI/ViewModels/Agents/Request/AgentMigrationModel.cs

@@ -0,0 +1,10 @@
+using System.Text.Json.Serialization;
+
+namespace BotSharp.OpenAPI.ViewModels.Agents;
+
+public class AgentMigrationModel
+{
+    [JsonPropertyName("agent_ids")]
+    [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+    public IEnumerable<string>? AgentIds { get; set; }
+}