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