Update ConversationViewModel.cs

LM-Kit · Dec 2, 2024 · 594d5e1 · 594d5e1
1 parent b2045d3
commit 594d5e1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/LM-Kit-Maestro/ViewModels/ConversationViewModel.cs b/LM-Kit-Maestro/ViewModels/ConversationViewModel.cs
@@ -219,7 +219,8 @@ private void OnNewlySubmittedPrompt(string prompt)
     private void OnTextGenerationResult(LMKitService.LMKitResult? result, Exception? exception = null)
     {
         AwaitingResponse = false;
-        Messages.Last().Status = result != null ? result.Status : LMKitTextGenerationStatus.GenericError;
+        Messages.Last().Status = result != null ? result.Status : exception is OperationCanceledException ?
+            LMKitTextGenerationStatus.Cancelled : LMKitTextGenerationStatus.GenericError;
 
         if (exception != null || result?.Exception != null)
         {