Merge pull request #1169 from iceljc/features/refine-model-settings

Features/refine model settings

Author: iceljc

src/Infrastructure/BotSharp.Core.SideCar/Services/BotSharpConversationSideCar.cs

@@ -15,9 +15,6 @@ limitations under the License.
 ******************************************************************************/
 
 using BotSharp.Core.Infrastructures;
-using NetTopologySuite.Index.KdTree;
-using Newtonsoft.Json.Linq;
-using static System.Runtime.InteropServices.JavaScript.JSType;
 
 namespace BotSharp.Core.SideCar.Services;
 

src/Infrastructure/BotSharp.Core/Conversations/Services/ConversationStateService.cs

@@ -14,8 +14,6 @@ You may obtain a copy of the License at
   limitations under the License.
 ******************************************************************************/
 
-using BotSharp.Abstraction.Conversations.Enums;
-using BotSharp.Abstraction.Hooks;
 using BotSharp.Abstraction.Options;
 using BotSharp.Abstraction.SideCar;
 

src/Infrastructure/BotSharp.Core/Demo/Functions/GetWeatherFn.cs

@@ -37,11 +37,11 @@ public async Task<bool> Execute(RoleDialogModel message)
         await Task.Delay(1500);
 
 #if DEBUG
-        var temp = RoleDialogModel.From(message, AgentRole.Assistant, $"Here is your weather in {args?.City}");
+        var intermediateMsg = RoleDialogModel.From(message, AgentRole.Assistant, $"Here is your weather in {args?.City}");
         messageHub.Push(new()
         {
             EventName = ChatEvent.OnIntermediateMessageReceivedFromAssistant,
-            Data = temp,
+            Data = intermediateMsg,
             RefId = conv.ConversationId,
             SaveDataToDb = true
         });

src/Infrastructure/BotSharp.Core/MessageHub/Observers/ConversationObserver.cs

@@ -30,6 +30,8 @@ public override void OnError(Exception error)
     public override void OnNext(HubObserveData<RoleDialogModel> value)
     {
         var conv = _services.GetRequiredService<IConversationService>();
+        var storage = _services.GetRequiredService<IConversationStorage>();
+        var routeCtx = _services.GetRequiredService<IRoutingContext>();
 
         if (value.EventName == ChatEvent.OnIndicationReceived)
         {
@@ -41,5 +43,16 @@ public override void OnNext(HubObserveData<RoleDialogModel> value)
                 func(value).ConfigureAwait(false).GetAwaiter().GetResult();
             }
         }
+        else if (value.EventName == ChatEvent.OnIntermediateMessageReceivedFromAssistant)
+        {
+            var dialogs = routeCtx.GetDialogs();
+            dialogs.Add(value.Data);
+            routeCtx.SetDialogs(dialogs);
+
+            if (value.SaveDataToDb)
+            {
+                storage.Append(conv.ConversationId, value.Data);
+            }
+        }
     }
 }

src/Plugins/BotSharp.Plugin.ChatHub/Observers/ChatHubObserver.cs

@@ -140,12 +140,6 @@ public override void OnNext(HubObserveData<RoleDialogModel> value)
                         Role = AgentRole.Assistant
                     }
                 };
-
-                if (value.SaveDataToDb)
-                {
-                    var storage = _services.GetRequiredService<IConversationStorage>();
-                    storage.Append(conv.ConversationId, message);
-                }
                 break;
         }