Revert change to construction of client

Signed-off-by: John Mazanec <jmazane@amazon.com>
opensearch-project · Oct 11, 2022 · 1721893 · 1721893
1 parent 24e743b
commit 1721893
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/main/java/org/opensearch/neuralsearch/plugin/NeuralSearch.java b/src/main/java/org/opensearch/neuralsearch/plugin/NeuralSearch.java
@@ -52,14 +52,12 @@ public Collection<Object> createComponents(
         final IndexNameExpressionResolver indexNameExpressionResolver,
         final Supplier<RepositoriesService> repositoriesServiceSupplier
     ) {
-        final MachineLearningNodeClient machineLearningNodeClient = new MachineLearningNodeClient(
-            new NodeClient(environment.settings(), threadPool)
-        );
+        final MachineLearningNodeClient machineLearningNodeClient = new MachineLearningNodeClient(client);
         final MLCommonsClientAccessor clientAccessor = new MLCommonsClientAccessor(machineLearningNodeClient);
         return List.of(clientAccessor);
     }
 
-    /**
+    /**z
      * Registering the Action Handlers
      *
      * @return A {@link List} of {@link ActionHandler}