diff --git a/plugin/src/main/java/org/opensearch/ml/rest/RestMLCreateConnectorAction.java b/plugin/src/main/java/org/opensearch/ml/rest/RestMLCreateConnectorAction.java index c41fe3784e..a0398deb6f 100644 --- a/plugin/src/main/java/org/opensearch/ml/rest/RestMLCreateConnectorAction.java +++ b/plugin/src/main/java/org/opensearch/ml/rest/RestMLCreateConnectorAction.java @@ -13,6 +13,7 @@ import java.io.IOException; import java.util.List; import java.util.Locale; +import java.util.Set; import org.opensearch.core.xcontent.XContentParser; import org.opensearch.ml.common.settings.MLFeatureEnabledSetting; @@ -21,13 +22,14 @@ import org.opensearch.ml.common.transport.connector.MLCreateConnectorRequest; 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 RestMLCreateConnectorAction extends BaseRestHandler { +public class RestMLCreateConnectorAction extends BaseRestHandler implements RestRequestFilter { private static final String ML_CREATE_CONNECTOR_ACTION = "ml_create_connector_action"; private final MLFeatureEnabledSetting mlFeatureEnabledSetting; @@ -76,4 +78,9 @@ MLCreateConnectorRequest getRequest(RestRequest request) throws IOException { mlCreateConnectorInput.setTenantId(tenantId); return new MLCreateConnectorRequest(mlCreateConnectorInput); } + + @Override + public Set getFilteredFields() { + return Set.of("credential.openai_key"); + } }