feat:db context settings.

arthuridea · Mar 5, 2024 · c2be716 · c2be716
1 parent 437c1e0
commit c2be716
Show file tree

Hide file tree

Showing 6 changed files with 45 additions and 25 deletions.
diff --git a/src/LLMService.DataProvider.Relational/Data/ChatDbContext.cs b/src/LLMService.DataProvider.Relational/Data/ChatDbContext.cs
@@ -1,7 +1,12 @@
 using LLMService.DataProvider.Relational.Entity;
+using LLMService.Shared.Extensions;
 using Microsoft.EntityFrameworkCore;
-using Microsoft.EntityFrameworkCore.Infrastructure;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
+using System.ComponentModel;
+using System.ComponentModel.DataAnnotations;
+using System.Text;
+using static LLMService.DataProvider.Relational.Entity.EntityConsts;
+using static LLMService.Shared.Models.LLMApiDefaults;
 
 namespace LLMService.DataProvider.Relational.Data
 {
@@ -218,12 +223,45 @@ private void _configureChatMessage(EntityTypeBuilder<TMessage> builder)
             builder.ToTable("ChatMessage");
             builder.HasKey(x => x.Id);
 
+            builder.Property(x => x.ConversationId)
+                   .IsRequired(true);
+
             builder.Property(x => x.Role)
                    .HasMaxLength(256);
 
-        }
+            // model type comments.
 
+            builder.Property(x => x.ModelType)
+                   .HasComment(_getCommentFromEnum<LLM_ModelType>());
 
+            // MessageDataType comments.
+            builder.Property(x => x.DataType)
+                   .HasComment(_getCommentFromEnum<MessageDataType>());
+
+            // MessageContentType comments.
+            builder.Property(x => x.ContentType)
+                   .HasComment(_getCommentFromEnum<MessageContentType>());
+        }
+
+        private static string _getCommentFromEnum<TEnum>()
+            where TEnum : Enum
+        {
+            StringBuilder comments = new();
+            var lists = Enum.GetValues(typeof(TEnum))
+                            .Cast<TEnum>()
+                            .Select(x => new
+                            {
+                                Value = (int)(object)x,
+                                Name = x.ToString(),
+                                DisplayName = x.GetAttribute<DisplayAttribute>()?.Name ?? "",
+                                Description = x.GetAttribute<DescriptionAttribute>()?.Description ?? "",
+                            });
+            foreach (var item in lists)
+            {
+                comments.Append($"{item.Name} = {item.Value}{(string.IsNullOrEmpty(item.Description) ? "" : $"({item.Description})")} | ");
+            }
+            return comments.ToString();
+        }
 
         #endregion
     }

diff --git a/src/LLMService.DataProvider.Relational/Entity/ChatFolder.cs b/src/LLMService.DataProvider.Relational/Entity/ChatFolder.cs
@@ -1,6 +1,4 @@
-using System.ComponentModel.DataAnnotations;
-
-namespace LLMService.DataProvider.Relational.Entity
+namespace LLMService.DataProvider.Relational.Entity
 {
     /// <summary>
     /// 

diff --git a/src/LLMService.DataProvider.Relational/Entity/ChatMessage.cs b/src/LLMService.DataProvider.Relational/Entity/ChatMessage.cs
@@ -1,11 +1,8 @@
-using System.ComponentModel;
-using System.ComponentModel.DataAnnotations;
-using static LLMService.DataProvider.Relational.Entity.EntityConsts;
+using static LLMService.DataProvider.Relational.Entity.EntityConsts;
 using static LLMService.Shared.Models.LLMApiDefaults;
 
 namespace LLMService.DataProvider.Relational.Entity
 {
-#nullable enable
     /// <summary>
     /// 
     /// </summary>
@@ -32,14 +29,14 @@ public class ChatMessage<TMessageKey, TConversationKey, TTenantKey, TUserKey>: I
         /// <value>
         /// The identifier.
         /// </value>
-        public TMessageKey Id {  get; set; }
+        public TMessageKey Id { get; set; }
         /// <summary>
         /// Gets or sets the conversation identifier.
         /// </summary>
         /// <value>
         /// The conversation identifier.
         /// </value>
-        public TConversationKey? ConversationId { get; set; }
+        public TConversationKey ConversationId { get; set; }
         /// <summary>
         /// Gets or sets the tenant identifier.
         /// </summary>
@@ -120,5 +117,4 @@ public class ChatMessage<TMessageKey, TConversationKey, TTenantKey, TUserKey>: I
 
 
     }
-#nullable disable
 }
diff --git a/src/LLMService.DataProvider.Relational/Entity/Conversation.cs b/src/LLMService.DataProvider.Relational/Entity/Conversation.cs
@@ -1,6 +1,4 @@
-using System.ComponentModel.DataAnnotations;
-
-namespace LLMService.DataProvider.Relational.Entity
+namespace LLMService.DataProvider.Relational.Entity
 {
     /// <summary>
     /// 

diff --git a/src/LLMService.DataProvider.Relational/Provider/ChatStorage.cs b/src/LLMService.DataProvider.Relational/Provider/ChatStorage.cs
@@ -1,10 +1,5 @@
 using LLMService.DataProvider.Relational.Entity;
 using LLMService.Shared.Models;
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
 
 namespace LLMService.DataProvider.Relational.Provider
 {

diff --git a/src/LLMService.DataProvider.Relational/Provider/IChatStorage.cs b/src/LLMService.DataProvider.Relational/Provider/IChatStorage.cs
@@ -1,9 +1,4 @@
 using LLMService.Shared.Models;
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
 
 namespace LLMService.DataProvider.Relational.Provider
 {