xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-03-25 02:23:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复付费用户验证

Browse Source
This commit is contained in:
ageer
2024-02-28 15:36:15 +08:00
parent 88e1d02ca5
commit 9bbf90458a
1 changed files with 29 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
ruoyi-modules/ruoyi-system/src/main/java/com/xmzs/system/service/impl/ISseServiceImpl.java
View File

@@ -74,8 +74,23 @@ public class ISseServiceImpl implements ISseService {
public SseEmitter sseChat(ChatRequest chatRequest) {
LocalCache.CACHE.put("userId",getUserId());
SseEmitter sseEmitter = new SseEmitter(0L);
SysUser sysUser = sysUserMapper.selectById(getUserId());
// TODO 添加枚举
if ("0".equals(sysUser.getUserGrade()) && !ChatCompletion.Model.GPT_3_5_TURBO.getName().equals(chatRequest.getModel())) {
// 创建并发送一个名为 "error" 的事件，带有错误消息和状态码
SseEmitter.SseEventBuilder event = SseEmitter.event()
.name("error") // 客户端将监听这个事件名
.data("免费用户暂时不支持此模型,请切换gpt-3.5-turbo模型或者点击《进入市场选购您的商品》充值后使用!");
try {
sseEmitter.send(event);
} catch (IOException e) {
throw new RuntimeException(e);
}
sseEmitter.complete();
return sseEmitter;
}
SSEEventSourceListener openAIEventSourceListener = new SSEEventSourceListener(sseEmitter);
checkUserGrade(sseEmitter, chatRequest.getModel());
// 获取对话消息列表
List<Message> msgList = chatRequest.getMessages();
// 图文识别上下文信息
@@ -255,18 +270,19 @@ public class ISseServiceImpl implements ISseService {
}
}
// TODO 添加枚举
if ("0".equals(sysUser.getUserGrade()) && !ChatCompletion.Model.GPT_3_5_TURBO.getName().equals(model)) {
// 创建并发送一个名为 "error" 的事件，带有错误消息和状态码
SseEmitter.SseEventBuilder event = SseEmitter.event()
.name("error") // 客户端将监听这个事件名
.data("免费用户暂时不支持此模型,请切换gpt-3.5-turbo模型或者点击《进入市场选购您的商品》充值后使用!");
try {
emitter.send(event);
} catch (IOException e) {
throw new RuntimeException(e);
}
emitter.complete();
}
// if ("0".equals(sysUser.getUserGrade()) && !ChatCompletion.Model.GPT_3_5_TURBO.getName().equals(model)) {
// // 创建并发送一个名为 "error" 的事件，带有错误消息和状态码
// SseEmitter.SseEventBuilder event = SseEmitter.event()
// .name("error") // 客户端将监听这个事件名
// .data("免费用户暂时不支持此模型,请切换gpt-3.5-turbo模型或者点击《进入市场选购您的商品》充值后使用!");
// try {
// emitter.send(event);
// } catch (IOException e) {
// throw new RuntimeException(e);
// }
// emitter.complete();
//
// }
}
/**