changing file structure for bwc

Signed-off-by: Amit Galitzky <[email protected]>
opensearch-project · Jun 8, 2024 · 50b44eb · 50b44eb
1 parent 1f9d215
commit 50b44eb
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/build.gradle b/build.gradle
@@ -494,7 +494,29 @@ List<Provider<RegularFile>> plugins = [
                 return new RegularFile() {
                     @Override
                     File getAsFile() {
-                        return configurations.zipArchive.asFileTree.getFiles()
+                        return configurations.zipArchive.asFileTree.matching{include "**/opensearch-ml-plugin-${opensearch_build}.zip"}.getSingleFile()
+                    }
+                }
+            }
+        }),
+        provider(new Callable<RegularFile>(){
+            @Override
+            RegularFile call() throws Exception {
+                return new RegularFile() {
+                    @Override
+                    File getAsFile() {
+                        return configurations.zipArchive.asFileTree.matching{include "**/opensearch-knn-${opensearch_build}.zip"}.getSingleFile()
+                    }
+                }
+            }
+        }),
+        provider(new Callable<RegularFile>(){
+            @Override
+            RegularFile call() throws Exception {
+                return new RegularFile() {
+                    @Override
+                    File getAsFile() {
+                        return configurations.zipArchive.asFileTree.matching{include "**/neural-search-${opensearch_build}.zip"}.getSingleFile()
                     }
                 }
             }

diff --git a/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java b/src/test/java/org/opensearch/flowframework/rest/FlowFrameworkRestApiIT.java
@@ -525,7 +525,11 @@ public void testAllDefaultUseCasesCreation() throws Exception {
     }
 
     public void testSemanticSearchWithLocalModelEndToEnd() throws Exception {
-
+        // Checking if plugins are part of the integration test cluster so we can continue with this test
+        List<String> plugins = catPlugins();
+        if (!plugins.contains("opensearch-knn") && plugins.contains("opensearch-neural-search")) {
+            return;
+        }
         Map<String, Object> defaults = new HashMap<>();
         defaults.put("register_local_pretrained_model.name", "huggingface/sentence-transformers/all-MiniLM-L6-v2");
         defaults.put("register_local_pretrained_model.version", "1.0.1");