Merge pull request #410 from uezo/improve-dialog-processing-20241214

Improve dialog processing
uezo · Dec 13, 2024 · a091d8a · a091d8a
2 parents d7854ac + 65ddef1
commit a091d8a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/Scripts/Dialog/DialogPriorityManager.cs b/Scripts/Dialog/DialogPriorityManager.cs
@@ -49,16 +49,16 @@ public void SetRequest(string text, Dictionary<string, object> payloads = null,
         {
             if (priority == 0)
             {
-                _ = dialogProcessor.StartDialogAsync(text);
+                _ = dialogProcessor.StartDialogAsync(text + textToAppendNext);
             }
             else
             {
                 dialogQueue.Enqueue(new DialogQueueItem() {
                     Priority = priority, Text = text + textToAppendNext, Payloads = payloads
                 }, priority);
-
-                textToAppendNext = string.Empty;
             }
+
+            textToAppendNext = string.Empty;
         }
 
         public void SetRequestToAppendNext(string text)

diff --git a/Scripts/Dialog/DialogProcessor.cs b/Scripts/Dialog/DialogProcessor.cs
@@ -27,6 +27,7 @@ public enum DialogStatus
         // Actions for each status
         public Func<string, Dictionary<string, object>, CancellationToken, UniTask> OnStartAsync { get; set; }
         public Func<string, Dictionary<string, object>, CancellationToken, UniTask> OnRequestRecievedAsync { get; set; }
+        public Func<ILLMSession, CancellationToken, UniTask> OnBeforeProcessContentStreamAsync { get; set; }
         public Func<string, Dictionary<string, object>, ILLMSession, CancellationToken, UniTask> OnResponseShownAsync { get; set; }
         public Func<bool, CancellationToken, UniTask> OnEndAsync { get; set; }
         public Func<bool, UniTask> OnStopAsync { get; set; }
@@ -109,7 +110,7 @@ public void LoadLLMTools()
         }
 
         // Start dialog
-        public async UniTask StartDialogAsync(string text, Dictionary<string, object> payloads = null)
+        public async UniTask StartDialogAsync(string text, Dictionary<string, object> payloads = null, bool overwrite = true)
         {
             if (string.IsNullOrEmpty(text) && (payloads == null || payloads.Count == 0))
             {
@@ -120,8 +121,11 @@ public async UniTask StartDialogAsync(string text, Dictionary<string, object> pa
             processingId = Guid.NewGuid().ToString();
             var currentProcessingId = processingId;
 
-            // Stop running dialog and get cancellation token
-            await StopDialog(true);
+            if (overwrite)
+            {
+                // Stop running dialog and get cancellation token
+                await StopDialog(true);
+            }
 
             var token = GetDialogToken();
 
@@ -166,6 +170,8 @@ public async UniTask StartDialogAsync(string text, Dictionary<string, object> pa
                     }
                 }
 
+                await OnBeforeProcessContentStreamAsync(llmSession, token);
+
                 // Start parsing voices, faces and animations
                 var processContentStreamTask = llmContentProcessor.ProcessContentStreamAsync(llmSession, token);