when key is null, skip running predict (#2402) (#2404)

opensearch-project · May 3, 2024 · 29f443f · 29f443f
1 parent 32c9ef8
commit 29f443f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLInferenceIngestProcessorIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLInferenceIngestProcessorIT.java
@@ -112,6 +112,10 @@ public void testMLInferenceProcessorWithObjectFieldType() throws Exception {
         String index_name = "daily_index";
         createPipelineProcessor(createPipelineRequestBody, "diary_embedding_pipeline");
         createIndex(index_name, createIndexRequestBody);
+        // Skip test if key is null
+        if (OPENAI_KEY == null) {
+            return;
+        }
         uploadDocument(index_name, "1", uploadDocumentRequestBody);
         Map document = getDocument(index_name, "1");
         List embeddingList = JsonPath.parse(document).read("_source.diary_embedding");
@@ -194,6 +198,10 @@ public void testMLInferenceProcessorWithNestedFieldType() throws Exception {
         String index_name = "book_index";
         createPipelineProcessor(createPipelineRequestBody, "embedding_pipeline");
         createIndex(index_name, createIndexRequestBody);
+        // Skip test if key is null
+        if (OPENAI_KEY == null) {
+            return;
+        }
         uploadDocument(index_name, "1", uploadDocumentRequestBody);
         Map document = getDocument(index_name, "1");