From 34880b012214800b9772b3c1c7bef587e0732a60 Mon Sep 17 00:00:00 2001 From: zpc Date: Tue, 3 Sep 2024 23:53:20 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/0-core/HuanMeng.MiaoYu.Code/Chat/ChatBLL.cs | 7 ++++++- src/0-core/HuanMeng.MiaoYu.Code/Users/UserBLL.cs | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/src/0-core/HuanMeng.MiaoYu.Code/Chat/ChatBLL.cs b/src/0-core/HuanMeng.MiaoYu.Code/Chat/ChatBLL.cs index be06bae..ddbdb31 100644 --- a/src/0-core/HuanMeng.MiaoYu.Code/Chat/ChatBLL.cs +++ b/src/0-core/HuanMeng.MiaoYu.Code/Chat/ChatBLL.cs @@ -66,6 +66,7 @@ public class ChatBLL : MiaoYuBase ModelConfigId = charact.ModelConfigId, TenantId = charact.TenantId, UserId = _UserId, + TotalToken = 0, }; Dao.daoDbMiaoYu.context.T_User_Chat.Add(userChatSession); Dao.daoDbMiaoYu.context.SaveChanges(); @@ -327,9 +328,13 @@ public class ChatBLL : MiaoYuBase t_Character.Token = charact.Token; } } - t_Chat1.Tokens = claudeChatResponse.InputTokens - charact.Token; + t_Chat1.Tokens = claudeChatResponse.InputTokens ; t_Chat1.Input_tokens = claudeChatResponse.InputTokens; //设置消耗的总token + if (userChatSession.TotalToken == null) + { + userChatSession.TotalToken = 0; + } userChatSession.TotalToken += claudeChatResponse.InputTokens + claudeChatResponse.OutputTokens; Dao.daoDbMiaoYu.context.Add(t_Chat1); Dao.daoDbMiaoYu.context.Add(t_Chat); diff --git a/src/0-core/HuanMeng.MiaoYu.Code/Users/UserBLL.cs b/src/0-core/HuanMeng.MiaoYu.Code/Users/UserBLL.cs index 9953aae..4c75ea7 100644 --- a/src/0-core/HuanMeng.MiaoYu.Code/Users/UserBLL.cs +++ b/src/0-core/HuanMeng.MiaoYu.Code/Users/UserBLL.cs @@ -127,7 +127,7 @@ namespace HuanMeng.MiaoYu.Code.Users var memoryCard = Dao.daoDbMiaoYu.context.T_User_MemoryCard.Where(it => it.UserId == _UserId && it.CharacterId == 0).Count(); //var memoryCard = user.GetUserCurrencyMoney(UserCurrencyType.记忆卡, Dao); //获取聊天次数 - var hasTalked = Dao.daoDbMiaoYu.context.T_User_Chat.Where(it => it.UserId == _UserId && !it.IsDelete && it.TotalToken > 0).Count(); + var hasTalked = Dao.daoDbMiaoYu.context.T_User_Chat.Where(it => it.UserId == _UserId && !it.IsDelete ).Count(); List characters = new List(); return new BaseResponse(ResonseCode.Success, "请求成功", new ResponseUserInfo {