370 lines
15 KiB
C#
370 lines
15 KiB
C#
using HuanMeng.DotNetCore.Base;
|
|
using HuanMeng.MiaoYu.Code.Cache;
|
|
using HuanMeng.MiaoYu.Code.Chat.Claude;
|
|
using HuanMeng.MiaoYu.Code.Chat.Claude.Model;
|
|
using HuanMeng.MiaoYu.Code.Chat.Contract;
|
|
using HuanMeng.MiaoYu.Model.DbSqlServer.Db_MiaoYu;
|
|
using HuanMeng.MiaoYu.Model.Dto.Chat;
|
|
using HuanMeng.MiaoYu.Model.EnumModel.Chat;
|
|
using HuanMeng.Utility;
|
|
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
|
|
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace HuanMeng.MiaoYu.Code.Chat
|
|
{
|
|
/// <summary>
|
|
/// 聊天类
|
|
/// </summary>
|
|
public class ChatBLL : MiaoYuBase
|
|
{
|
|
public ChatBLL(IServiceProvider serviceProvider) : base(serviceProvider)
|
|
{
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取用户和角色的聊天记录
|
|
/// </summary>
|
|
/// <param name="characterId"></param>
|
|
/// <returns></returns>
|
|
/// <exception cref="ArgumentException"></exception>
|
|
public async Task<BaseResponse<List<ChatMessageDto>>> GetChatMessage(int characterId)
|
|
{
|
|
|
|
var charact = MiaoYuCache.CharacterList.FirstOrDefault(it => it.Id == characterId);
|
|
if (charact == null)
|
|
{
|
|
throw new ArgumentException("角色不存在");
|
|
}
|
|
var list = new List<ChatMessageDto>();
|
|
ChatMessageDto chatMessageDto2 = new ChatMessageDto()
|
|
{
|
|
ClaudeType = ChatMessageType.text.ToString(),
|
|
Content = charact.Biography ?? "",
|
|
Role = ChatRole.profile.ToString(),
|
|
Timestamp = DateTime.Now,
|
|
UserIcon = "",
|
|
Id = 0
|
|
};
|
|
list.Add(chatMessageDto2);
|
|
if (_UserId == 0)
|
|
{
|
|
|
|
|
|
var chat = new ChatMessageDto()
|
|
{
|
|
Id = 0,
|
|
Role = ChatRole.assistant.ToString(),
|
|
ClaudeType = "text",
|
|
Content = charact.Prologue,
|
|
Timestamp = DateTime.Now,
|
|
UserIcon = charact.IconImage
|
|
};
|
|
|
|
list.Add(chat);
|
|
return new BaseResponse<List<ChatMessageDto>>(ResonseCode.Success, "", list);
|
|
}
|
|
//if(timeStamp.)
|
|
var userChatSession = await Dao.daoDbMiaoYu.context.T_User_Chat.Where(it => it.CharacterId == characterId && it.UserId == _UserId && !it.IsDelete).FirstOrDefaultAsync();
|
|
if (userChatSession == null)
|
|
{
|
|
userChatSession = new T_User_Chat()
|
|
{
|
|
SessionId = Guid.NewGuid(),
|
|
CharacterId = characterId,
|
|
CreateAt = DateTime.Now,
|
|
IsDelete = false,
|
|
SessionName = "新会话",
|
|
UpdateAt = DateTime.Now,
|
|
ModelConfigId = charact.ModelConfigId,
|
|
TenantId = charact.TenantId,
|
|
UserId = _UserId,
|
|
};
|
|
Dao.daoDbMiaoYu.context.T_User_Chat.Add(userChatSession);
|
|
Dao.daoDbMiaoYu.context.SaveChanges();
|
|
}
|
|
var chatList = await Dao.daoDbMiaoYu.context.T_Chat.Where(it => it.SessionId == userChatSession.SessionId && it.UserId == _UserId && it.Type == (int)ChatType.正常).OrderByDescending(it => it.SendMessageDay).ToListAsync();
|
|
if (chatList == null || chatList.Count == 0)
|
|
{
|
|
//添加第一句
|
|
var chat = new T_Chat()
|
|
{
|
|
Role = ChatRole.initialization.ToString(),
|
|
CharacterId = characterId,
|
|
ClaudeId = "",
|
|
ClaudeType = "text",
|
|
CreateTime = DateTime.Now,
|
|
Content = charact.Prologue,
|
|
Input_tokens = 0,
|
|
Output_tokens = 0,
|
|
SendDateDay = int.Parse(DateTime.Now.ToString("yyyyMMdd")),
|
|
TimeStamp = DateTime.Now,//DateTimeExtensions.ToUnixTimestamp(),.ToUnixTimestamp()
|
|
SendMessageDay = DateTime.Now.ToUnixTimestamp(),
|
|
SessionId = userChatSession.SessionId,
|
|
TenantId = userChatSession.TenantId,
|
|
Type = 0,
|
|
UpdateTime = DateTime.Now,
|
|
UserId = userChatSession.UserId,
|
|
ClaudeModel = ""
|
|
};
|
|
Dao.daoDbMiaoYu.context.T_Chat.Add(chat);
|
|
Dao.daoDbMiaoYu.context.SaveChanges();
|
|
chatList = new List<T_Chat>();
|
|
chatList.Add(chat);
|
|
}
|
|
var message = Mapper.Map<List<ChatMessageDto>>(chatList);
|
|
//设置头像
|
|
message.Where(it =>
|
|
it.Role == ChatRole.assistant.ToString() ||
|
|
it.Role == ChatRole.initialization.ToString()
|
|
).ToList()
|
|
.ForEach(it => it.UserIcon = charact.IconImage);
|
|
//用户头像
|
|
var userData = await Dao.daoDbMiaoYu.context.T_User_Data.FirstOrDefaultAsync(it => it.UserId == _UserId);
|
|
var userIconUrl = userData?.UserIconUrl ?? "";
|
|
message.Where(it => it.Role == ChatRole.user.ToString()).ToList().ForEach(it => it.UserIcon = userIconUrl);
|
|
|
|
//将第一句修改角色
|
|
var initia = message.FirstOrDefault(it => it.Role == ChatRole.initialization.ToString());
|
|
if (initia != null)
|
|
{
|
|
initia.Role = ChatRole.assistant.ToString();
|
|
}
|
|
message.Insert(0, chatMessageDto2);
|
|
return new BaseResponse<List<ChatMessageDto>>(ResonseCode.Success, "", message);
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// 发送消息接口
|
|
/// </summary>
|
|
/// <param name="characterId">角色Id</param>
|
|
/// <param name="message">消息内容</param>
|
|
/// <returns></returns>
|
|
/// <exception cref="ArgumentException"></exception>
|
|
public async Task<BaseResponse<ChatMessageDataDto>> Message(int characterId, string message)
|
|
{
|
|
ChatMessageDataDto chatListDto = new ChatMessageDataDto();
|
|
List<ChatMessageDto> chatMessageDtos = new List<ChatMessageDto>();
|
|
var charact = MiaoYuCache.CharacterList.FirstOrDefault(it => it.Id == characterId);
|
|
if (charact == null)
|
|
{
|
|
throw new ArgumentException("角色不存在");
|
|
}
|
|
|
|
if (_UserId == 0)
|
|
{
|
|
|
|
var chatMessage = new ChatMessageDto()
|
|
{
|
|
Id = 0,
|
|
Role = ChatRole.assistant.ToString(),
|
|
ClaudeType = "text",
|
|
Content = charact.Prologue,
|
|
Timestamp = DateTime.Now,
|
|
UserIcon = charact.IconImage
|
|
};
|
|
chatMessageDtos.Add(chatMessage);
|
|
|
|
chatListDto.ChatList = chatMessageDtos;
|
|
return new BaseResponse<ChatMessageDataDto>(ResonseCode.Success, "", chatListDto);
|
|
}
|
|
//if(timeStamp.)
|
|
var userChatSession = await Dao.daoDbMiaoYu.context.T_User_Chat.Where(it => it.CharacterId == characterId && it.UserId == _UserId && !it.IsDelete).FirstOrDefaultAsync();
|
|
if (userChatSession == null)
|
|
{
|
|
userChatSession = new T_User_Chat()
|
|
{
|
|
SessionId = Guid.NewGuid(),
|
|
CharacterId = characterId,
|
|
CreateAt = DateTime.Now,
|
|
IsDelete = false,
|
|
SessionName = "新会话",
|
|
UpdateAt = DateTime.Now,
|
|
ModelConfigId = charact.ModelConfigId,
|
|
TenantId = charact.TenantId,
|
|
UserId = _UserId,
|
|
};
|
|
Dao.daoDbMiaoYu.context.T_User_Chat.Add(userChatSession);
|
|
Dao.daoDbMiaoYu.context.SaveChanges();
|
|
}
|
|
var role = new List<string>();
|
|
role.Add(ChatRole.assistant.ToString());
|
|
role.Add(ChatRole.user.ToString());
|
|
|
|
var chatList = await Dao.daoDbMiaoYu.context.T_Chat.Where(it => it.SessionId == userChatSession.SessionId && it.UserId == _UserId && role.Contains(it.Role)).OrderBy(it => it.SendMessageDay).ToListAsync();
|
|
if (chatList == null)
|
|
{
|
|
chatList = new List<T_Chat>();
|
|
}
|
|
List<ClaudeChatMessage> mess = new List<ClaudeChatMessage>();
|
|
for (int i = 0; i < chatList.Count; i++)
|
|
{
|
|
mess.Add(new ClaudeChatMessage
|
|
{
|
|
Role = chatList[i].Role,
|
|
Content = chatList[i].Content,
|
|
});
|
|
}
|
|
mess.Add(new ClaudeChatMessage
|
|
{
|
|
Role = ChatRole.user.ToString(),
|
|
Content = message,
|
|
});
|
|
#region 添加数据库
|
|
T_Chat t_Chat1 = new T_Chat
|
|
{
|
|
CharacterId = charact.Id,
|
|
ClaudeModel = "",
|
|
ClaudeId = "",
|
|
ClaudeType = "text",
|
|
Content = message,
|
|
CreateTime = DateTime.Now,
|
|
Input_tokens = 0,
|
|
Output_tokens = 0,
|
|
Role = ChatRole.user.ToString(),
|
|
SendDateDay = int.Parse(DateTime.Now.ToString("yyyyMMdd")),
|
|
SendMessageDay = DateTime.Now.ToUnixTimestamp(),
|
|
SessionId = userChatSession.SessionId,
|
|
TenantId = userChatSession.TenantId,
|
|
TimeStamp = DateTime.Now,
|
|
Type = 0,
|
|
UpdateTime = DateTime.Now,
|
|
UserId = _UserId,
|
|
};
|
|
|
|
|
|
#endregion
|
|
#region 调用api
|
|
ClaudeChatChatParams baseChatParams = new ClaudeChatChatParams();
|
|
baseChatParams.Messages = mess.ToArray();
|
|
baseChatParams.System = charact.System ?? "";
|
|
baseChatParams.MaxTokens = charact.ModelConfig.MaxTokens;
|
|
var claude = charact.ModelConfig.GetClaudeChatConfig();
|
|
IChat chat = new ClaudeChat(claude, HttpClientFactory);
|
|
var response = await chat.MessagesAsync(baseChatParams);
|
|
var claudeChatResponse = response as ClaudeChatResponse;
|
|
#endregion
|
|
T_Chat t_Chat = new T_Chat
|
|
{
|
|
CharacterId = charact.Id,
|
|
ClaudeModel = claudeChatResponse.Model,
|
|
ClaudeId = claudeChatResponse.Id,
|
|
ClaudeType = claudeChatResponse.Content[0]?.Type,
|
|
Content = claudeChatResponse.Content[0]?.Text,
|
|
CreateTime = DateTime.Now,
|
|
Input_tokens = claudeChatResponse.Usage.InputTokens,
|
|
Output_tokens = claudeChatResponse.Usage.OutputTokens,
|
|
Role = claudeChatResponse.Role,
|
|
SendDateDay = int.Parse(DateTime.Now.ToString("yyyyMMdd")),
|
|
SendMessageDay = DateTime.Now.ToUnixTimestamp(),
|
|
SessionId = userChatSession.SessionId,
|
|
TenantId = userChatSession.TenantId,
|
|
TimeStamp = DateTime.Now,
|
|
Type = 0,
|
|
UpdateTime = DateTime.Now,
|
|
UserId = _UserId,
|
|
};
|
|
Dao.daoDbMiaoYu.context.Add(t_Chat1);
|
|
Dao.daoDbMiaoYu.context.Add(t_Chat);
|
|
userChatSession.UpdateAt = DateTime.Now;
|
|
await Dao.daoDbMiaoYu.context.SaveChangesAsync();
|
|
//claudeChatResponse.
|
|
ChatMessageDto chatMessageDto = new ChatMessageDto()
|
|
{
|
|
ClaudeType = ChatMessageType.text.ToString(),
|
|
Content = claudeChatResponse.Content[0].Text,
|
|
Role = ChatRole.assistant.ToString(),
|
|
Timestamp = DateTime.Now,
|
|
UserIcon = charact.IconImage,
|
|
Id = t_Chat.Id
|
|
};
|
|
chatMessageDtos.Add(chatMessageDto);
|
|
ChatMessageDto chatMessageDto1 = new ChatMessageDto()
|
|
{
|
|
ClaudeType = ChatMessageType.text.ToString(),
|
|
Content = "您还剩余1次聊天次数",
|
|
Role = ChatRole.tips.ToString(),
|
|
Timestamp = DateTime.Now,
|
|
UserIcon = "",
|
|
Id = t_Chat.Id
|
|
};
|
|
chatMessageDtos.Add(chatMessageDto1);
|
|
chatListDto.ChatList = chatMessageDtos;
|
|
chatListDto.LastMessageId = t_Chat1.Id;
|
|
chatListDto.RemainingChatCount = 1;
|
|
return new BaseResponse<ChatMessageDataDto>(ResonseCode.Success, "", chatListDto);
|
|
}
|
|
|
|
/// <summary>
|
|
/// 删除聊天记录
|
|
/// </summary>
|
|
/// <param name="id">聊天id</param>
|
|
/// <param name="characterId"></param>
|
|
/// <returns></returns>
|
|
public async Task<bool> DelChatByIds(List<int> id, int characterId)
|
|
{
|
|
var chatsToDelete = await Dao.daoDbMiaoYu.context.T_Chat.Where(t => id.Contains(t.Id)).ToListAsync();
|
|
if (chatsToDelete.Count <= 0)
|
|
{
|
|
throw new Exception("删除失败");
|
|
}
|
|
var chatList = chatsToDelete.Where(t => t.UserId == _UserId && t.CharacterId == characterId && (t.Type == (int)ChatType.正常 || t.Type == (int)ChatType.重新生成)).ToList();
|
|
if (chatList.Count > 0)
|
|
{
|
|
chatList.ForEach(t => t.Type = 2);
|
|
}
|
|
Dao.daoDbMiaoYu.context.SaveChanges();
|
|
return true;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 清空聊天记录
|
|
/// </summary>
|
|
/// <param name="characterId"></param>
|
|
/// <returns></returns>
|
|
public async Task<bool> DelChat(int characterId)
|
|
{
|
|
var tempList = await Dao.daoDbMiaoYu.context.T_User_Chat.Where(t => t.UserId == _UserId && t.CharacterId == characterId && t.IsDelete == false).FirstOrDefaultAsync();
|
|
if (tempList != null)
|
|
{
|
|
tempList.IsDelete = true;
|
|
}
|
|
Dao.daoDbMiaoYu.context.SaveChanges();
|
|
return true;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取消息聊天记录列表
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public async Task<BaseResponse<List<ChatHistoryInfo>>> GetChatHistoryList()
|
|
{
|
|
var userChatSessions = await Dao.daoDbMiaoYu.context.T_User_Chat.Where(it => it.UserId == _UserId && !it.IsDelete).ToListAsync();
|
|
List<ChatHistoryInfo> chatHistoryInfos = new List<ChatHistoryInfo>();
|
|
var charactersIds = MiaoYuCache.CharacterList.ToList();
|
|
userChatSessions.ForEach(it =>
|
|
{
|
|
var model = charactersIds.FirstOrDefault(item => item.Id == it.CharacterId);
|
|
if (model != null)
|
|
{
|
|
var info = Mapper.Map<ChatHistoryInfo>(model);
|
|
info.LastContactTime = it.UpdateAt;
|
|
chatHistoryInfos.Add(info);
|
|
}
|
|
|
|
});
|
|
chatHistoryInfos = chatHistoryInfos.OrderByDescending(it => it.LastContactTime).ToList();
|
|
return new BaseResponse<List<ChatHistoryInfo>>(ResonseCode.Success, "", chatHistoryInfos) { };
|
|
}
|
|
}
|
|
}
|