Changes

rag-api/Data/Entities/RagChatCompletionCacheEntity.cs

@@ -0,0 +1,10 @@
+namespace Api.Data.Entities;
+
+public sealed class RagChatCompletionCacheEntity
+{
+    public string CacheKey { get; set; } = string.Empty;
+    public string Model { get; set; } = string.Empty;
+    public decimal Temperature { get; set; }
+    public string ResponseText { get; set; } = string.Empty;
+    public DateTime CreatedAt { get; set; } = DateTime.UtcNow;
+}

rag-api/Data/Entities/RagChunkEntity.cs

@@ -0,0 +1,12 @@
+namespace Api.Data.Entities;
+
+public sealed class RagChunkEntity
+{
+    public string Id { get; set; } = string.Empty;
+    public string DocumentId { get; set; } = string.Empty;
+    public int ChunkIndex { get; set; }
+    public string Text { get; set; } = string.Empty;
+    public byte[] Embedding { get; set; } = [];
+
+    public RagDocumentEntity? Document { get; set; }
+}

rag-api/Data/Entities/RagDocumentEntity.cs

@@ -0,0 +1,16 @@
+namespace Api.Data.Entities;
+
+public sealed class RagDocumentEntity
+{
+    public string Id { get; set; } = string.Empty;
+    public string DocumentType { get; set; } = string.Empty;
+    public string Title { get; set; } = string.Empty;
+    public string? SourceUrl { get; set; }
+    public string RawText { get; set; } = string.Empty;
+    public string TextHash { get; set; } = string.Empty;
+    public double TypeConfidence { get; set; }
+    public string MetadataJson { get; set; } = "{}";
+    public DateTime CreatedAt { get; set; } = DateTime.UtcNow;
+
+    public ICollection<RagChunkEntity> Chunks { get; set; } = [];
+}

rag-api/Data/Entities/RagEmbeddingCacheEntity.cs

@@ -0,0 +1,10 @@
+namespace Api.Data.Entities;
+
+public sealed class RagEmbeddingCacheEntity
+{
+    public string CacheKey { get; set; } = string.Empty;
+    public string Model { get; set; } = string.Empty;
+    public string TextHash { get; set; } = string.Empty;
+    public byte[] Vector { get; set; } = [];
+    public DateTime CreatedAt { get; set; } = DateTime.UtcNow;
+}

rag-api/Data/RagDbContext.cs

@@ -0,0 +1,72 @@
+using Api.Data.Entities;
+using Microsoft.EntityFrameworkCore;
+
+namespace Api.Data;
+
+public sealed class RagDbContext : DbContext
+{
+    public RagDbContext(DbContextOptions<RagDbContext> options) : base(options)
+    {
+    }
+
+    public DbSet<RagDocumentEntity> RagDocuments => Set<RagDocumentEntity>();
+    public DbSet<RagChunkEntity> RagChunks => Set<RagChunkEntity>();
+    public DbSet<RagEmbeddingCacheEntity> RagEmbeddingCache => Set<RagEmbeddingCacheEntity>();
+    public DbSet<RagChatCompletionCacheEntity> RagChatCompletionCache => Set<RagChatCompletionCacheEntity>();
+
+    protected override void OnModelCreating(ModelBuilder modelBuilder)
+    {
+        modelBuilder.Entity<RagDocumentEntity>(entity =>
+        {
+            entity.ToTable("RagDocuments");
+            entity.HasKey(x => x.Id);
+            entity.Property(x => x.Id).HasMaxLength(64);
+            entity.Property(x => x.DocumentType).HasMaxLength(80).IsRequired();
+            entity.Property(x => x.Title).HasMaxLength(300).IsRequired();
+            entity.Property(x => x.SourceUrl).HasMaxLength(1200);
+            entity.Property(x => x.RawText).IsRequired();
+            entity.Property(x => x.TextHash).HasMaxLength(64).IsRequired();
+            entity.Property(x => x.MetadataJson).HasDefaultValue("{}").IsRequired();
+            entity.Property(x => x.CreatedAt).HasDefaultValueSql("SYSUTCDATETIME()");
+            entity.HasIndex(x => x.TextHash);
+            entity.HasIndex(x => x.DocumentType);
+        });
+
+        modelBuilder.Entity<RagChunkEntity>(entity =>
+        {
+            entity.ToTable("RagChunks");
+            entity.HasKey(x => x.Id);
+            entity.Property(x => x.Id).HasMaxLength(64);
+            entity.Property(x => x.DocumentId).HasMaxLength(64).IsRequired();
+            entity.Property(x => x.Text).IsRequired();
+            entity.Property(x => x.Embedding).IsRequired();
+            entity.HasOne(x => x.Document)
+                .WithMany(x => x.Chunks)
+                .HasForeignKey(x => x.DocumentId)
+                .OnDelete(DeleteBehavior.Cascade);
+        });
+
+        modelBuilder.Entity<RagEmbeddingCacheEntity>(entity =>
+        {
+            entity.ToTable("RagEmbeddingCache");
+            entity.HasKey(x => x.CacheKey);
+            entity.Property(x => x.CacheKey).HasMaxLength(64);
+            entity.Property(x => x.Model).HasMaxLength(120).IsRequired();
+            entity.Property(x => x.TextHash).HasMaxLength(64).IsRequired();
+            entity.Property(x => x.Vector).IsRequired();
+            entity.Property(x => x.CreatedAt).HasDefaultValueSql("SYSUTCDATETIME()");
+            entity.HasIndex(x => x.TextHash);
+        });
+
+        modelBuilder.Entity<RagChatCompletionCacheEntity>(entity =>
+        {
+            entity.ToTable("RagChatCompletionCache");
+            entity.HasKey(x => x.CacheKey);
+            entity.Property(x => x.CacheKey).HasMaxLength(64);
+            entity.Property(x => x.Model).HasMaxLength(120).IsRequired();
+            entity.Property(x => x.Temperature).HasColumnType("decimal(4,2)");
+            entity.Property(x => x.ResponseText).IsRequired();
+            entity.Property(x => x.CreatedAt).HasDefaultValueSql("SYSUTCDATETIME()");
+        });
+    }
+}