supeng 3 weeks ago
parent
commit
96133c351d

+ 6 - 6
content-understanding-core/src/main/java/com/tzld/piaoquan/content/understanding/service/impl/GeminiGenerateContentAction.java

@@ -51,12 +51,12 @@ public class GeminiGenerateContentAction implements Action {
     @Override
     public String execute(ActionParam param) {
         Integer type = param.getType();
-//        ThreadLocalRandom random = ThreadLocalRandom.current();
-//        int index = random.nextInt(apiKeyList.size());
-//        String apiKey = apiKeyList.get(index);
-        int size = apiKeyList.size();
-        int index = getIndex(size);
-        String apiKey = apiKeyList.get(index);
+        String apiKey = "";
+        if (Objects.nonNull(apiKeyList) && !apiKeyList.isEmpty()) {
+            int size = apiKeyList.size();
+            int index = getIndex(size);
+            apiKey = apiKeyList.get(index);
+        }
         GeminiParam geminiParam = new GeminiParam();
         geminiParam.setType(type);
         geminiParam.setPrompt(param.getPrompt());