refactoring

LM-Kit · Nov 28, 2024 · 7071e63 · 7071e63
1 parent aaba4b0
commit 7071e63
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/Services/LMKitMaestroTestsHelpers.cs b/tests/Services/LMKitMaestroTestsHelpers.cs
@@ -7,11 +7,11 @@ namespace LMKit.Maestro.Tests
 {
     internal static class MaestroTestsHelpers
     {
-        public static void AssertPromptResponseIsSuccessful(LMKitService.PromptResult promptResult)
+        public static void AssertPromptResponseIsSuccessful(LMKitService.LMKitResult promptResult)
         {
             Assert.Equal(promptResult.Status, LMKitTextGenerationStatus.Undefined);
             Assert.Null(promptResult.Exception);
-            Assert.NotNull(promptResult.TextGenerationResult);
+            Assert.NotNull(promptResult.Result);
         }
 
         public static void AssertConversationPromptSuccessState(ConversationViewModelWrapper testConversation, int expectedMessageCount = 2)

diff --git a/tests/Services/LmKitDummyConversation.cs b/tests/Services/LmKitDummyConversation.cs
@@ -6,7 +6,7 @@ internal class LMKitDummyConversation
     {
         LMKitService.Conversation Conversation { get; }
 
-        public TaskCompletionSource<LMKitService.PromptResult?> PromptResultTask { get; private set; } = new TaskCompletionSource<LMKitService.PromptResult?>();
+        public TaskCompletionSource<LMKitService.LMKitResult?> PromptResultTask { get; private set; } = new TaskCompletionSource<LMKitService.LMKitResult?>();
 
         public LMKitDummyConversation(LMKitService lmKitService)
         {
@@ -15,7 +15,7 @@ public LMKitDummyConversation(LMKitService lmKitService)
 
         public void SubmitPrompt(LMKitService lmKitService, string prompt)
         {
-            LMKitService.PromptResult? promptResult = null;
+            LMKitService.LMKitResult? promptResult = null;
 
             Task.Run(async () =>
             {