Fix the problem with chat history

mehrandvd · Dec 29, 2023 · 8141322 · 8141322
1 parent 5e5df20
commit 8141322
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/skUnit/Asserts/SemanticKernelAssert_Chat.cs b/src/skUnit/Asserts/SemanticKernelAssert_Chat.cs
@@ -73,8 +73,6 @@ public async Task CheckChatScenarioAsync(Kernel kernel, ChatScenario scenario, F
 
             if (chatItem.Role == AuthorRole.Assistant)
             {
-                chatHistory.AddAssistantMessage(chatItem.Content);
-
                 Log($"## [EXPECTED ANSWER]");
                 Log(chatItem.Content);
                 Log();
@@ -88,7 +86,13 @@ public async Task CheckChatScenarioAsync(Kernel kernel, ChatScenario scenario, F
                     """);
             }
 
+
             var answer = await getAnswerFunc(chatHistory);
+
+            // To let chatHistory stay clean for gettig the answer
+            chatHistory.AddAssistantMessage(chatItem.Content);
+
+
             Log($"### [ACTUAL ANSWER]");
             Log(answer);
             Log();