xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-03-13 20:53:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(billing): 统一计费代理类BillingChatServiceProxy

Browse Source
This commit is contained in:
Administrator
2025-09-04 16:41:14 +08:00
parent 4a8d21a742
commit 2b5fd810a4
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/proxy/BillingChatServiceProxy.java
View File

@@ -18,6 +18,7 @@ import java.util.function.Consumer;
/**
* 统一计费代理类
* 自动处理所有ChatService的AI回复保存和计费逻辑
*
*/
@Slf4j
@RequiredArgsConstructor
@@ -31,12 +32,12 @@ public class BillingChatServiceProxy implements IChatService {
// 🔥 在AI回复开始前检查余额是否充足
if (!chatCostService.checkBalanceSufficient(chatRequest)) {
String errorMsg = "余额不足，无法使用AI服务，请充值后再试";
log.warn("余额不足阻止AI回复，用户ID: {}, 模型: {}",
log.warn("余额不足阻止AI回复，用户ID: {}, 模型: {}",
chatRequest.getUserId(), chatRequest.getModel());
throw new RuntimeException(errorMsg);
}
log.debug("余额检查通过，开始AI回复，用户ID: {}, 模型: {}",
log.debug("余额检查通过，开始AI回复，用户ID: {}, 模型: {}",
chatRequest.getUserId(), chatRequest.getModel());
// 创建增强的SseEmitter，自动收集AI回复