CreateModelPackageRequest.h

 
#pragma once
#include <aws/core/utils/UUID.h>
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMakerRequest.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/AdditionalInferenceSpecificationDefinition.h>
#include <aws/sagemaker/model/DriftCheckBaselines.h>
#include <aws/sagemaker/model/InferenceSpecification.h>
#include <aws/sagemaker/model/ManagedStorageType.h>
#include <aws/sagemaker/model/MetadataProperties.h>
#include <aws/sagemaker/model/ModelApprovalStatus.h>
#include <aws/sagemaker/model/ModelLifeCycle.h>
#include <aws/sagemaker/model/ModelMetrics.h>
#include <aws/sagemaker/model/ModelPackageModelCard.h>
#include <aws/sagemaker/model/ModelPackageRegistrationType.h>
#include <aws/sagemaker/model/ModelPackageSecurityConfig.h>
#include <aws/sagemaker/model/ModelPackageValidationSpecification.h>
#include <aws/sagemaker/model/SkipModelValidation.h>
#include <aws/sagemaker/model/SourceAlgorithmSpecification.h>
#include <aws/sagemaker/model/Tag.h>
 
#include <utility>
 
namespace Aws {
namespace SageMaker {
namespace Model {
 
class CreateModelPackageRequest : public SageMakerRequest {
 public:
  AWS_SAGEMAKER_API CreateModelPackageRequest() = default;
 
  // Service request name is the Operation name which will send this request out,
  // each operation should has unique request name, so that we can get operation's name from this request.
  // Note: this is not true for response, multiple operations may have the same response name,
  // so we can not get operation's name from response.
  inline virtual const char* GetServiceRequestName() const override { return "CreateModelPackage"; }
 
  AWS_SAGEMAKER_API Aws::String SerializePayload() const override;
 
  AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetModelPackageName() const { return m_modelPackageName; }
  inline bool ModelPackageNameHasBeenSet() const { return m_modelPackageNameHasBeenSet; }
  template <typename ModelPackageNameT = Aws::String>
  void SetModelPackageName(ModelPackageNameT&& value) {
    m_modelPackageNameHasBeenSet = true;
    m_modelPackageName = std::forward<ModelPackageNameT>(value);
  }
  template <typename ModelPackageNameT = Aws::String>
  CreateModelPackageRequest& WithModelPackageName(ModelPackageNameT&& value) {
    SetModelPackageName(std::forward<ModelPackageNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetModelPackageGroupName() const { return m_modelPackageGroupName; }
  inline bool ModelPackageGroupNameHasBeenSet() const { return m_modelPackageGroupNameHasBeenSet; }
  template <typename ModelPackageGroupNameT = Aws::String>
  void SetModelPackageGroupName(ModelPackageGroupNameT&& value) {
    m_modelPackageGroupNameHasBeenSet = true;
    m_modelPackageGroupName = std::forward<ModelPackageGroupNameT>(value);
  }
  template <typename ModelPackageGroupNameT = Aws::String>
  CreateModelPackageRequest& WithModelPackageGroupName(ModelPackageGroupNameT&& value) {
    SetModelPackageGroupName(std::forward<ModelPackageGroupNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetModelPackageDescription() const { return m_modelPackageDescription; }
  inline bool ModelPackageDescriptionHasBeenSet() const { return m_modelPackageDescriptionHasBeenSet; }
  template <typename ModelPackageDescriptionT = Aws::String>
  void SetModelPackageDescription(ModelPackageDescriptionT&& value) {
    m_modelPackageDescriptionHasBeenSet = true;
    m_modelPackageDescription = std::forward<ModelPackageDescriptionT>(value);
  }
  template <typename ModelPackageDescriptionT = Aws::String>
  CreateModelPackageRequest& WithModelPackageDescription(ModelPackageDescriptionT&& value) {
    SetModelPackageDescription(std::forward<ModelPackageDescriptionT>(value));
    return *this;
  }
 
 
  inline ModelPackageRegistrationType GetModelPackageRegistrationType() const { return m_modelPackageRegistrationType; }
  inline bool ModelPackageRegistrationTypeHasBeenSet() const { return m_modelPackageRegistrationTypeHasBeenSet; }
  inline void SetModelPackageRegistrationType(ModelPackageRegistrationType value) {
    m_modelPackageRegistrationTypeHasBeenSet = true;
    m_modelPackageRegistrationType = value;
  }
  inline CreateModelPackageRequest& WithModelPackageRegistrationType(ModelPackageRegistrationType value) {
    SetModelPackageRegistrationType(value);
    return *this;
  }
 
 
  inline const InferenceSpecification& GetInferenceSpecification() const { return m_inferenceSpecification; }
  inline bool InferenceSpecificationHasBeenSet() const { return m_inferenceSpecificationHasBeenSet; }
  template <typename InferenceSpecificationT = InferenceSpecification>
  void SetInferenceSpecification(InferenceSpecificationT&& value) {
    m_inferenceSpecificationHasBeenSet = true;
    m_inferenceSpecification = std::forward<InferenceSpecificationT>(value);
  }
  template <typename InferenceSpecificationT = InferenceSpecification>
  CreateModelPackageRequest& WithInferenceSpecification(InferenceSpecificationT&& value) {
    SetInferenceSpecification(std::forward<InferenceSpecificationT>(value));
    return *this;
  }
 
 
  inline const ModelPackageValidationSpecification& GetValidationSpecification() const { return m_validationSpecification; }
  inline bool ValidationSpecificationHasBeenSet() const { return m_validationSpecificationHasBeenSet; }
  template <typename ValidationSpecificationT = ModelPackageValidationSpecification>
  void SetValidationSpecification(ValidationSpecificationT&& value) {
    m_validationSpecificationHasBeenSet = true;
    m_validationSpecification = std::forward<ValidationSpecificationT>(value);
  }
  template <typename ValidationSpecificationT = ModelPackageValidationSpecification>
  CreateModelPackageRequest& WithValidationSpecification(ValidationSpecificationT&& value) {
    SetValidationSpecification(std::forward<ValidationSpecificationT>(value));
    return *this;
  }
 
 
  inline const SourceAlgorithmSpecification& GetSourceAlgorithmSpecification() const { return m_sourceAlgorithmSpecification; }
  inline bool SourceAlgorithmSpecificationHasBeenSet() const { return m_sourceAlgorithmSpecificationHasBeenSet; }
  template <typename SourceAlgorithmSpecificationT = SourceAlgorithmSpecification>
  void SetSourceAlgorithmSpecification(SourceAlgorithmSpecificationT&& value) {
    m_sourceAlgorithmSpecificationHasBeenSet = true;
    m_sourceAlgorithmSpecification = std::forward<SourceAlgorithmSpecificationT>(value);
  }
  template <typename SourceAlgorithmSpecificationT = SourceAlgorithmSpecification>
  CreateModelPackageRequest& WithSourceAlgorithmSpecification(SourceAlgorithmSpecificationT&& value) {
    SetSourceAlgorithmSpecification(std::forward<SourceAlgorithmSpecificationT>(value));
    return *this;
  }
 
 
  inline bool GetCertifyForMarketplace() const { return m_certifyForMarketplace; }
  inline bool CertifyForMarketplaceHasBeenSet() const { return m_certifyForMarketplaceHasBeenSet; }
  inline void SetCertifyForMarketplace(bool value) {
    m_certifyForMarketplaceHasBeenSet = true;
    m_certifyForMarketplace = value;
  }
  inline CreateModelPackageRequest& WithCertifyForMarketplace(bool value) {
    SetCertifyForMarketplace(value);
    return *this;
  }
 
 
  inline const Aws::Vector<Tag>& GetTags() const { return m_tags; }
  inline bool TagsHasBeenSet() const { return m_tagsHasBeenSet; }
  template <typename TagsT = Aws::Vector<Tag>>
  void SetTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags = std::forward<TagsT>(value);
  }
  template <typename TagsT = Aws::Vector<Tag>>
  CreateModelPackageRequest& WithTags(TagsT&& value) {
    SetTags(std::forward<TagsT>(value));
    return *this;
  }
  template <typename TagsT = Tag>
  CreateModelPackageRequest& AddTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags.emplace_back(std::forward<TagsT>(value));
    return *this;
  }
 
 
  inline ModelApprovalStatus GetModelApprovalStatus() const { return m_modelApprovalStatus; }
  inline bool ModelApprovalStatusHasBeenSet() const { return m_modelApprovalStatusHasBeenSet; }
  inline void SetModelApprovalStatus(ModelApprovalStatus value) {
    m_modelApprovalStatusHasBeenSet = true;
    m_modelApprovalStatus = value;
  }
  inline CreateModelPackageRequest& WithModelApprovalStatus(ModelApprovalStatus value) {
    SetModelApprovalStatus(value);
    return *this;
  }
 
 
  inline const MetadataProperties& GetMetadataProperties() const { return m_metadataProperties; }
  inline bool MetadataPropertiesHasBeenSet() const { return m_metadataPropertiesHasBeenSet; }
  template <typename MetadataPropertiesT = MetadataProperties>
  void SetMetadataProperties(MetadataPropertiesT&& value) {
    m_metadataPropertiesHasBeenSet = true;
    m_metadataProperties = std::forward<MetadataPropertiesT>(value);
  }
  template <typename MetadataPropertiesT = MetadataProperties>
  CreateModelPackageRequest& WithMetadataProperties(MetadataPropertiesT&& value) {
    SetMetadataProperties(std::forward<MetadataPropertiesT>(value));
    return *this;
  }
 
 
  inline const ModelMetrics& GetModelMetrics() const { return m_modelMetrics; }
  inline bool ModelMetricsHasBeenSet() const { return m_modelMetricsHasBeenSet; }
  template <typename ModelMetricsT = ModelMetrics>
  void SetModelMetrics(ModelMetricsT&& value) {
    m_modelMetricsHasBeenSet = true;
    m_modelMetrics = std::forward<ModelMetricsT>(value);
  }
  template <typename ModelMetricsT = ModelMetrics>
  CreateModelPackageRequest& WithModelMetrics(ModelMetricsT&& value) {
    SetModelMetrics(std::forward<ModelMetricsT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetClientToken() const { return m_clientToken; }
  inline bool ClientTokenHasBeenSet() const { return m_clientTokenHasBeenSet; }
  template <typename ClientTokenT = Aws::String>
  void SetClientToken(ClientTokenT&& value) {
    m_clientTokenHasBeenSet = true;
    m_clientToken = std::forward<ClientTokenT>(value);
  }
  template <typename ClientTokenT = Aws::String>
  CreateModelPackageRequest& WithClientToken(ClientTokenT&& value) {
    SetClientToken(std::forward<ClientTokenT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetDomain() const { return m_domain; }
  inline bool DomainHasBeenSet() const { return m_domainHasBeenSet; }
  template <typename DomainT = Aws::String>
  void SetDomain(DomainT&& value) {
    m_domainHasBeenSet = true;
    m_domain = std::forward<DomainT>(value);
  }
  template <typename DomainT = Aws::String>
  CreateModelPackageRequest& WithDomain(DomainT&& value) {
    SetDomain(std::forward<DomainT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTask() const { return m_task; }
  inline bool TaskHasBeenSet() const { return m_taskHasBeenSet; }
  template <typename TaskT = Aws::String>
  void SetTask(TaskT&& value) {
    m_taskHasBeenSet = true;
    m_task = std::forward<TaskT>(value);
  }
  template <typename TaskT = Aws::String>
  CreateModelPackageRequest& WithTask(TaskT&& value) {
    SetTask(std::forward<TaskT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetSamplePayloadUrl() const { return m_samplePayloadUrl; }
  inline bool SamplePayloadUrlHasBeenSet() const { return m_samplePayloadUrlHasBeenSet; }
  template <typename SamplePayloadUrlT = Aws::String>
  void SetSamplePayloadUrl(SamplePayloadUrlT&& value) {
    m_samplePayloadUrlHasBeenSet = true;
    m_samplePayloadUrl = std::forward<SamplePayloadUrlT>(value);
  }
  template <typename SamplePayloadUrlT = Aws::String>
  CreateModelPackageRequest& WithSamplePayloadUrl(SamplePayloadUrlT&& value) {
    SetSamplePayloadUrl(std::forward<SamplePayloadUrlT>(value));
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, Aws::String>& GetCustomerMetadataProperties() const { return m_customerMetadataProperties; }
  inline bool CustomerMetadataPropertiesHasBeenSet() const { return m_customerMetadataPropertiesHasBeenSet; }
  template <typename CustomerMetadataPropertiesT = Aws::Map<Aws::String, Aws::String>>
  void SetCustomerMetadataProperties(CustomerMetadataPropertiesT&& value) {
    m_customerMetadataPropertiesHasBeenSet = true;
    m_customerMetadataProperties = std::forward<CustomerMetadataPropertiesT>(value);
  }
  template <typename CustomerMetadataPropertiesT = Aws::Map<Aws::String, Aws::String>>
  CreateModelPackageRequest& WithCustomerMetadataProperties(CustomerMetadataPropertiesT&& value) {
    SetCustomerMetadataProperties(std::forward<CustomerMetadataPropertiesT>(value));
    return *this;
  }
  template <typename CustomerMetadataPropertiesKeyT = Aws::String, typename CustomerMetadataPropertiesValueT = Aws::String>
  CreateModelPackageRequest& AddCustomerMetadataProperties(CustomerMetadataPropertiesKeyT&& key, CustomerMetadataPropertiesValueT&& value) {
    m_customerMetadataPropertiesHasBeenSet = true;
    m_customerMetadataProperties.emplace(std::forward<CustomerMetadataPropertiesKeyT>(key),
                                         std::forward<CustomerMetadataPropertiesValueT>(value));
    return *this;
  }
 
 
  inline const DriftCheckBaselines& GetDriftCheckBaselines() const { return m_driftCheckBaselines; }
  inline bool DriftCheckBaselinesHasBeenSet() const { return m_driftCheckBaselinesHasBeenSet; }
  template <typename DriftCheckBaselinesT = DriftCheckBaselines>
  void SetDriftCheckBaselines(DriftCheckBaselinesT&& value) {
    m_driftCheckBaselinesHasBeenSet = true;
    m_driftCheckBaselines = std::forward<DriftCheckBaselinesT>(value);
  }
  template <typename DriftCheckBaselinesT = DriftCheckBaselines>
  CreateModelPackageRequest& WithDriftCheckBaselines(DriftCheckBaselinesT&& value) {
    SetDriftCheckBaselines(std::forward<DriftCheckBaselinesT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<AdditionalInferenceSpecificationDefinition>& GetAdditionalInferenceSpecifications() const {
    return m_additionalInferenceSpecifications;
  }
  inline bool AdditionalInferenceSpecificationsHasBeenSet() const { return m_additionalInferenceSpecificationsHasBeenSet; }
  template <typename AdditionalInferenceSpecificationsT = Aws::Vector<AdditionalInferenceSpecificationDefinition>>
  void SetAdditionalInferenceSpecifications(AdditionalInferenceSpecificationsT&& value) {
    m_additionalInferenceSpecificationsHasBeenSet = true;
    m_additionalInferenceSpecifications = std::forward<AdditionalInferenceSpecificationsT>(value);
  }
  template <typename AdditionalInferenceSpecificationsT = Aws::Vector<AdditionalInferenceSpecificationDefinition>>
  CreateModelPackageRequest& WithAdditionalInferenceSpecifications(AdditionalInferenceSpecificationsT&& value) {
    SetAdditionalInferenceSpecifications(std::forward<AdditionalInferenceSpecificationsT>(value));
    return *this;
  }
  template <typename AdditionalInferenceSpecificationsT = AdditionalInferenceSpecificationDefinition>
  CreateModelPackageRequest& AddAdditionalInferenceSpecifications(AdditionalInferenceSpecificationsT&& value) {
    m_additionalInferenceSpecificationsHasBeenSet = true;
    m_additionalInferenceSpecifications.emplace_back(std::forward<AdditionalInferenceSpecificationsT>(value));
    return *this;
  }
 
 
  inline SkipModelValidation GetSkipModelValidation() const { return m_skipModelValidation; }
  inline bool SkipModelValidationHasBeenSet() const { return m_skipModelValidationHasBeenSet; }
  inline void SetSkipModelValidation(SkipModelValidation value) {
    m_skipModelValidationHasBeenSet = true;
    m_skipModelValidation = value;
  }
  inline CreateModelPackageRequest& WithSkipModelValidation(SkipModelValidation value) {
    SetSkipModelValidation(value);
    return *this;
  }
 
 
  inline const Aws::String& GetSourceUri() const { return m_sourceUri; }
  inline bool SourceUriHasBeenSet() const { return m_sourceUriHasBeenSet; }
  template <typename SourceUriT = Aws::String>
  void SetSourceUri(SourceUriT&& value) {
    m_sourceUriHasBeenSet = true;
    m_sourceUri = std::forward<SourceUriT>(value);
  }
  template <typename SourceUriT = Aws::String>
  CreateModelPackageRequest& WithSourceUri(SourceUriT&& value) {
    SetSourceUri(std::forward<SourceUriT>(value));
    return *this;
  }
 
 
  inline const ModelPackageSecurityConfig& GetSecurityConfig() const { return m_securityConfig; }
  inline bool SecurityConfigHasBeenSet() const { return m_securityConfigHasBeenSet; }
  template <typename SecurityConfigT = ModelPackageSecurityConfig>
  void SetSecurityConfig(SecurityConfigT&& value) {
    m_securityConfigHasBeenSet = true;
    m_securityConfig = std::forward<SecurityConfigT>(value);
  }
  template <typename SecurityConfigT = ModelPackageSecurityConfig>
  CreateModelPackageRequest& WithSecurityConfig(SecurityConfigT&& value) {
    SetSecurityConfig(std::forward<SecurityConfigT>(value));
    return *this;
  }
 
 
  inline const ModelPackageModelCard& GetModelCard() const { return m_modelCard; }
  inline bool ModelCardHasBeenSet() const { return m_modelCardHasBeenSet; }
  template <typename ModelCardT = ModelPackageModelCard>
  void SetModelCard(ModelCardT&& value) {
    m_modelCardHasBeenSet = true;
    m_modelCard = std::forward<ModelCardT>(value);
  }
  template <typename ModelCardT = ModelPackageModelCard>
  CreateModelPackageRequest& WithModelCard(ModelCardT&& value) {
    SetModelCard(std::forward<ModelCardT>(value));
    return *this;
  }
 
 
  inline const ModelLifeCycle& GetModelLifeCycle() const { return m_modelLifeCycle; }
  inline bool ModelLifeCycleHasBeenSet() const { return m_modelLifeCycleHasBeenSet; }
  template <typename ModelLifeCycleT = ModelLifeCycle>
  void SetModelLifeCycle(ModelLifeCycleT&& value) {
    m_modelLifeCycleHasBeenSet = true;
    m_modelLifeCycle = std::forward<ModelLifeCycleT>(value);
  }
  template <typename ModelLifeCycleT = ModelLifeCycle>
  CreateModelPackageRequest& WithModelLifeCycle(ModelLifeCycleT&& value) {
    SetModelLifeCycle(std::forward<ModelLifeCycleT>(value));
    return *this;
  }
 
 
  inline ManagedStorageType GetManagedStorageType() const { return m_managedStorageType; }
  inline bool ManagedStorageTypeHasBeenSet() const { return m_managedStorageTypeHasBeenSet; }
  inline void SetManagedStorageType(ManagedStorageType value) {
    m_managedStorageTypeHasBeenSet = true;
    m_managedStorageType = value;
  }
  inline CreateModelPackageRequest& WithManagedStorageType(ManagedStorageType value) {
    SetManagedStorageType(value);
    return *this;
  }
 private:
  Aws::String m_modelPackageName;
 
  Aws::String m_modelPackageGroupName;
 
  Aws::String m_modelPackageDescription;
 
  ModelPackageRegistrationType m_modelPackageRegistrationType{ModelPackageRegistrationType::NOT_SET};
 
  InferenceSpecification m_inferenceSpecification;
 
  ModelPackageValidationSpecification m_validationSpecification;
 
  SourceAlgorithmSpecification m_sourceAlgorithmSpecification;
 
  bool m_certifyForMarketplace{false};
 
  Aws::Vector<Tag> m_tags;
 
  ModelApprovalStatus m_modelApprovalStatus{ModelApprovalStatus::NOT_SET};
 
  MetadataProperties m_metadataProperties;
 
  ModelMetrics m_modelMetrics;
 
  Aws::String m_clientToken{Aws::Utils::UUID::PseudoRandomUUID()};
 
  Aws::String m_domain;
 
  Aws::String m_task;
 
  Aws::String m_samplePayloadUrl;
 
  Aws::Map<Aws::String, Aws::String> m_customerMetadataProperties;
 
  DriftCheckBaselines m_driftCheckBaselines;
 
  Aws::Vector<AdditionalInferenceSpecificationDefinition> m_additionalInferenceSpecifications;
 
  SkipModelValidation m_skipModelValidation{SkipModelValidation::NOT_SET};
 
  Aws::String m_sourceUri;
 
  ModelPackageSecurityConfig m_securityConfig;
 
  ModelPackageModelCard m_modelCard;
 
  ModelLifeCycle m_modelLifeCycle;
 
  ManagedStorageType m_managedStorageType{ManagedStorageType::NOT_SET};
  bool m_modelPackageNameHasBeenSet = false;
  bool m_modelPackageGroupNameHasBeenSet = false;
  bool m_modelPackageDescriptionHasBeenSet = false;
  bool m_modelPackageRegistrationTypeHasBeenSet = false;
  bool m_inferenceSpecificationHasBeenSet = false;
  bool m_validationSpecificationHasBeenSet = false;
  bool m_sourceAlgorithmSpecificationHasBeenSet = false;
  bool m_certifyForMarketplaceHasBeenSet = false;
  bool m_tagsHasBeenSet = false;
  bool m_modelApprovalStatusHasBeenSet = false;
  bool m_metadataPropertiesHasBeenSet = false;
  bool m_modelMetricsHasBeenSet = false;
  bool m_clientTokenHasBeenSet = true;
  bool m_domainHasBeenSet = false;
  bool m_taskHasBeenSet = false;
  bool m_samplePayloadUrlHasBeenSet = false;
  bool m_customerMetadataPropertiesHasBeenSet = false;
  bool m_driftCheckBaselinesHasBeenSet = false;
  bool m_additionalInferenceSpecificationsHasBeenSet = false;
  bool m_skipModelValidationHasBeenSet = false;
  bool m_sourceUriHasBeenSet = false;
  bool m_securityConfigHasBeenSet = false;
  bool m_modelCardHasBeenSet = false;
  bool m_modelLifeCycleHasBeenSet = false;
  bool m_managedStorageTypeHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws