Merge pull request #126 from evan-cao-wb/Add-filter-when-saving-dialogue

Add filter when saving dialogue

Author: Oceania2018

src/Plugins/BotSharp.Plugin.RoutingSpeeder/RoutingConversationHook.cs

@@ -13,6 +13,7 @@
 using System.Runtime.InteropServices;
 using BotSharp.Abstraction.Agents;
 using System.IO;
+using BotSharp.Abstraction.Routing.Settings;
 
 namespace BotSharp.Plugin.RoutingSpeeder;
 
@@ -49,20 +50,26 @@ public override async Task BeforeCompletion(RoleDialogModel message)
 
     public override async Task AfterCompletion(RoleDialogModel message)
     {
-        // save train data
-        var agentService = _services.CreateScope().ServiceProvider.GetRequiredService<IAgentService>();
-        var rootDataPath = agentService.GetDataDir();
+        var routerSettings = _services.GetRequiredService<RoutingSettings>();
+        bool saveFlag = (message.CurrentAgentId != routerSettings.RouterId) && (message.CurrentAgentId != routerSettings.ReasonerId);
 
-        string rawDataDir = Path.Combine(rootDataPath, "raw_data", $"{message.CurrentAgentId}.txt");
-        var lastThreeDialogs = _dialogs.Where(x => x.Role == AgentRole.User).Select(x => x.Content).Reverse().Take(3).ToArray();
-
-        if (!File.Exists(rawDataDir))
-        {
-            await File.WriteAllLinesAsync(rawDataDir, lastThreeDialogs);
-        }
-        else
+        if (saveFlag)
         {
-            await File.AppendAllLinesAsync(rawDataDir, lastThreeDialogs);
+            // save train data
+            var agentService = _services.CreateScope().ServiceProvider.GetRequiredService<IAgentService>();
+            var rootDataPath = agentService.GetDataDir();
+
+            string rawDataDir = Path.Combine(rootDataPath, "raw_data", $"{message.CurrentAgentId}.txt");
+            var lastThreeDialogs = _dialogs.Where(x => x.Role == AgentRole.User).Select(x => x.Content).Reverse().Take(3).ToArray();
+
+            if (!File.Exists(rawDataDir))
+            {
+                await File.WriteAllLinesAsync(rawDataDir, lastThreeDialogs);
+            }
+            else
+            {
+                await File.AppendAllLinesAsync(rawDataDir, lastThreeDialogs);
+            }
         }
     }
 }