| | | 1 | | namespace UIBlazor.Models; |
| | | 2 | | |
| | | 3 | | public class ConversationSession |
| | | 4 | | { |
| | | 5 | | /// <summary> |
| | | 6 | | /// Gets or sets the unique identifier for the conversation session. |
| | | 7 | | /// </summary> |
| | | 8 | | [JsonIgnore] |
| | 21 | 9 | | public string Id { get; set; } = string.Empty; |
| | | 10 | | |
| | | 11 | | /// <summary> |
| | | 12 | | /// Gets or sets the list of messages in the conversation. |
| | | 13 | | /// </summary> |
| | 19 | 14 | | public List<VisualChatMessage> Messages { get; set; } = []; |
| | | 15 | | |
| | | 16 | | /// <summary> |
| | | 17 | | /// Gets or sets the timestamp when the conversation was created. |
| | | 18 | | /// </summary> |
| | 8 | 19 | | public DateTime CreatedAt { get; set; } = DateTime.Now; |
| | | 20 | | |
| | | 21 | | /// <summary> |
| | | 22 | | /// Gets or sets the timestamp when the conversation was last updated. |
| | | 23 | | /// </summary> |
| | 9 | 24 | | public DateTime LastUpdated { get; set; } = DateTime.Now; |
| | | 25 | | |
| | | 26 | | /// <summary> |
| | | 27 | | /// Gets or sets the maximum number of messages to keep in memory. |
| | | 28 | | /// </summary> |
| | | 29 | | [JsonIgnore] |
| | 12 | 30 | | public int MaxMessages { get; set; } = 50; |
| | | 31 | | |
| | | 32 | | /// <summary> |
| | | 33 | | /// Gets or sets the total tokens used in the conversation. |
| | | 34 | | /// </summary> |
| | 0 | 35 | | public int TotalTokens { get; set; } |
| | | 36 | | |
| | | 37 | | /// <summary> |
| | | 38 | | /// Gets or sets the current application mode for this session. |
| | | 39 | | /// </summary> |
| | 0 | 40 | | public AppMode Mode { get; set; } = AppMode.Chat; |
| | | 41 | | |
| | | 42 | | /// <summary> |
| | | 43 | | /// Adds a message to the conversation and manages memory limits. |
| | | 44 | | /// </summary> |
| | | 45 | | /// <param name="role">The role of the message sender.</param> |
| | | 46 | | /// <param name="content">The message content.</param> |
| | | 47 | | public void AddMessage(string role, string content) |
| | | 48 | | { |
| | 1 | 49 | | AddMessage(new VisualChatMessage |
| | 1 | 50 | | { |
| | 1 | 51 | | Role = role, |
| | 1 | 52 | | Content = content, |
| | 1 | 53 | | Timestamp = DateTime.Now |
| | 1 | 54 | | }); |
| | 1 | 55 | | } |
| | | 56 | | |
| | | 57 | | /// <summary> |
| | | 58 | | /// Adds a message object to the conversation and manages memory limits. |
| | | 59 | | /// </summary> |
| | | 60 | | public void AddMessage(VisualChatMessage message) |
| | | 61 | | { |
| | 1 | 62 | | Messages.Add(message); |
| | 1 | 63 | | LastUpdated = DateTime.Now; |
| | | 64 | | |
| | | 65 | | // Remove oldest messages if we exceed the limit |
| | 1 | 66 | | while (Messages.Count > MaxMessages) |
| | | 67 | | { |
| | 0 | 68 | | Messages.RemoveAt(0); |
| | | 69 | | } |
| | 1 | 70 | | } |
| | | 71 | | |
| | | 72 | | /// <summary> |
| | | 73 | | /// Removes a message from the conversation. |
| | | 74 | | /// </summary> |
| | | 75 | | public void RemoveMessage(string id) |
| | | 76 | | { |
| | 0 | 77 | | var message = Messages.FirstOrDefault(m => m.Id == id); |
| | 0 | 78 | | if (message != null) |
| | | 79 | | { |
| | 0 | 80 | | Messages.Remove(message); |
| | 0 | 81 | | LastUpdated = DateTime.Now; |
| | | 82 | | } |
| | 0 | 83 | | } |
| | | 84 | | |
| | | 85 | | /// <summary> |
| | | 86 | | /// Updates the content of a message. |
| | | 87 | | /// </summary> |
| | | 88 | | public void UpdateMessage(string id, string content) |
| | | 89 | | { |
| | 0 | 90 | | var message = Messages.FirstOrDefault(m => m.Id == id); |
| | 0 | 91 | | if (message != null) |
| | | 92 | | { |
| | 0 | 93 | | message.Content = content; |
| | 0 | 94 | | LastUpdated = DateTime.Now; |
| | | 95 | | } |
| | 0 | 96 | | } |
| | | 97 | | |
| | | 98 | | /// <summary> |
| | | 99 | | /// Gets the conversation messages formatted for the AI API. |
| | | 100 | | /// </summary> |
| | | 101 | | /// <param name="systemPrompt">The system prompt to include.</param> |
| | | 102 | | /// <returns>A list of message objects for the AI API.</returns> |
| | | 103 | | public List<object> GetFormattedMessages(string systemPrompt) |
| | | 104 | | { |
| | 0 | 105 | | var messages = new List<object> |
| | 0 | 106 | | { |
| | 0 | 107 | | // Add system message |
| | 0 | 108 | | new { role = "system", content = systemPrompt } |
| | 0 | 109 | | }; |
| | | 110 | | |
| | | 111 | | // Add conversation messages |
| | 0 | 112 | | foreach (var message in Messages) |
| | | 113 | | { |
| | 0 | 114 | | messages.Add(new { role = message.Role, content = message.Content }); |
| | | 115 | | } |
| | | 116 | | |
| | 0 | 117 | | return messages; |
| | | 118 | | } |
| | | 119 | | } |
| | | 120 | | |