孙永军 пре 1 година
родитељ
комит
2fb8b407c9

+ 13 - 6
src/main/java/com/sxtvs/open/api/chat/service/impl/KimiChatServiceImpl.java

@@ -32,10 +32,7 @@ import java.net.http.HttpClient;
 import java.net.http.HttpRequest;
 import java.net.http.HttpResponse;
 import java.time.LocalDateTime;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Collections;
-import java.util.List;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -52,7 +49,17 @@ public class KimiChatServiceImpl extends ServiceImpl<KimiChatMapper, KimiChat> i
 
     private final RestTemplate restTemplate = new RestTemplate();
 
-    private final String token = "sk-eKQV5KAqkQHINWZV9IBbsuefO21A6oTToY9pv9nyDeXPH0zu";
+    private final List<String> token = Arrays.asList(
+            "sk-eKQV5KAqkQHINWZV9IBbsuefO21A6oTToY9pv9nyDeXPH0zu",
+            "sk-wMy3vyR7mCQc5vvr4uFm6lkfhdxFJZtQxx54pKvaJcVIa3XQ",
+            "sk-ftWQ35trs1omUIUuKzsdeb6idKK9p6PzK4C9U5YTkHkt0PtD",
+            "sk-9VL5bvkhGhtM3GhBD80orZm75vdGLrRH8xRF3Takl8Q82oea",
+            "sk-mu6zmA4ZYg3xtayd5TLq0gXCOZNk8eOAGpTd7pri4inpFOXs",
+            "sk-22Rh0RPkx7Z1uCKERWNSq8dgUaYZdSVy9d3d2otDyecGJZny",
+            "sk-VRMUyGAnxqaCfnTOhttLUvysVLRz1MdwrBOyV6kvv6YKfk0s"
+            );
+
+    private final Random random = new Random();
 
     @Resource
     private SSEService sseService;
@@ -73,7 +80,7 @@ public class KimiChatServiceImpl extends ServiceImpl<KimiChatMapper, KimiChat> i
                 .uri(uri)
                 .header("Accept", "text/event-stream")
 //                .header("Content-Type", "application/json")
-                .header("Authorization", "Bearer " + token)
+                .header("Authorization", "Bearer " + token.get(random.nextInt(token.size())))
                 .POST( HttpRequest.BodyPublishers.ofString(JSON.toJSONString(
                     kimiRequest
                 )))