| | | 1 | | using Elsa.AI.Persistence.EFCore.Entities; |
| | | 2 | | using Elsa.Persistence.EFCore; |
| | | 3 | | using Microsoft.Extensions.DependencyInjection; |
| | | 4 | | |
| | | 5 | | namespace Elsa.AI.Persistence.EFCore; |
| | | 6 | | |
| | | 7 | | public class AIDbContext : ElsaDbContextBase |
| | | 8 | | { |
| | 1 | 9 | | private static readonly IServiceProvider EmptyServiceProvider = new ServiceCollection().BuildServiceProvider(); |
| | | 10 | | |
| | 34 | 11 | | public AIDbContext(DbContextOptions<AIDbContext> options) : this(options, EmptyServiceProvider) |
| | | 12 | | { |
| | 34 | 13 | | } |
| | | 14 | | |
| | 38 | 15 | | public AIDbContext(DbContextOptions<AIDbContext> options, IServiceProvider serviceProvider) : base(options, serviceP |
| | | 16 | | { |
| | 38 | 17 | | } |
| | | 18 | | |
| | 58 | 19 | | public DbSet<AIConversationRecord> Conversations => Set<AIConversationRecord>(); |
| | 40 | 20 | | public DbSet<AIProposalRecord> Proposals => Set<AIProposalRecord>(); |
| | 10 | 21 | | public DbSet<AIAuditRecord> AuditRecords => Set<AIAuditRecord>(); |
| | | 22 | | |
| | | 23 | | protected override void OnModelCreating(ModelBuilder modelBuilder) |
| | | 24 | | { |
| | 2 | 25 | | modelBuilder.Entity<AIProposalRecord>(entity => |
| | 2 | 26 | | { |
| | 2 | 27 | | entity.ToTable("AIProposals"); |
| | 2 | 28 | | entity.HasKey(x => x.Id); |
| | 2 | 29 | | entity.HasIndex(x => new { x.TenantId, x.ConversationId }); |
| | 2 | 30 | | entity.HasIndex(x => x.Status); |
| | 2 | 31 | | entity.Property(x => x.WorkflowPayload).IsRequired(); |
| | 2 | 32 | | entity.Property(x => x.ValidationDiagnostics).IsRequired(); |
| | 2 | 33 | | entity.Property(x => x.Warnings).IsRequired(); |
| | 4 | 34 | | }); |
| | | 35 | | |
| | 2 | 36 | | modelBuilder.Entity<AIAuditRecord>(entity => |
| | 2 | 37 | | { |
| | 2 | 38 | | entity.ToTable("AIAuditRecords"); |
| | 2 | 39 | | entity.HasKey(x => x.Id); |
| | 2 | 40 | | entity.HasIndex(x => new { x.TenantId, x.ConversationId }); |
| | 2 | 41 | | entity.HasIndex(x => x.ActorId); |
| | 2 | 42 | | entity.HasIndex(x => new { x.TenantId, x.Timestamp }); |
| | 2 | 43 | | entity.HasIndex(x => x.ProposalId); |
| | 2 | 44 | | entity.HasIndex(x => x.ToolInvocationId); |
| | 2 | 45 | | entity.Property(x => x.Type).IsRequired(); |
| | 2 | 46 | | entity.Property(x => x.Data).IsRequired(); |
| | 4 | 47 | | }); |
| | | 48 | | |
| | 2 | 49 | | modelBuilder.Entity<AIConversationRecord>(entity => |
| | 2 | 50 | | { |
| | 2 | 51 | | entity.ToTable("AIConversations"); |
| | 2 | 52 | | entity.HasKey(x => x.Id); |
| | 2 | 53 | | entity.HasIndex(x => new { x.TenantId, x.UserId }); |
| | 2 | 54 | | entity.HasIndex(x => x.Status); |
| | 2 | 55 | | entity.HasIndex(x => x.RetentionExpiresAt); |
| | 2 | 56 | | entity.Property(x => x.UserId).IsRequired(); |
| | 2 | 57 | | entity.Property(x => x.Status).IsRequired(); |
| | 2 | 58 | | entity.Property(x => x.RetentionMode).IsRequired(); |
| | 2 | 59 | | entity.Property(x => x.Messages).IsRequired(); |
| | 4 | 60 | | }); |
| | | 61 | | |
| | 2 | 62 | | base.OnModelCreating(modelBuilder); |
| | 2 | 63 | | } |
| | | 64 | | } |