Changes

2026-05-14 14:12:29 +03:00
parent 92278ae375
commit 75bc9509c5
137 changed files with 0 additions and 371 deletions
@@ -0,0 +1,77 @@
+using Api.Data.Entities;
+using Microsoft.EntityFrameworkCore;
+
+namespace Api.Data;
+
+public sealed class RagDbContext : DbContext
+{
+    public const string SchemaName = "rag";
+    public const string MigrationTableName = "_Migrations";
+
+    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.HasDefaultSchema(SchemaName);
+
+        modelBuilder.Entity<RagDocumentEntity>(entity =>
+        {
+            entity.ToTable("Documents");
+            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("Chunks");
+            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("EmbeddingCache");
+            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("ChatCompletionCache");
+            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()");
+        });
+    }
+}