Add support for system role in assertions

mehrandvd · Dec 25, 2023 · 0baa635 · 0baa635
1 parent b86e3b0
commit 0baa635
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/skUnit/Asserts/SemanticKernelAssert_Chat.cs b/src/skUnit/Asserts/SemanticKernelAssert_Chat.cs
@@ -24,6 +24,12 @@ public static async Task ScenarioChatSuccessAsync(Kernel kernel, ChatScenario sc
             Log(chatItem.Content);
             Log();
 
+            if (chatItem.Role == AuthorRole.System)
+            {
+                chatHistory.AddUserMessage(chatItem.Content);
+                continue;
+            }
+
             if (chatItem.Role == AuthorRole.User)
             {
                 chatHistory.AddUserMessage(chatItem.Content);