package com.acumenvelocity.ath.model;
   
   import java.util.Date;
   import java.util.Objects;
   import java.util.UUID;
   
   import com.fasterxml.jackson.annotation.JsonCreator;
   import com.fasterxml.jackson.annotation.JsonProperty;
   import com.fasterxml.jackson.annotation.JsonValue;
  
  import io.swagger.annotations.ApiModelProperty;
  
  @javax.annotation.Generated(value = "org.openapitools.codegen.languages.JavaInflectorServerCodegen", comments = "Generator version: 7.15.0")
  public class TrainingCompletedResponse {
    /**
     * Training completed successfully.
     */
    public enum StatusEnum {
      COMPLETED("COMPLETED");
  
      private String value;
  
      StatusEnum(String value) {
        this.value = value;
      }
  
      @Override
      @JsonValue
      public String toString() {
        return String.valueOf(value);
      }
  
      @JsonCreator
      public static StatusEnum fromValue(String text) {
        for (StatusEnum b : StatusEnum.values()) {
          if (String.valueOf(b.value).equals(text)) {
            return b;
          }
        }
        throw new IllegalArgumentException("Unexpected value '" + text + "'");
      }
    }
  
    @JsonProperty("status")
    private StatusEnum status;
  
    @JsonProperty("training_job_id")
    private UUID trainingJobId;
  
    @JsonProperty("model_id")
    private String modelId = "";
  
    @JsonProperty("model_name")
    private String modelName = "";
  
    @JsonProperty("dataset_id")
    private String datasetId = "";
  
    @JsonProperty("completed_at")
    private Date completedAt;
  
    @JsonProperty("dataset_stats")
    private DatasetStats datasetStats;
  
    /**
     * Training completed successfully.
     **/
    public TrainingCompletedResponse status(StatusEnum status) {
      this.status = status;
      return this;
    }
  
    @ApiModelProperty(example = "COMPLETED", required = true, value = "Training completed successfully.")
    @JsonProperty("status")
    public StatusEnum getStatus() {
      return status;
    }
  
    public void setStatus(StatusEnum status) {
      this.status = status;
    }
  
    /**
     * Unique identifier of the training job.
     **/
    public TrainingCompletedResponse trainingJobId(UUID trainingJobId) {
      this.trainingJobId = trainingJobId;
      return this;
    }
  
    @ApiModelProperty(example = "550e8400-e29b-41d4-a716-446655440000", required = true, value = "Unique identifier of the training job.")
    @JsonProperty("training_job_id")
    public UUID getTrainingJobId() {
      return trainingJobId;
    }
  
    public void setTrainingJobId(UUID trainingJobId) {
      this.trainingJobId = trainingJobId;
    }
 
   /**
    * AutoML model resource name.
    **/
   public TrainingCompletedResponse modelId(String modelId) {
     this.modelId = modelId;
     return this;
   }
 
   @ApiModelProperty(example = "projects/my-project/locations/us-central1/models/12345", required = true, value = "AutoML model resource name.")
   @JsonProperty("model_id")
   public String getModelId() {
     return modelId;
   }
 
   public void setModelId(String modelId) {
     this.modelId = modelId;
   }
 
   /**
    * Display name of the trained model.
    **/
   public TrainingCompletedResponse modelName(String modelName) {
     this.modelName = modelName;
     return this;
   }
 
   @ApiModelProperty(example = "my-custom-en-es-model", required = true, value = "Display name of the trained model.")
   @JsonProperty("model_name")
   public String getModelName() {
     return modelName;
   }
 
   public void setModelName(String modelName) {
     this.modelName = modelName;
   }
 
   /**
    * AutoML dataset resource name used for training.
    **/
   public TrainingCompletedResponse datasetId(String datasetId) {
     this.datasetId = datasetId;
     return this;
   }
 
   @ApiModelProperty(example = "projects/my-project/locations/us-central1/datasets/67890", value = "AutoML dataset resource name used for training.")
   @JsonProperty("dataset_id")
   public String getDatasetId() {
     return datasetId;
   }
 
   public void setDatasetId(String datasetId) {
     this.datasetId = datasetId;
   }
 
   /**
    * Timestamp when training completed.
    **/
   public TrainingCompletedResponse completedAt(Date completedAt) {
     this.completedAt = completedAt;
     return this;
   }
 
   @ApiModelProperty(example = "2025-12-16T18:30Z", required = true, value = "Timestamp when training completed.")
   @JsonProperty("completed_at")
   public Date getCompletedAt() {
     return completedAt;
   }
 
   public void setCompletedAt(Date completedAt) {
     this.completedAt = completedAt;
   }
 
   /**
    **/
   public TrainingCompletedResponse datasetStats(DatasetStats datasetStats) {
     this.datasetStats = datasetStats;
     return this;
   }
 
   @ApiModelProperty(value = "")
   @JsonProperty("dataset_stats")
   public DatasetStats getDatasetStats() {
     return datasetStats;
   }
 
   public void setDatasetStats(DatasetStats datasetStats) {
     this.datasetStats = datasetStats;
   }
 
   @Override
   public boolean equals(Object o) {
     if (this == o) {
       return true;
     }
     if (o == null || getClass() != o.getClass()) {
       return false;
     }
     TrainingCompletedResponse trainingCompletedResponse = (TrainingCompletedResponse) o;
     return Objects.equals(status, trainingCompletedResponse.status) &&
         Objects.equals(trainingJobId, trainingCompletedResponse.trainingJobId) &&
         Objects.equals(modelId, trainingCompletedResponse.modelId) &&
         Objects.equals(modelName, trainingCompletedResponse.modelName) &&
         Objects.equals(datasetId, trainingCompletedResponse.datasetId) &&
         Objects.equals(completedAt, trainingCompletedResponse.completedAt) &&
         Objects.equals(datasetStats, trainingCompletedResponse.datasetStats);
   }
 
   @Override
   public int hashCode() {
     return Objects.hash(status, trainingJobId, modelId, modelName, datasetId, completedAt,
         datasetStats);
   }
 
   @Override
   public String toString() {
     StringBuilder sb = new StringBuilder();
     sb.append("class TrainingCompletedResponse {\n");
 
     sb.append("    status: ").append(toIndentedString(status)).append("\n");
     sb.append("    trainingJobId: ").append(toIndentedString(trainingJobId)).append("\n");
     sb.append("    modelId: ").append(toIndentedString(modelId)).append("\n");
     sb.append("    modelName: ").append(toIndentedString(modelName)).append("\n");
     sb.append("    datasetId: ").append(toIndentedString(datasetId)).append("\n");
     sb.append("    completedAt: ").append(toIndentedString(completedAt)).append("\n");
     sb.append("    datasetStats: ").append(toIndentedString(datasetStats)).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    ");
   }
 }