From 9bbf90458a71c40fbb559ef7149a9b5af9a7309b Mon Sep 17 00:00:00 2001
From: ageer <ageerle@163.com>
Date: 星期三, 28 二月 2024 15:36:15 +0800
Subject: [PATCH] 修复付费用户验证
---
ruoyi-modules/ruoyi-system/src/main/java/com/xmzs/system/service/impl/ISseServiceImpl.java | 42 +++++++++++++++++++++++++++++-------------
1 files changed, 29 insertions(+), 13 deletions(-)
diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/xmzs/system/service/impl/ISseServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/com/xmzs/system/service/impl/ISseServiceImpl.java
index b2462c8..ff024fb 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/xmzs/system/service/impl/ISseServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/xmzs/system/service/impl/ISseServiceImpl.java
@@ -74,8 +74,23 @@
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("鍏嶈垂鐢ㄦ埛鏆傛椂涓嶆敮鎸佹妯″瀷,璇峰垏鎹pt-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 @@
}
}
// TODO 娣诲姞鏋氫妇
- if ("0".equals(sysUser.getUserGrade()) && !ChatCompletion.Model.GPT_3_5_TURBO.getName().equals(model)) {
- // 鍒涘缓骞跺彂閫佷竴涓悕涓� "error" 鐨勪簨浠讹紝甯︽湁閿欒娑堟伅鍜岀姸鎬佺爜
- SseEmitter.SseEventBuilder event = SseEmitter.event()
- .name("error") // 瀹㈡埛绔皢鐩戝惉杩欎釜浜嬩欢鍚�
- .data("鍏嶈垂鐢ㄦ埛鏆傛椂涓嶆敮鎸佹妯″瀷,璇峰垏鎹pt-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("鍏嶈垂鐢ㄦ埛鏆傛椂涓嶆敮鎸佹妯″瀷,璇峰垏鎹pt-3.5-turbo妯″瀷鎴栬�呯偣鍑汇�婅繘鍏ュ競鍦洪�夎喘鎮ㄧ殑鍟嗗搧銆嬪厖鍊煎悗浣跨敤!");
+// try {
+// emitter.send(event);
+// } catch (IOException e) {
+// throw new RuntimeException(e);
+// }
+// emitter.complete();
+//
+// }
}
/**
--
Gitblit v1.9.3