ソースを参照

add gemini key

supeng 2 日 前
コミット
6bb6f81a7d

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

@@ -43,9 +43,9 @@ public class GeminiGenerateContentAction implements Action {
     @Override
     @Override
     public String execute(ActionParam param) {
     public String execute(ActionParam param) {
         Integer type = param.getType();
         Integer type = param.getType();
-//        ThreadLocalRandom random = ThreadLocalRandom.current();
-//        int index = random.nextInt(apiKeyList.size());
-//        String apiKey = apiKeyList.get(index);
+        ThreadLocalRandom random = ThreadLocalRandom.current();
+        int index = random.nextInt(apiKeyList.size());
+        String apiKey = apiKeyList.get(index);
         GeminiParam geminiParam = new GeminiParam();
         GeminiParam geminiParam = new GeminiParam();
         geminiParam.setType(type);
         geminiParam.setType(type);
         geminiParam.setPrompt(param.getPrompt());
         geminiParam.setPrompt(param.getPrompt());
@@ -54,7 +54,7 @@ public class GeminiGenerateContentAction implements Action {
                 || Objects.equals(ContentTypeEnum.VTT.getValue(), type)) {
                 || Objects.equals(ContentTypeEnum.VTT.getValue(), type)) {
             geminiParam.setMediaUrl(param.getInput());
             geminiParam.setMediaUrl(param.getInput());
         }
         }
-//        geminiParam.setApiKey(apiKey);
+        geminiParam.setApiKey(apiKey);
         geminiParam.setModel(MODEL);
         geminiParam.setModel(MODEL);
         geminiParam.setTemperature(TEMPERATURE);
         geminiParam.setTemperature(TEMPERATURE);
         return RetryUtil.executeWithRetry(() -> {
         return RetryUtil.executeWithRetry(() -> {