OnConversationRedirected

Author: Haiping Chen

src/Infrastructure/BotSharp.Abstraction/Conversations/ConversationHookBase.cs

@@ -91,4 +91,9 @@ public virtual Task OnUserAgentConnectedInitially(Conversation conversation)
     {
         return Task.CompletedTask;
     }
+
+    public virtual Task OnConversationRedirected(string toAgentId, RoleDialogModel message)
+    {
+        return Task.CompletedTask;
+    }
 }

src/Infrastructure/BotSharp.Abstraction/Conversations/IConversationHook.cs

@@ -82,4 +82,12 @@ public interface IConversationHook
     /// <param name="conversation"></param>
     /// <returns></returns>
     Task OnHumanInterventionNeeded(RoleDialogModel message);
+
+    /// <summary>
+    /// Conversation is redirected to another agent
+    /// </summary>
+    /// <param name="toAgentId"></param>
+    /// <param name="message"></param>
+    /// <returns></returns>
+    Task OnConversationRedirected(string toAgentId, RoleDialogModel message);
 }

src/Infrastructure/BotSharp.Abstraction/Loggers/Enums/ContentLogSource.cs

@@ -6,4 +6,5 @@ public static class ContentLogSource
     public const string Prompt = "prompt";
     public const string FunctionCall = "function call";
     public const string AgentResponse = "agent response";
+    public const string HardRule = "hard rule";
 }

src/Infrastructure/BotSharp.Core/Routing/Functions/RouteToAgentFn.cs

@@ -153,6 +153,11 @@ private bool HasMissingRequiredField(RoleDialogModel message, out string agentId
 #else
                 logger.LogInformation($"*** Routing redirect to {record.Name.ToUpper()} ***");
 #endif
+                var hooks = _services.GetServices<IConversationHook>();
+                foreach (var hook in hooks)
+                {
+                    hook.OnConversationRedirected(routingRule.RedirectTo, message);
+                }
             }
             else
             {

src/Plugins/BotSharp.Plugin.ChatHub/Hooks/StreamingLogHook.cs

@@ -46,6 +46,18 @@ await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerat
             BuildContentLog(conversationId, _user.UserName, log, ContentLogSource.UserInput, message));
     }
 
+    public override async Task OnConversationRedirected(string toAgentId, RoleDialogModel message)
+    {
+        var agentService = _services.GetRequiredService<IAgentService>();
+        var conversationId = _state.GetConversationId();
+        var fromAgent = await agentService.LoadAgent(message.CurrentAgentId);
+        var toAgent = await agentService.LoadAgent(toAgentId);
+
+        var log = $"{message.Content}\r\n=====\r\nREDIRECTED TO {toAgent.Name}";
+        await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated",
+            BuildContentLog(conversationId, fromAgent.Name, log, ContentLogSource.HardRule, message));
+    }
+
     public async Task BeforeGenerating(Agent agent, List<RoleDialogModel> conversations)
     {
         if (!_convSettings.ShowVerboseLog) return;
@@ -83,23 +95,26 @@ public async Task AfterGenerated(RoleDialogModel message, TokenStatsModel tokenS
         var agent = await agentService.LoadAgent(message.CurrentAgentId);
         var logSource = string.Empty;
 
+        var log = tokenStats.Prompt;
+        logSource = ContentLogSource.Prompt;
+        await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated",
+            BuildContentLog(conversationId, agent?.Name, log, logSource, message));
+
         // Log routing output
         try
         {
             var inst = message.Content.JsonContent<FunctionCallFromLlm>();
-            logSource = ContentLogSource.AgentResponse;
-            await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated",
-                BuildContentLog(conversationId, agent?.Name, message.Content, logSource, message));
+            if (!string.IsNullOrEmpty(inst.Function))
+            {
+                logSource = ContentLogSource.AgentResponse;
+                await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated",
+                    BuildContentLog(conversationId, agent?.Name, message.Content, logSource, message));
+            }
         }
         catch
         {
             // ignore
         }
-
-        var log = tokenStats.Prompt;
-        logSource = ContentLogSource.Prompt;
-        await _chatHub.Clients.User(_user.Id).SendAsync("OnConversationContentLogGenerated",
-            BuildContentLog(conversationId, agent?.Name, log, logSource, message));
     }
 
     /// <summary>
@@ -118,7 +133,7 @@ public override async Task OnResponseGenerated(RoleDialogModel message)
         {
             var agentService = _services.GetRequiredService<IAgentService>();
             var agent = await agentService.LoadAgent(message.CurrentAgentId);
-            var log = $"[{agent?.Name}]: {message.Content}";
+            var log = $"{message.Content}";
             if (message.RichContent != null && message.RichContent.Message.RichType != "text")
             {
                 var richContent = JsonSerializer.Serialize(message.RichContent, _serializerOptions);