79 lines
2.7 KiB
C#
79 lines
2.7 KiB
C#
namespace Admin.NET.Core.Ai.Services.DataBase;
|
|
|
|
using Admin.NET.Core.Ai.Entitys;
|
|
using Admin.NET.Core.Ai.Enums;
|
|
using Admin.NET.Core.Ai.Services.DataBase.Dto;
|
|
|
|
/// <summary>
|
|
/// 异步对数据库进行操作
|
|
/// </summary>
|
|
public class ChatChannelActionService : IScoped
|
|
{
|
|
private readonly ILogger<ChatChannelActionService> _logger;
|
|
private readonly SqlSugarRepository<LLMChatSummaryHistory> _chatSummaryHistoryService;
|
|
private readonly SqlSugarRepository<LLMChatHistory> _chatHistoryService;
|
|
|
|
/// <summary>
|
|
/// 构造函数
|
|
/// </summary>
|
|
/// <param name="logger"></param>
|
|
/// <param name="chatSummaryHistoryService"></param>
|
|
/// <param name="chatHistoryService"></param>
|
|
public ChatChannelActionService(ILogger<ChatChannelActionService> logger,
|
|
SqlSugarRepository<LLMChatSummaryHistory> chatSummaryHistoryService,
|
|
SqlSugarRepository<LLMChatHistory> chatHistoryService)
|
|
{
|
|
_logger = logger;
|
|
_chatSummaryHistoryService = chatSummaryHistoryService;
|
|
_chatHistoryService = chatHistoryService;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 追加聊天记录
|
|
/// </summary>
|
|
public async Task AppendAsync(LLMChatSummaryHistory chatSummaryHistory)
|
|
{
|
|
await _chatSummaryHistoryService.Context.CopyNew()
|
|
.InsertNav(chatSummaryHistory)
|
|
.Include(z => z.Histories)
|
|
.ExecuteCommandAsync();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 追加聊天明细
|
|
/// </summary>
|
|
public async Task AppendItemAsync(LLMChatSummaryHistory chatSummaryHistory)
|
|
{
|
|
chatSummaryHistory.Histories.ForEach(item => item.SummaryId = chatSummaryHistory.Id);
|
|
await _chatHistoryService.Context.CopyNew()
|
|
.Insertable(chatSummaryHistory.Histories)
|
|
.ExecuteCommandAsync();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 删除所有聊天记录
|
|
/// </summary>
|
|
public async Task DeleteAllAsync(LLMChatSummaryHistory chatSummaryHistory)
|
|
{
|
|
await _chatSummaryHistoryService.Context.CopyNew()
|
|
.DeleteNav<LLMChatSummaryHistory>(u => u.Id == chatSummaryHistory.Id)
|
|
.Include(u => u.Histories)
|
|
.ExecuteCommandAsync();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 重命名摘要
|
|
/// </summary>
|
|
public async Task RenameSummaryAsync(LLMChatSummaryHistory chatSummaryHistory)
|
|
{
|
|
var oldChatSummaryHistory = await _chatSummaryHistoryService.CopyNew()
|
|
.AsQueryable()
|
|
.Where(u => u.Id == chatSummaryHistory.Id)
|
|
.FirstAsync();
|
|
oldChatSummaryHistory.Summary = chatSummaryHistory.Summary;
|
|
await _chatSummaryHistoryService.Context.CopyNew()
|
|
.Updateable(oldChatSummaryHistory)
|
|
.ExecuteCommandAsync();
|
|
}
|
|
}
|