xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-14 04:13:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 扣费时无法获取用户id

Browse Source
This commit is contained in:
ageerle
2025-04-22 10:40:36 +08:00
parent 6281840f36
commit 1b793e822a
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ChatCostServiceImpl.java
View File

@@ -54,6 +54,12 @@ public class ChatCostServiceImpl implements IChatCostService {
ChatMessageBo chatMessageBo = new ChatMessageBo(); ChatMessageBo chatMessageBo = new ChatMessageBo();
Object userId = LocalCache.CACHE.get("userId");
if(userId!=null){
chatMessageBo.setUserId((Long) userId);
}else {
chatMessageBo.setUserId(getUserId());
}
// 计算总token数 // 计算总token数
ChatToken chatToken = chatTokenService.queryByUserId(getUserId(), modelName); ChatToken chatToken = chatTokenService.queryByUserId(getUserId(), modelName);
if (chatToken == null) { if (chatToken == null) {
@@ -97,12 +103,6 @@ public class ChatCostServiceImpl implements IChatCostService {
chatToken.setUserId(chatMessageBo.getUserId()); chatToken.setUserId(chatMessageBo.getUserId());
chatTokenService.editToken(chatToken); chatTokenService.editToken(chatToken);
} }
Object userId = LocalCache.CACHE.get("userId");
if(userId!=null){
chatMessageBo.setUserId((Long) userId);
}else {
chatMessageBo.setUserId(getUserId());
}
// 保存消息记录 // 保存消息记录
chatMessageService.insertByBo(chatMessageBo); chatMessageService.insertByBo(chatMessageBo);
} }

4
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/SseServiceImpl.java
View File

@@ -73,7 +73,7 @@ public class SseServiceImpl implements ISseService {
public SseEmitter sseChat(ChatRequest chatRequest, HttpServletRequest request) { public SseEmitter sseChat(ChatRequest chatRequest, HttpServletRequest request) {
SseEmitter sseEmitter = new SseEmitter(); SseEmitter sseEmitter = new SseEmitter();
try { try {
// 构建消息列表增加联网、知识库等内容 // 构建消息列表
buildChatMessageList(chatRequest); buildChatMessageList(chatRequest);
if (!StpUtil.isLogin()) { if (!StpUtil.isLogin()) {
// 未登录用户限制对话次数 // 未登录用户限制对话次数
@@ -145,7 +145,7 @@ public class SseServiceImpl implements ISseService {
if(StringUtils.isEmpty(sysPrompt)){ if(StringUtils.isEmpty(sysPrompt)){
sysPrompt ="你是一个由RuoYI-AI开发的人工智能助手，名字叫熊猫助手。你擅长中英文对话，能够理解并处理各种问题，提供安全、有帮助、准确的回答。" + sysPrompt ="你是一个由RuoYI-AI开发的人工智能助手，名字叫熊猫助手。你擅长中英文对话，能够理解并处理各种问题，提供安全、有帮助、准确的回答。" +
"当前时间："+ DateUtils.getDate()+ "当前时间："+ DateUtils.getDate()+
"#注意：回复之前注意结合上下文内容。 "; "#注意：回复之前注意结合上下文和工具返回内容。";
} }
// 设置系统默认提示词 // 设置系统默认提示词