Update LmKitService.RequestSchedule.cs

LM-Kit · Nov 29, 2024 · 8a9155c · 8a9155c
1 parent 23236e5
commit 8a9155c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/LM-Kit-Maestro/Services/LmKitService.RequestSchedule.cs b/LM-Kit-Maestro/Services/LmKitService.RequestSchedule.cs
@@ -79,7 +79,12 @@ public void Remove(LMKitRequest scheduledPrompt)
             {
                 foreach (var scheduledPrompt in _scheduledPrompts)
                 {
-                    if (scheduledPrompt.Parameters is LMKitRequest.PromptRequestParameters parameter && parameter.Conversation == conversation)
+                    if (scheduledPrompt.Parameters is LMKitRequest.PromptRequestParameters promptParameters && promptParameters.Conversation == conversation)
+                    {
+                        prompt = scheduledPrompt;
+                        break;
+                    }
+                    else if (scheduledPrompt.Parameters is LMKitRequest.RegenerateResponseParameters regenerateParameters && regenerateParameters.Conversation == conversation)
                     {
                         prompt = scheduledPrompt;
                         break;