Apply the same to Register Model Action

Signed-off-by: Craig Perkins <[email protected]>

Author: cwperks

plugin/src/main/java/org/opensearch/ml/rest/RestMLRegisterModelAction.java

@@ -17,6 +17,7 @@
 import java.io.IOException;
 import java.util.List;
 import java.util.Locale;
+import java.util.Set;
 
 import org.opensearch.cluster.service.ClusterService;
 import org.opensearch.common.settings.Settings;
@@ -28,13 +29,14 @@
 import org.opensearch.ml.common.transport.register.MLRegisterModelRequest;
 import org.opensearch.rest.BaseRestHandler;
 import org.opensearch.rest.RestRequest;
+import org.opensearch.rest.RestRequestFilter;
 import org.opensearch.rest.action.RestToXContentListener;
 import org.opensearch.transport.client.node.NodeClient;
 
 import com.google.common.annotations.VisibleForTesting;
 import com.google.common.collect.ImmutableList;
 
-public class RestMLRegisterModelAction extends BaseRestHandler {
+public class RestMLRegisterModelAction extends BaseRestHandler implements RestRequestFilter {
     private static final String ML_REGISTER_MODEL_ACTION = "ml_register_model_action";
     private final MLFeatureEnabledSetting mlFeatureEnabledSetting;
 
@@ -107,4 +109,9 @@ MLRegisterModelRequest getRequest(RestRequest request) throws IOException {
         }
         return new MLRegisterModelRequest(mlInput);
     }
+
+    @Override
+    public Set<String> getFilteredFields() {
+        return Set.of("connector.credential", "*.Authorization");
+    }
 }