72 lines
2.5 KiB
C#
72 lines
2.5 KiB
C#
using Mapster;
|
|
using SqlSugar;
|
|
using Volo.Abp.Domain.Entities.Auditing;
|
|
using Yi.Framework.AiHub.Domain.Entities.ValueObjects;
|
|
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
|
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
|
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
|
|
|
namespace Yi.Framework.AiHub.Domain.Entities.Chat;
|
|
|
|
[SugarTable("Ai_Message")]
|
|
[SugarIndex($"index_{{table}}_{nameof(UserId)}_{nameof(SessionId)}",
|
|
$"{nameof(UserId)}", OrderByType.Desc,
|
|
$"{nameof(SessionId)}", OrderByType.Desc
|
|
)]
|
|
public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
|
|
{
|
|
public MessageAggregateRoot()
|
|
{
|
|
}
|
|
|
|
public MessageAggregateRoot(Guid? userId, Guid? sessionId, string content, string role, string modelId,
|
|
ThorUsageResponse? tokenUsage)
|
|
{
|
|
UserId = userId;
|
|
SessionId = sessionId;
|
|
//如果没有会话，不存储对话内容
|
|
Content = sessionId is null ? null : content;
|
|
Role = role;
|
|
ModelId = modelId;
|
|
if (tokenUsage is not null)
|
|
{
|
|
long inputTokenCount =
|
|
(tokenUsage.PromptTokens.HasValue && tokenUsage.PromptTokens.Value != 0)
|
|
? tokenUsage.PromptTokens.Value
|
|
: (tokenUsage.InputTokens.HasValue && tokenUsage.InputTokens.Value != 0)
|
|
? tokenUsage.InputTokens.Value
|
|
: 0;
|
|
|
|
long outputTokenCount =
|
|
(tokenUsage.CompletionTokens.HasValue && tokenUsage.CompletionTokens.Value != 0)
|
|
? tokenUsage.CompletionTokens.Value
|
|
: (tokenUsage.OutputTokens.HasValue && tokenUsage.OutputTokens.Value != 0)
|
|
? tokenUsage.OutputTokens.Value
|
|
: 0;
|
|
|
|
|
|
this.TokenUsage = new TokenUsageValueObject
|
|
{
|
|
OutputTokenCount = outputTokenCount,
|
|
InputTokenCount = inputTokenCount,
|
|
TotalTokenCount = tokenUsage.TotalTokens ?? 0
|
|
};
|
|
}
|
|
|
|
this.MessageType = sessionId is null ? MessageTypeEnum.Api : MessageTypeEnum.Web;
|
|
}
|
|
|
|
public Guid? UserId { get; set; }
|
|
public Guid? SessionId { get; set; }
|
|
|
|
[SugarColumn(ColumnDataType = StaticConfig.CodeFirst_BigString)]
|
|
public string? Content { get; set; }
|
|
|
|
public string Role { get; set; }
|
|
public string ModelId { get; set; }
|
|
public string? Remark { get; set; }
|
|
|
|
[SugarColumn(IsOwnsOne = true)] public TokenUsageValueObject TokenUsage { get; set; } = new TokenUsageValueObject();
|
|
|
|
public MessageTypeEnum MessageType { get; set; }
|
|
} |