AlignDocumentRequest

package com.acumenvelocity.ath.model;

import java.net.URI;
import java.util.Objects;
import java.util.UUID;

import com.fasterxml.jackson.annotation.JsonProperty;

import io.swagger.annotations.ApiModelProperty;

@javax.annotation.Generated(value = "org.openapitools.codegen.languages.JavaInflectorServerCodegen", comments = "Generator version: 7.15.0")
public class AlignDocumentRequest {
  @JsonProperty("src_lang")
  private String srcLang = "en";

  @JsonProperty("trg_lang")
  private String trgLang = "fr";

  @JsonProperty("doc_trl_gcs_url")
  private URI docTrlGcsUrl = URI.create("");

  @JsonProperty("doc_trl_encoding")
  private String docTrlEncoding = "";

  @JsonProperty("src_srx")
  private String srcSrx = "";

  @JsonProperty("trg_srx")
  private String trgSrx = "";

  @JsonProperty("use_alignment_model")
  private Boolean useAlignmentModel = false;

  @JsonProperty("alignment_model_name")
  private String alignmentModelName = "gemini-2.5-pro";

  @JsonProperty("use_codes_reinsertion_model")
  private Boolean useCodesReinsertionModel = false;

  @JsonProperty("codes_reinsertion_model_name")
  private String codesReinsertionModelName = "gemini-2.5-flash";

  @JsonProperty("user_id")
  private UUID userId;

  /**
   * Source language ISO code.
   **/
  public AlignDocumentRequest srcLang(String srcLang) {
    this.srcLang = srcLang;
    return this;
  }

  @ApiModelProperty(required = true, value = "Source language ISO code.")
  @JsonProperty("src_lang")
  public String getSrcLang() {
    return srcLang;
  }

  public void setSrcLang(String srcLang) {
    this.srcLang = srcLang;
  }

  /**
   * Target language ISO code.
   **/
  public AlignDocumentRequest trgLang(String trgLang) {
    this.trgLang = trgLang;
    return this;
  }

  @ApiModelProperty(required = true, value = "Target language ISO code.")
  @JsonProperty("trg_lang")
  public String getTrgLang() {
    return trgLang;
  }

  public void setTrgLang(String trgLang) {
    this.trgLang = trgLang;
  }

  /**
   * GCS URL (gs://bucket/blob) of an existing translation if to be imported along with the original
   * document. If specified, the passed translation is segmented and aligned with extracted original
   * document's segments. If the number of segments is different, the translation is ignored and
   * regular import pipeline with TM and MT is run on the original document. The caller takes the
   * risk of the possible segments' misalignment, in which case manual postediting of misaligned
   * segments is requered.
   **/
  public AlignDocumentRequest docTrlGcsUrl(URI docTrlGcsUrl) {
    this.docTrlGcsUrl = docTrlGcsUrl;
    return this;
  }

  @ApiModelProperty(required = true, value = "GCS URL (gs://bucket/blob) of an existing translation if to be  imported along with the original document. If specified, the passed translation  is segmented and aligned with extracted original document's segments. If the  number of segments is different, the translation is ignored and regular import  pipeline with TM and MT is run on the original document. The caller takes the  risk of the possible segments' misalignment, in which case manual postediting of  misaligned segments is requered.")
  @JsonProperty("doc_trl_gcs_url")
  public URI getDocTrlGcsUrl() {
    return docTrlGcsUrl;
  }

  public void setDocTrlGcsUrl(URI docTrlGcsUrl) {
    this.docTrlGcsUrl = docTrlGcsUrl;
  }

  /**
   * Optional translation encoding if `doc_trl_gcs_url` is specified. If omitted and
   * `doc_trl_gcs_url` is specified, the original document's encoding will be used.
   **/
  public AlignDocumentRequest docTrlEncoding(String docTrlEncoding) {
    this.docTrlEncoding = docTrlEncoding;
    return this;
  }

  @ApiModelProperty(value = "Optional translation encoding if `doc_trl_gcs_url` is specified.  If omitted and `doc_trl_gcs_url` is specified, the original document's encoding  will be used.")
  @JsonProperty("doc_trl_encoding")
  public String getDocTrlEncoding() {
    return docTrlEncoding;
  }

  public void setDocTrlEncoding(String docTrlEncoding) {
    this.docTrlEncoding = docTrlEncoding;
  }

  /**
   * Optional SRX string for source segmentation.
   **/
  public AlignDocumentRequest srcSrx(String srcSrx) {
    this.srcSrx = srcSrx;
    return this;
  }

  @ApiModelProperty(value = "Optional SRX string for source segmentation.")
  @JsonProperty("src_srx")
  public String getSrcSrx() {
    return srcSrx;
  }

  public void setSrcSrx(String srcSrx) {
    this.srcSrx = srcSrx;
  }

  /**
   * Optional SRX string for target segmentation.
   **/
  public AlignDocumentRequest trgSrx(String trgSrx) {
    this.trgSrx = trgSrx;
    return this;
  }

  @ApiModelProperty(value = "Optional SRX string for target segmentation.")
  @JsonProperty("trg_srx")
  public String getTrgSrx() {
    return trgSrx;
  }

  public void setTrgSrx(String trgSrx) {
    this.trgSrx = trgSrx;
  }

  /**
   * True to use LLM-based documents alignment, false to have Okapi align documents.
   **/
  public AlignDocumentRequest useAlignmentModel(Boolean useAlignmentModel) {
    this.useAlignmentModel = useAlignmentModel;
    return this;
  }

