/*
 * Copyright 2018 IBM Corp. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package com.ibm.watson.discovery.v1.model;

import java.util.List;

import com.google.gson.annotations.SerializedName;
import com.ibm.cloud.sdk.core.service.model.GenericModel;

/**
 * A list of Document Segmentation settings.
 */
public class SegmentSettings extends GenericModel {

  private Boolean enabled;
  @SerializedName("selector_tags")
  private List<String> selectorTags;

  /**
   * Gets the enabled.
   *
   * Enables/disables the Document Segmentation feature.
   *
   * @return the enabled
   */
  public Boolean isEnabled() {
    return enabled;
  }

  /**
   * Gets the selectorTags.
   *
   * Defines the heading level that splits into document segments. Valid values are h1, h2, h3, h4, h5, h6. The content
   * of the header field that the segmentation splits at is used as the **title** field for that segmented result.
   *
   * @return the selectorTags
   */
  public List<String> getSelectorTags() {
    return selectorTags;
  }

  /**
   * Sets the enabled.
   *
   * @param enabled the new enabled
   */
  public void setEnabled(final Boolean enabled) {
    this.enabled = enabled;
  }

  /**
   * Sets the selectorTags.
   *
   * @param selectorTags the new selectorTags
   */
  public void setSelectorTags(final List<String> selectorTags) {
    this.selectorTags = selectorTags;
  }
}

/*
 * Copyright 2018 IBM Corp. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package com.ibm.watson.discovery.v1.model;

import java.util.List;

import com.google.gson.annotations.SerializedName;
import com.ibm.cloud.sdk.core.service.model.GenericModel;

/**
 * A list of Document Segmentation settings.
 */
public class SegmentSettings extends GenericModel {

  private Boolean enabled;
  @SerializedName("selector_tags")
  private List<String> selectorTags;
  @SerializedName("annotated_fields")
  private List<String> annotatedFields;

  /**
   * Gets the enabled.
   *
   * Enables/disables the Document Segmentation feature.
   *
   * @return the enabled
   */
  public Boolean isEnabled() {
    return enabled;
  }

  /**
   * Gets the selectorTags.
   *
   * Defines the heading level that splits into document segments. Valid values are h1, h2, h3, h4, h5, h6. The content
   * of the header field that the segmentation splits at is used as the **title** field for that segmented result. Only
   * valid if used with a collection that has **enabled** set to `false` in the **smart_document_understanding** object.
   *
   * @return the selectorTags
   */
  public List<String> getSelectorTags() {
    return selectorTags;
  }

  /**
   * Gets the annotatedFields.
   *
   * Defines the annotated smart document understanding fields that the document is split on. The content of the
   * annotated field that the segmentation splits at is used as the **title** field for that segmented result. For
   * example, if the field `sub-title` is specified, when a document is uploaded each time the smart documement
   * understanding conversion encounters a field of type `sub-title` the document is split at that point and the content
   * of the field used as the title of the remaining content. Thnis split is performed for all instances of the listed
   * fields in the uploaded document. Only valid if used with a collection that has **enabled** set to `true` in the
   * **smart_document_understanding** object.
   *
   * @return the annotatedFields
   */
  public List<String> getAnnotatedFields() {
    return annotatedFields;
  }

  /**
   * Sets the enabled.
   *
   * @param enabled the new enabled
   */
  public void setEnabled(final Boolean enabled) {
    this.enabled = enabled;
  }

  /**
   * Sets the selectorTags.
   *
   * @param selectorTags the new selectorTags
   */
  public void setSelectorTags(final List<String> selectorTags) {
    this.selectorTags = selectorTags;
  }

  /**
   * Sets the annotatedFields.
   *
   * @param annotatedFields the new annotatedFields
   */
  public void setAnnotatedFields(final List<String> annotatedFields) {
    this.annotatedFields = annotatedFields;
  }
}

Add annotatedFields prop to SegmentSettings