You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
1.0 KiB
32 lines
1.0 KiB
using LINGYUN.Abp.AIManagement.Chats;
|
|
using LINGYUN.Abp.AIManagement.EntityFrameworkCore;
|
|
using LINGYUN.Abp.AIManagement.Tokens;
|
|
using LINGYUN.Abp.AIManagement.Workspaces;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Volo.Abp.Data;
|
|
using Volo.Abp.EntityFrameworkCore;
|
|
|
|
namespace LINGYUN.Abp.MicroService.AIService;
|
|
|
|
[ConnectionStringName("Default")]
|
|
public class AIServiceMigrationsDbContext :
|
|
AbpDbContext<AIServiceMigrationsDbContext>,
|
|
IAIManagementDbContext
|
|
{
|
|
public DbSet<WorkspaceDefinitionRecord> WorkspaceDefinitions { get; set; }
|
|
public DbSet<TextChatMessageRecord> TextChatMessageRecords { get; set; }
|
|
public DbSet<ConversationRecord> ConversationRecords { get; set; }
|
|
public DbSet<TokenUsageRecord> TokenUsageRecords { get; set; }
|
|
|
|
public AIServiceMigrationsDbContext(
|
|
DbContextOptions<AIServiceMigrationsDbContext> options) : base(options)
|
|
{
|
|
}
|
|
|
|
protected override void OnModelCreating(ModelBuilder builder)
|
|
{
|
|
base.OnModelCreating(builder);
|
|
|
|
builder.ConfigureAIManagement();
|
|
}
|
|
}
|
|
|