Merge pull request #330 from uezo/workaround-clear-llm-context

Provide workaround to clear state data, including LLM context
uezo · Sep 1, 2024 · 10bcb5f · 10bcb5f
2 parents ded4a4d + f4b4cd7
commit 10bcb5f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/Scripts/Dialog/DialogController.cs b/Scripts/Dialog/DialogController.cs
@@ -569,6 +569,18 @@ public CancellationToken GetDialogToken()
             return dialogTokenSource.Token;
         }
 
+        public async UniTask ClearStateAsync(string userId = null)
+        {
+            if (requestProcessor is LocalRequestProcessor)
+            {
+                if (string.IsNullOrEmpty(userId))
+                {
+                    userId = GetClientIdDefault();
+                }
+                await ((LocalRequestProcessor)requestProcessor).ClearStateAsync(userId);
+            }           
+        }
+
         private void Update()
         {
             // Control mute

diff --git a/Scripts/Dialog/Processor/LocalRequestProcessor.cs b/Scripts/Dialog/Processor/LocalRequestProcessor.cs
@@ -151,5 +151,10 @@ public virtual async UniTask<Response> ProcessRequestAsync(Request request, Canc
                 throw ex;
             }
         }
+
+        public virtual async UniTask ClearStateAsync(string userId)
+        {
+            await StateStore.DeleteStateAsync(userId);
+        }
     }
 }