Dynamically set capabilities endpoint based on configuration

common/transform/src/main/java/ai/h2o/mojos/deploy/common/transform/MojoScorer.java

@@ -302,6 +302,10 @@ public Model getModelInfo() {
  return modelInfoConverter.apply(pipeline);
  }
 
+ public ShapleyLoadOption getEnabledShapleyTypes() {
+ return enabledShapleyTypes;
+ }
+
  /**
  * Method to load mojo pipelines for shapley scoring based on configuration
  *

local-rest-scorer/src/main/java/ai/h2o/mojos/deploy/local/rest/controller/ModelsApiController.java

@@ -9,9 +9,10 @@
 import ai.h2o.mojos.deploy.common.rest.model.ScoreResponse;
 import ai.h2o.mojos.deploy.common.transform.MojoScorer;
 import ai.h2o.mojos.deploy.common.transform.SampleRequestBuilder;
+import ai.h2o.mojos.deploy.common.transform.ShapleyLoadOption;
 import com.google.common.base.Strings;
 import java.io.IOException;
-import java.util.Arrays;
+import java.util.ArrayList;
 import java.util.List;
 
 import org.slf4j.Logger;
@@ -24,13 +25,13 @@
 @Controller
 public class ModelsApiController implements ModelApi {
 
- private static final List<CapabilityType> SUPPORTED_CAPABILITIES
- = Arrays.asList(CapabilityType.SCORE, CapabilityType.CONTRIBUTION_TRANSFORMED);
  private static final Logger log = LoggerFactory.getLogger(ModelsApiController.class);
 
  private final MojoScorer scorer;
  private final SampleRequestBuilder sampleRequestBuilder;
 
+ private final List<CapabilityType> supportedCapabilities;
+
  /**
  * Simple Api controller. Inherits from {@link ModelApi}, which controls global, expected request
  * mappings for the rest service.
@@ -44,6 +45,7 @@ public class ModelsApiController implements ModelApi {
  public ModelsApiController(MojoScorer scorer, SampleRequestBuilder sampleRequestBuilder) {
  this.scorer = scorer;
  this.sampleRequestBuilder = sampleRequestBuilder;
+ this.supportedCapabilities = setSupportedCapabilities();
  }
 
  @Override
@@ -58,7 +60,7 @@ public ResponseEntity<String> getModelId() {
 
  @Override
  public ResponseEntity<List<CapabilityType>> getCapabilities() {
- return ResponseEntity.ok(SUPPORTED_CAPABILITIES);
+ return ResponseEntity.ok(supportedCapabilities);
  }
 
  @Override
@@ -116,4 +118,26 @@ public ResponseEntity<ContributionResponse> getContribution(
  public ResponseEntity<ScoreRequest> getSampleRequest() {
  return ResponseEntity.ok(sampleRequestBuilder.build(scorer.getPipeline().getInputMeta()));
  }
+
+ private List<CapabilityType> setSupportedCapabilities() {
+ List<CapabilityType> capabilityTypes = new ArrayList<>();
+ capabilityTypes.add(CapabilityType.SCORE);
+ ShapleyLoadOption enabledShapleyTypes = scorer.getEnabledShapleyTypes();
+ switch (enabledShapleyTypes) {
+ case ALL:
+ capabilityTypes.add(CapabilityType.CONTRIBUTION_ORIGINAL);
+ capabilityTypes.add(CapabilityType.CONTRIBUTION_TRANSFORMED);
+ break;
+ case ORIGINAL:
+ capabilityTypes.add(CapabilityType.CONTRIBUTION_ORIGINAL);
+ break;
+ case TRANSFORMED:
+ capabilityTypes.add(CapabilityType.CONTRIBUTION_TRANSFORMED);
+ break;
+ case NONE:
+ default:
+ break;
+ }
+ return capabilityTypes;
+ }
 }