Browse Source
feat(ai): Automatically generate a conversation name for the first dialogue.pull/1467/head
yx lin
2 months ago
committed by
GitHub
GitHub
15 changed files with 290 additions and 53 deletions
@ -0,0 +1,11 @@ |
|||||
|
using System; |
||||
|
using Volo.Abp.Domain.Entities.Events.Distributed; |
||||
|
using Volo.Abp.MultiTenancy; |
||||
|
|
||||
|
namespace LINGYUN.Abp.AIManagement.Chats; |
||||
|
public abstract class ChatMessageRecordEto : EntityEto<Guid>, IMultiTenant |
||||
|
{ |
||||
|
public Guid? TenantId { get; set; } |
||||
|
public string Workspace { get; set; } |
||||
|
public Guid? ConversationId { get; set; } |
||||
|
} |
||||
@ -0,0 +1,4 @@ |
|||||
|
namespace LINGYUN.Abp.AIManagement.Chats; |
||||
|
public class TextChatMessageRecordEto : ChatMessageRecordEto |
||||
|
{ |
||||
|
} |
||||
@ -0,0 +1,127 @@ |
|||||
|
using LINGYUN.Abp.AI; |
||||
|
using LINGYUN.Abp.AIManagement.Localization; |
||||
|
using LINGYUN.Abp.AIManagement.Tokens; |
||||
|
using Microsoft.Extensions.AI; |
||||
|
using Microsoft.Extensions.Localization; |
||||
|
using System.Globalization; |
||||
|
using System.Threading.Tasks; |
||||
|
using Volo.Abp.Data; |
||||
|
using Volo.Abp.DependencyInjection; |
||||
|
using Volo.Abp.DistributedLocking; |
||||
|
using Volo.Abp.Domain.Entities.Events.Distributed; |
||||
|
using Volo.Abp.EventBus.Distributed; |
||||
|
using Volo.Abp.Guids; |
||||
|
using Volo.Abp.Localization; |
||||
|
using Volo.Abp.Specifications; |
||||
|
using Volo.Abp.Uow; |
||||
|
|
||||
|
namespace LINGYUN.Abp.AIManagement.Chats; |
||||
|
public class ConversationChangeNameHandler : |
||||
|
IDistributedEventHandler<EntityCreatedEto<TextChatMessageRecordEto>>, |
||||
|
ITransientDependency |
||||
|
{ |
||||
|
private readonly IGuidGenerator _guidGenerator; |
||||
|
private readonly IAbpDistributedLock _distributedLock; |
||||
|
private readonly IChatClientFactory _chatClientFactory; |
||||
|
private readonly IStringLocalizer<AIManagementResource> _stringLocalizer; |
||||
|
private readonly ITokenUsageRecordRepository _tokenUsageRecordRepository; |
||||
|
private readonly IConversationRecordRepository _conversationRecordRepository; |
||||
|
private readonly ITextChatMessageRecordRepository _textChatMessageRecordRepository; |
||||
|
|
||||
|
public ConversationChangeNameHandler( |
||||
|
IGuidGenerator guidGenerator, |
||||
|
IAbpDistributedLock distributedLock, |
||||
|
IChatClientFactory chatClientFactory, |
||||
|
IStringLocalizer<AIManagementResource> stringLocalizer, |
||||
|
ITokenUsageRecordRepository tokenUsageRecordRepository, |
||||
|
IConversationRecordRepository conversationRecordRepository, |
||||
|
ITextChatMessageRecordRepository textChatMessageRecordRepository) |
||||
|
{ |
||||
|
_guidGenerator = guidGenerator; |
||||
|
_distributedLock = distributedLock; |
||||
|
_chatClientFactory = chatClientFactory; |
||||
|
_stringLocalizer = stringLocalizer; |
||||
|
_tokenUsageRecordRepository = tokenUsageRecordRepository; |
||||
|
_conversationRecordRepository = conversationRecordRepository; |
||||
|
_textChatMessageRecordRepository = textChatMessageRecordRepository; |
||||
|
} |
||||
|
|
||||
|
[UnitOfWork] |
||||
|
public async virtual Task HandleEventAsync(EntityCreatedEto<TextChatMessageRecordEto> eventData) |
||||
|
{ |
||||
|
/* |
||||
|
* 业务逻辑: 当用户第一次向智能体发送消息时,系统根据用户消息创建一个额外的智能体 |
||||
|
* 通过消息摘要生成一个简短有效的问题描述,用作本次对话的名称, 设计灵感: Deepseek |
||||
|
*/ |
||||
|
|
||||
|
if (!eventData.Entity.ConversationId.HasValue) |
||||
|
{ |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
await using var lockHandle = await _distributedLock.TryAcquireAsync($"{nameof(ConversationChangeNameHandler)}_{eventData.Entity.ConversationId}_DesignConversationName"); |
||||
|
|
||||
|
if (lockHandle == null) |
||||
|
{ |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
var conversation = await _conversationRecordRepository.FindAsync(eventData.Entity.ConversationId.Value); |
||||
|
if (conversation == null) |
||||
|
{ |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
var specification = new ExpressionSpecification<TextChatMessageRecord>( |
||||
|
x => x.ConversationId == eventData.Entity.ConversationId && x.Role == ChatRole.User); |
||||
|
|
||||
|
var historyChatMessages = await _textChatMessageRecordRepository.GetListAsync( |
||||
|
specification: specification, |
||||
|
sorting: nameof(TextChatMessageRecord.CreationTime), |
||||
|
maxResultCount: 2); |
||||
|
|
||||
|
if (historyChatMessages.Count > 1) |
||||
|
{ |
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
var chatMessage = historyChatMessages[0]; |
||||
|
var currentCulture = chatMessage.GetProperty(nameof(CultureInfo.CurrentCulture), CultureInfo.CurrentCulture.Name); |
||||
|
|
||||
|
using (CultureHelper.Use(currentCulture!)) |
||||
|
{ |
||||
|
var chatClient = await _chatClientFactory.CreateAsync(chatMessage.Workspace); |
||||
|
|
||||
|
var aiAgent = chatClient |
||||
|
.AsBuilder() |
||||
|
.ConfigureOptions(options => |
||||
|
{ |
||||
|
// 不受工具影响
|
||||
|
options.Tools = []; |
||||
|
}) |
||||
|
.BuildAIAgent(_stringLocalizer["DesignConversationNamePrompt"].Value); |
||||
|
|
||||
|
var agentRunRes = await aiAgent.RunAsync(chatMessage.Content); |
||||
|
|
||||
|
conversation.SetName(agentRunRes.Text); |
||||
|
|
||||
|
await _conversationRecordRepository.UpdateAsync(conversation); |
||||
|
|
||||
|
if (agentRunRes.Usage != null) |
||||
|
{ |
||||
|
var tokenUsageRecord = new TokenUsageRecord( |
||||
|
_guidGenerator.Create(), |
||||
|
chatMessage.Id, |
||||
|
conversation.Id, |
||||
|
agentRunRes.Usage.InputTokenCount, |
||||
|
agentRunRes.Usage.OutputTokenCount, |
||||
|
agentRunRes.Usage.TotalTokenCount, |
||||
|
agentRunRes.Usage.CachedInputTokenCount, |
||||
|
agentRunRes.Usage.ReasoningTokenCount, |
||||
|
chatMessage.TenantId); |
||||
|
|
||||
|
await _tokenUsageRecordRepository.InsertAsync(tokenUsageRecord); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue