From 9ca3cd0b1a99d93dd7b08d8e408f5379ccd1aca6 Mon Sep 17 00:00:00 2001 From: chenchun Date: Wed, 24 Dec 2025 12:20:09 +0800 Subject: [PATCH] =?UTF-8?q?style:=20=E6=A0=BC=E5=BC=8F=E5=8C=96=20ChatMana?= =?UTF-8?q?ger.cs=20=E7=9A=84=E5=8F=82=E6=95=B0=E4=B8=8E=E7=A9=BA=E7=99=BD?= =?UTF-8?q?=EF=BC=8C=E8=B0=83=E6=95=B4=E6=8D=A2=E8=A1=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 对构造函数参数、局部变量赋值和方法内空白进行了排版调整(换行与缩进、空格规范化)。 - 删除/添加了一些空行以提高可读性。 - 未修改任何业务逻辑或行为,仅代码样式层面的变更。 --- .../Managers/ChatManager.cs | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/ChatManager.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/ChatManager.cs index 106bd23a..468e7e60 100644 --- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/ChatManager.cs +++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/ChatManager.cs @@ -33,9 +33,12 @@ public class ChatManager : DomainService private readonly UsageStatisticsManager _usageStatisticsManager; private readonly PremiumPackageManager _premiumPackageManager; private readonly AiGateWayManager _aiGateWayManager; + public ChatManager(ILoggerFactory loggerFactory, ISqlSugarRepository messageRepository, - ISqlSugarRepository agentStoreRepository, AiMessageManager aiMessageManager, UsageStatisticsManager usageStatisticsManager, PremiumPackageManager premiumPackageManager, AiGateWayManager aiGateWayManager) + ISqlSugarRepository agentStoreRepository, AiMessageManager aiMessageManager, + UsageStatisticsManager usageStatisticsManager, PremiumPackageManager premiumPackageManager, + AiGateWayManager aiGateWayManager) { _loggerFactory = loggerFactory; _messageRepository = messageRepository; @@ -64,8 +67,8 @@ public class ChatManager : DomainService response.Headers.TryAdd("Cache-Control", "no-cache"); response.Headers.TryAdd("Connection", "keep-alive"); - var modelDescribe=await _aiGateWayManager.GetModelAsync(ModelApiTypeEnum.OpenAi,modelId); - + var modelDescribe = await _aiGateWayManager.GetModelAsync(ModelApiTypeEnum.OpenAi, modelId); + //token状态检查,在应用层统一处理 var client = new OpenAIClient(new ApiKeyCredential(token), new OpenAIClientOptions @@ -177,7 +180,7 @@ public class ChatManager : DomainService }; //设置倍率 usage.SetSupplementalMultiplier(modelDescribe.Multiplier); - + //创建系统回答,用于计费统计 await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId, new MessageInputDto { @@ -185,7 +188,7 @@ public class ChatManager : DomainService ModelId = modelId, TokenUsage = usage }, tokenId); - + //创建用量统计,用于统计分析 await _usageStatisticsManager.SetUsageAsync(userId, modelId, usage, tokenId); @@ -198,7 +201,7 @@ public class ChatManager : DomainService await _premiumPackageManager.TryConsumeTokensAsync(userId, totalTokens); } } - + await SendHttpStreamMessageAsync(httpContext, new AgentResultOutput {