refactoring

LM-Kit · Nov 28, 2024 · aaba4b0 · aaba4b0
1 parent 07f6be1
commit aaba4b0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/LM-Kit-Maestro/ViewModels/ConversationViewModel.cs b/LM-Kit-Maestro/ViewModels/ConversationViewModel.cs
@@ -163,7 +163,7 @@ protected override void HandleSubmit()
         string prompt = InputText;
         OnNewlySubmittedPrompt(prompt);
 
-        LMKitService.PromptResult? promptResult = null;
+        LMKitService.LMKitResult? promptResult = null;
 
         Task.Run(async () =>
         {
@@ -179,7 +179,7 @@ protected override void HandleSubmit()
         });
     }
 
-    private void OnPromptResult(LMKitService.PromptResult? promptResult, Exception? submitPromptException = null)
+    private void OnPromptResult(LMKitService.LMKitResult? promptResult, Exception? submitPromptException = null)
     {
         AwaitingResponse = false;
 
@@ -205,9 +205,9 @@ private void OnPromptResult(LMKitService.PromptResult? promptResult, Exception?
             _pendingResponse!.Status = LatestPromptStatus;
             _pendingPrompt!.Status = LatestPromptStatus;
 
-            if (promptResult.Status == LMKitTextGenerationStatus.Undefined && promptResult.TextGenerationResult != null)
+            if (promptResult.Status == LMKitTextGenerationStatus.Undefined && promptResult.Result is TextGenerationResult textGenerationResult)
             {
-                OnTextGenerationSuccess(promptResult.TextGenerationResult);
+                OnTextGenerationSuccess(textGenerationResult);
             }
             else
             {