From 373424bd010cfae218570e602199e2ac6754a0ef Mon Sep 17 00:00:00 2001
From: ageerle <ageerle@163.com>
Date: 星期六, 24 五月 2025 16:18:18 +0800
Subject: [PATCH] feat(更新日志):
---
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java | 22 ++++++++++------------
1 files changed, 10 insertions(+), 12 deletions(-)
diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
index da2e589..cac159f 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
@@ -13,7 +13,7 @@
import org.ruoyi.common.core.exception.ServiceException;
import org.ruoyi.common.core.exception.base.BaseException;
import org.ruoyi.common.satoken.utils.LoginHelper;
-import org.ruoyi.domain.ChatToken;
+import org.ruoyi.domain.ChatUsageToken;
import org.ruoyi.domain.bo.ChatMessageBo;
import org.ruoyi.domain.vo.ChatModelVo;
import org.ruoyi.service.IChatMessageService;
@@ -48,6 +48,9 @@
*/
@Override
public void deductToken(ChatRequest chatRequest) {
+ if(chatRequest.getUserId()==null || chatRequest.getSessionId()==null){
+ return;
+ }
int tokens = TikTokensUtil.tokens(chatRequest.getModel(), chatRequest.getPrompt());
@@ -55,24 +58,19 @@
ChatMessageBo chatMessageBo = new ChatMessageBo();
- if(chatRequest.getSessionId() == null){
- Object sessionId = LocalCache.CACHE.get("sessionId");
- chatRequest.setSessionId((Long) sessionId);
- }
-
- Object userId = LocalCache.CACHE.get("userId");
- chatMessageBo.setUserId((Long) userId);
-
+ // 璁剧疆鐢ㄦ埛id
+ chatMessageBo.setUserId(chatRequest.getUserId());
// 璁剧疆瀵硅瘽瑙掕壊
chatMessageBo.setRole(chatRequest.getRole());
-
+ // 璁剧疆浼氳瘽id
chatMessageBo.setSessionId(chatRequest.getSessionId());
+ // 璁剧疆瀵硅瘽鍐呭
chatMessageBo.setContent(chatRequest.getPrompt());
// 璁＄畻鎬籺oken鏁�
- ChatToken chatToken = chatTokenService.queryByUserId(chatMessageBo.getUserId(), modelName);
+ ChatUsageToken chatToken = chatTokenService.queryByUserId(chatMessageBo.getUserId(), modelName);
if (chatToken == null) {
- chatToken = new ChatToken();
+ chatToken = new ChatUsageToken();
chatToken.setToken(0);
}
int totalTokens = chatToken.getToken() + tokens;
--
Gitblit v1.9.3