| | | 1 | | using static System.Collections.Specialized.BitVector32; |
| | | 2 | | |
| | | 3 | | namespace UIBlazor.Models; |
| | | 4 | | |
| | | 5 | | public class ConversationSession : BaseOptions |
| | | 6 | | { |
| | | 7 | | /// <summary> |
| | | 8 | | /// Gets or sets the unique identifier for the conversation session. |
| | | 9 | | /// </summary> |
| | | 10 | | [JsonIgnore] |
| | 198 | 11 | | public string Id { get; set; } = string.Empty; |
| | | 12 | | |
| | | 13 | | /// <summary> |
| | | 14 | | /// Gets or sets the list of messages in the conversation. |
| | | 15 | | /// </summary> |
| | 199 | 16 | | public List<VisualChatMessage> Messages { get; set; } = []; |
| | | 17 | | |
| | | 18 | | /// <summary> |
| | | 19 | | /// Gets or sets the timestamp when the conversation was created. |
| | | 20 | | /// </summary> |
| | 107 | 21 | | public DateTime CreatedAt { get; set; } = DateTime.Now; |
| | | 22 | | |
| | | 23 | | /// <summary> |
| | | 24 | | /// Gets or sets the timestamp when the conversation was last updated. |
| | | 25 | | /// </summary> |
| | 109 | 26 | | public DateTime LastUpdated { get; set; } = DateTime.Now; |
| | | 27 | | |
| | | 28 | | /// <summary> |
| | | 29 | | /// Gets or sets the total tokens used in the conversation. |
| | | 30 | | /// </summary> |
| | 285 | 31 | | public int TotalTokens { get; set; } |
| | | 32 | | |
| | | 33 | | /// <summary> |
| | | 34 | | /// Gets or sets the current application mode for this session. |
| | | 35 | | /// </summary> |
| | 708 | 36 | | public AppMode Mode { get; set => SetIfChanged(ref field, value); } = AppMode.Chat; |
| | | 37 | | |
| | | 38 | | /// <summary> |
| | | 39 | | /// Adds a message object to the conversation and manages memory limits. |
| | | 40 | | /// </summary> |
| | | 41 | | public void AddMessage(VisualChatMessage message) |
| | | 42 | | { |
| | 2 | 43 | | Messages.Add(message); |
| | 2 | 44 | | LastUpdated = DateTime.Now; |
| | 2 | 45 | | } |
| | | 46 | | |
| | | 47 | | /// <summary> |
| | | 48 | | /// Removes a message from the conversation. |
| | | 49 | | /// </summary> |
| | | 50 | | public void RemoveMessage(string id) |
| | | 51 | | { |
| | 0 | 52 | | var message = Messages.FirstOrDefault(m => m.Id == id); |
| | 0 | 53 | | if (message != null) |
| | | 54 | | { |
| | 0 | 55 | | Messages.Remove(message); |
| | 0 | 56 | | LastUpdated = DateTime.Now; |
| | | 57 | | } |
| | 0 | 58 | | } |
| | | 59 | | |
| | | 60 | | /// <summary> |
| | | 61 | | /// Updates the content of a message. |
| | | 62 | | /// </summary> |
| | | 63 | | public void UpdateMessage(string id, string content) |
| | | 64 | | { |
| | 0 | 65 | | var message = Messages.FirstOrDefault(m => m.Id == id); |
| | 0 | 66 | | if (message != null) |
| | | 67 | | { |
| | 0 | 68 | | message.Content = content; |
| | 0 | 69 | | LastUpdated = DateTime.Now; |
| | | 70 | | } |
| | 0 | 71 | | } |
| | | 72 | | |
| | | 73 | | /// <summary> |
| | | 74 | | /// Gets the conversation messages formatted for the AI API. |
| | | 75 | | /// </summary> |
| | | 76 | | /// <param name="systemPrompt">The system prompt to include.</param> |
| | | 77 | | /// <returns>A list of message objects for the AI API.</returns> |
| | | 78 | | public IEnumerable<object> GetFormattedMessages(string systemPrompt) |
| | | 79 | | { |
| | 5 | 80 | | var messages = new List<object> |
| | 5 | 81 | | { |
| | 5 | 82 | | // Add system message |
| | 5 | 83 | | new { role = ChatMessageRole.System, content = systemPrompt } |
| | 5 | 84 | | }; |
| | | 85 | | |
| | | 86 | | // Add conversation messages, flattening nested tool messages |
| | 10 | 87 | | foreach (var message in Messages) |
| | | 88 | | { |
| | 0 | 89 | | messages.Add(new { role = message.Role, content = message.Content }); |
| | | 90 | | |
| | | 91 | | // Tool results stored nested must be sent as separate messages to the LLM |
| | 0 | 92 | | foreach (var toolMessage in message.ToolResults) |
| | | 93 | | { |
| | 0 | 94 | | messages.Add(new { role = ChatMessageRole.User, content = toolMessage.Content }); |
| | | 95 | | } |
| | | 96 | | } |
| | | 97 | | |
| | 5 | 98 | | return Messages is [.., { IsStreaming: true }] // не отправлять последнее сообщение, если оно стримится |
| | 5 | 99 | | ? messages.SkipLast(1) |
| | 5 | 100 | | : messages; |
| | | 101 | | } |
| | | 102 | | |
| | | 103 | | public (IEnumerable<object> Messages, VisualChatMessage? LastUserMessage) GetFormattedMessagesForCompress(string sys |
| | | 104 | | { |
| | 0 | 105 | | var messages = new List<object> |
| | 0 | 106 | | { |
| | 0 | 107 | | // Add system message |
| | 0 | 108 | | new { role = ChatMessageRole.System, content = systemPrompt } |
| | 0 | 109 | | }; |
| | | 110 | | |
| | 0 | 111 | | var lastUserMessage = Messages.TakeLast(2).FirstOrDefault(m => m.Role == ChatMessageRole.User); |
| | 0 | 112 | | var compressedMessages = Messages.SkipLast(lastUserMessage is null ? 1 : 2); |
| | | 113 | | |
| | | 114 | | // Add conversation messages, flattening nested tool messages |
| | 0 | 115 | | foreach (var message in compressedMessages) |
| | | 116 | | { |
| | 0 | 117 | | messages.Add(new { role = message.Role, content = message.Content }); |
| | | 118 | | |
| | | 119 | | // Tool results stored nested must be sent as separate messages to the LLM |
| | 0 | 120 | | foreach (var toolMessage in message.ToolResults) |
| | | 121 | | { |
| | 0 | 122 | | messages.Add(new { role = ChatMessageRole.User, content = toolMessage.Content }); |
| | | 123 | | } |
| | | 124 | | } |
| | | 125 | | |
| | 0 | 126 | | messages.Add(new |
| | 0 | 127 | | { |
| | 0 | 128 | | role = ChatMessageRole.User, |
| | 0 | 129 | | content = """ |
| | 0 | 130 | | Act as a Context Compressor. |
| | 0 | 131 | | Summarize the following tool calls and assistant responses. |
| | 0 | 132 | | Preserve: |
| | 0 | 133 | | 1. Final results of searches |
| | 0 | 134 | | 2. Key variable names/file paths |
| | 0 | 135 | | 3. Errors encountered. Remove: redundant JSON structures and intermediate steps. |
| | 0 | 136 | | Format: 'Previous actions: [Concise Summary]' |
| | 0 | 137 | | """ |
| | 0 | 138 | | }); |
| | | 139 | | |
| | 0 | 140 | | return (messages, lastUserMessage); |
| | | 141 | | } |
| | | 142 | | } |
| | | 143 | | |