getFirst is not allowed in java 17 (opensearch-project#3226)

Signed-off-by: Dhrubo Saha <[email protected]>
ryanbogan · Nov 18, 2024 · 4f21953 · 4f21953
1 parent 7041c22
commit 4f21953
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...ml/common/connector/functions/preprocess/CohereMultiModalEmbeddingPreProcessFunction.java b/...ml/common/connector/functions/preprocess/CohereMultiModalEmbeddingPreProcessFunction.java
@@ -25,9 +25,10 @@ public CohereMultiModalEmbeddingPreProcessFunction() {
     public void validate(MLInput mlInput) {
         validateTextDocsInput(mlInput);
         List<String> docs = ((TextDocsInputDataSet) mlInput.getInputDataset()).getDocs();
-        if (docs.isEmpty() || (docs.size() == 1 && docs.getFirst() == null)) {
+        if (docs == null || docs.isEmpty() || (docs.size() == 1 && docs.get(0) == null)) {
             throw new IllegalArgumentException("No image provided");
         }
+
     }
 
     @Override
@@ -41,7 +42,7 @@ public RemoteInferenceInputDataSet process(MLInput mlInput) {
          * connector.pre_process.cohere.embedding
          * Cohere expects An array of image data URIs for the model to embed. Maximum number of images per call is 1.
          */
-        parametersMap.put("images", inputData.getDocs().getFirst());
+        parametersMap.put("images", inputData.getDocs().get(0));
         return RemoteInferenceInputDataSet
             .builder()
             .parameters(convertScriptStringToJsonString(Map.of("parameters", parametersMap)))