  @ApiModelProperty(value = "True to use LLM-based documents alignment, false to have Okapi align documents.")
  @JsonProperty("use_alignment_model")
  public Boolean getUseAlignmentModel() {
    return useAlignmentModel;
  }

  public void setUseAlignmentModel(Boolean useAlignmentModel) {
    this.useAlignmentModel = useAlignmentModel;
  }

  /**
   * Gemini AI model name to use for documents alignment.
   **/
  public AlignDocumentRequest alignmentModelName(String alignmentModelName) {
    this.alignmentModelName = alignmentModelName;
    return this;
  }

  @ApiModelProperty(value = "Gemini AI model name to use for documents alignment.")
  @JsonProperty("alignment_model_name")
  public String getAlignmentModelName() {
    return alignmentModelName;
  }

  public void setAlignmentModelName(String alignmentModelName) {
    this.alignmentModelName = alignmentModelName;
  }

  /**
   * True to use LLM-based target codes reinsertion, false to have Okapi reinsert. This option has
   * effect only when plainTextMT is true.
   **/
  public AlignDocumentRequest useCodesReinsertionModel(Boolean useCodesReinsertionModel) {
    this.useCodesReinsertionModel = useCodesReinsertionModel;
    return this;
  }

  @ApiModelProperty(value = "True to use LLM-based target codes reinsertion, false to have Okapi reinsert. This option has effect only when plainTextMT is true.")
  @JsonProperty("use_codes_reinsertion_model")
  public Boolean getUseCodesReinsertionModel() {
    return useCodesReinsertionModel;
  }

  public void setUseCodesReinsertionModel(Boolean useCodesReinsertionModel) {
    this.useCodesReinsertionModel = useCodesReinsertionModel;
  }

  /**
   * Gemini AI model name to use for target codes reinsertion.
   **/
  public AlignDocumentRequest codesReinsertionModelName(String codesReinsertionModelName) {
    this.codesReinsertionModelName = codesReinsertionModelName;
    return this;
  }

  @ApiModelProperty(value = "Gemini AI model name to use for target codes reinsertion.")
  @JsonProperty("codes_reinsertion_model_name")
  public String getCodesReinsertionModelName() {
    return codesReinsertionModelName;
  }

  public void setCodesReinsertionModelName(String codesReinsertionModelName) {
    this.codesReinsertionModelName = codesReinsertionModelName;
  }

  /**
   * UUID of the user importing the document.
   **/
  public AlignDocumentRequest userId(UUID userId) {
    this.userId = userId;
    return this;
  }

  @ApiModelProperty(required = true, value = "UUID of the user importing the document.")
  @JsonProperty("user_id")
  public UUID getUserId() {
    return userId;
  }

  public void setUserId(UUID userId) {
    this.userId = userId;
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }
    AlignDocumentRequest alignDocumentRequest = (AlignDocumentRequest) o;
    return Objects.equals(srcLang, alignDocumentRequest.srcLang) &&
        Objects.equals(trgLang, alignDocumentRequest.trgLang) &&
        Objects.equals(docTrlGcsUrl, alignDocumentRequest.docTrlGcsUrl) &&
        Objects.equals(docTrlEncoding, alignDocumentRequest.docTrlEncoding) &&
        Objects.equals(srcSrx, alignDocumentRequest.srcSrx) &&
        Objects.equals(trgSrx, alignDocumentRequest.trgSrx) &&
        Objects.equals(useAlignmentModel, alignDocumentRequest.useAlignmentModel) &&
        Objects.equals(alignmentModelName, alignDocumentRequest.alignmentModelName) &&
        Objects.equals(useCodesReinsertionModel, alignDocumentRequest.useCodesReinsertionModel) &&
        Objects.equals(codesReinsertionModelName, alignDocumentRequest.codesReinsertionModelName) &&
        Objects.equals(userId, alignDocumentRequest.userId);
  }

  @Override
  public int hashCode() {
    return Objects.hash(srcLang, trgLang, docTrlGcsUrl, docTrlEncoding, srcSrx, trgSrx,
        useAlignmentModel, alignmentModelName, useCodesReinsertionModel, codesReinsertionModelName,
        userId);
  }

  @Override
  public String toString() {
    StringBuilder sb = new StringBuilder();
    sb.append("class AlignDocumentRequest {\n");

    sb.append("    srcLang: ").append(toIndentedString(srcLang)).append("\n");
    sb.append("    trgLang: ").append(toIndentedString(trgLang)).append("\n");
    sb.append("    docTrlGcsUrl: ").append(toIndentedString(docTrlGcsUrl)).append("\n");
    sb.append("    docTrlEncoding: ").append(toIndentedString(docTrlEncoding)).append("\n");
    sb.append("    srcSrx: ").append(toIndentedString(srcSrx)).append("\n");
    sb.append("    trgSrx: ").append(toIndentedString(trgSrx)).append("\n");
    sb.append("    useAlignmentModel: ").append(toIndentedString(useAlignmentModel)).append("\n");
    sb.append("    alignmentModelName: ").append(toIndentedString(alignmentModelName)).append("\n");
    sb.append("    useCodesReinsertionModel: ").append(toIndentedString(useCodesReinsertionModel))
        .append("\n");
    sb.append("    codesReinsertionModelName: ").append(toIndentedString(codesReinsertionModelName))
        .append("\n");
    sb.append("    userId: ").append(toIndentedString(userId)).append("\n");
    sb.append("}");
    return sb.toString();
  }

  /**
   * Convert the given object to string with each line indented by 4 spaces
   * (except the first line).
   */
  private String toIndentedString(Object o) {
    if (o == null) {
      return "null";
    }
    return o.toString().replace("\n", "\n    ");
  }
}