Update ToLLamaSharpChatHistory extension method to be public and support semantic-kernel author roles

This commit is contained in:
Chirag Karia 2023-11-11 00:43:11 -05:00
parent bb03376594
commit 1b4659dff9
1 changed files with 3 additions and 3 deletions

View File

@ -3,9 +3,9 @@ using Microsoft.SemanticKernel.AI.ChatCompletion;
namespace LLamaSharp.SemanticKernel;
internal static class ExtensionMethods
public static class ExtensionMethods
{
internal static global::LLama.Common.ChatHistory ToLLamaSharpChatHistory(this ChatHistory chatHistory)
public static global::LLama.Common.ChatHistory ToLLamaSharpChatHistory(this ChatHistory chatHistory)
{
if (chatHistory is null)
{
@ -16,7 +16,7 @@ internal static class ExtensionMethods
foreach (var chat in chatHistory)
{
var role = Enum.TryParse<global::LLama.Common.AuthorRole>(chat.Role.Label, out var _role) ? _role : global::LLama.Common.AuthorRole.Unknown;
var role = Enum.TryParse<global::LLama.Common.AuthorRole>(chat.Role.Label, true, out var _role) ? _role : global::LLama.Common.AuthorRole.Unknown;
history.AddMessage(role, chat.Content);
}