ChatMessageExtensions.cs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

namespace Lilith.Core.Chat;

public static class ChatMessageExtensions
{
    public static ChatMessage Snapshot(this ChatMessage message) =>
        message with { Images = message.Images is null ? null : [.. message.Images] };

    public static SavedChatMessage ToSavedMessage(this ChatMessage message) => new()
    {
        Role = message.Role,
        Content = message.Content ?? string.Empty,
        Tokens = message.Tokens,
        Images = message.Images is { Count: > 0 } ? [.. message.Images] : null
    };

    public static ChatMessage ToChatMessage(this SavedChatMessage message) =>
        new(message.Role, message.Content, message.Images is null ? null : [.. message.Images], message.Tokens);
}