progress

LM-Kit · Nov 28, 2024 · 5feb001 · 5feb001
1 parent d2a5adb
commit 5feb001
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/LM-Kit-Maestro/Services/LMKitService.cs b/LM-Kit-Maestro/Services/LMKitService.cs
@@ -210,11 +210,11 @@ private async Task<LMKitResult> HandleLmKitRequest(LMKitRequest promptRequest)
         // Ensuring we don't touch anything until Lm-Kit objects' state has been set to handle this prompt request.
         _lmKitServiceSemaphore.Wait();
 
-        LMKitResult promptResult;
+        LMKitResult result;
 
         if (promptRequest.CancellationTokenSource.IsCancellationRequested || ModelLoadingState == LMKitModelLoadingState.Unloaded)
         {
-            promptResult = new LMKitResult()
+            result = new LMKitResult()
             {
                 Status = LMKitTextGenerationStatus.Cancelled
             };
@@ -223,20 +223,24 @@ private async Task<LMKitResult> HandleLmKitRequest(LMKitRequest promptRequest)
         }
         else
         {
-            BeforeSubmittingPrompt(((PromptRequestParameters)promptRequest.Parameters!).Conversation);
+            if (promptRequest.RequestType == LMKitRequestType.Prompt)
+            {
+                BeforeSubmittingPrompt(((PromptRequestParameters)promptRequest.Parameters!).Conversation);
+            }
+
             _lmKitServiceSemaphore.Release();
 
-            promptResult = await SubmitRequest(promptRequest);
+            result = await SubmitRequest(promptRequest);
         }
 
         if (_requestSchedule.Contains(promptRequest))
         {
             _requestSchedule.Remove(promptRequest);
         }
 
-        promptRequest.ResponseTask.TrySetResult(promptResult);
+        promptRequest.ResponseTask.TrySetResult(result);
 
-        return promptResult;
+        return result;
 
     }
 

diff --git a/LM-Kit-Maestro/ViewModels/ConversationViewModel.cs b/LM-Kit-Maestro/ViewModels/ConversationViewModel.cs
@@ -155,7 +155,7 @@ public async Task Delete()
     [RelayCommand]
     private async Task RegenerateResponse(MessageViewModel message)
     {
-        var response = await _lmKitService.RegenerateResponse(_lmKitConversation, message.LMKitMessage);
+        var response = await _lmKitService.RegenerateResponse(_lmKitConversation, message.LMKitMessage!);
     }
 
     protected override void HandleSubmit()