Fix the output for assistant.

mehrandvd · Jan 2, 2024 · 18da9ad · 18da9ad
1 parent 81916f4
commit 18da9ad
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/skUnit/Asserts/SemanticKernelAssert_Chat.cs b/src/skUnit/Asserts/SemanticKernelAssert_Chat.cs
@@ -40,19 +40,21 @@ public async Task CheckChatScenarioAsync(Kernel kernel, ChatScenario scenario, F
         while (queue.Count > 0)
         {
             var chatItem = queue.Dequeue();
-            Log($"## [{chatItem.Role.ToString().ToUpper()}]");
-            Log(chatItem.Content);
-            Log();
+
 
             if (chatItem.Role == AuthorRole.System)
             {
                 chatHistory.AddSystemMessage(chatItem.Content);
-                continue;
-            }
-
-            if (chatItem.Role == AuthorRole.User)
+                Log($"## [{chatItem.Role.ToString().ToUpper()}]");
+                Log(chatItem.Content);
+                Log();
+            } 
+            else if (chatItem.Role == AuthorRole.User)
             {
                 chatHistory.AddUserMessage(chatItem.Content);
+                Log($"## [{chatItem.Role.ToString().ToUpper()}]");
+                Log(chatItem.Content);
+                Log();
             } 
             else if (chatItem.Role == AuthorRole.Assistant)
             {