InferenceComponentSpecification.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/InferenceComponentComputeResourceRequirements.h>
#include <aws/sagemaker/model/InferenceComponentContainerSpecification.h>
#include <aws/sagemaker/model/InferenceComponentDataCacheConfig.h>
#include <aws/sagemaker/model/InferenceComponentSchedulingConfig.h>
#include <aws/sagemaker/model/InferenceComponentStartupParameters.h>
#include <aws/sagemaker/model/ProductionVariantInstanceType.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class InferenceComponentSpecification {
 public:
  AWS_SAGEMAKER_API InferenceComponentSpecification() = default;
  AWS_SAGEMAKER_API InferenceComponentSpecification(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API InferenceComponentSpecification& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline ProductionVariantInstanceType GetInstanceType() const { return m_instanceType; }
  inline bool InstanceTypeHasBeenSet() const { return m_instanceTypeHasBeenSet; }
  inline void SetInstanceType(ProductionVariantInstanceType value) {
    m_instanceTypeHasBeenSet = true;
    m_instanceType = value;
  }
  inline InferenceComponentSpecification& WithInstanceType(ProductionVariantInstanceType value) {
    SetInstanceType(value);
    return *this;
  }
 
 
  inline const Aws::String& GetModelName() const { return m_modelName; }
  inline bool ModelNameHasBeenSet() const { return m_modelNameHasBeenSet; }
  template <typename ModelNameT = Aws::String>
  void SetModelName(ModelNameT&& value) {
    m_modelNameHasBeenSet = true;
    m_modelName = std::forward<ModelNameT>(value);
  }
  template <typename ModelNameT = Aws::String>
  InferenceComponentSpecification& WithModelName(ModelNameT&& value) {
    SetModelName(std::forward<ModelNameT>(value));
    return *this;
  }
 
 
  inline const InferenceComponentContainerSpecification& GetContainer() const { return m_container; }
  inline bool ContainerHasBeenSet() const { return m_containerHasBeenSet; }
  template <typename ContainerT = InferenceComponentContainerSpecification>
  void SetContainer(ContainerT&& value) {
    m_containerHasBeenSet = true;
    m_container = std::forward<ContainerT>(value);
  }
  template <typename ContainerT = InferenceComponentContainerSpecification>
  InferenceComponentSpecification& WithContainer(ContainerT&& value) {
    SetContainer(std::forward<ContainerT>(value));
    return *this;
  }
 
 
  inline const InferenceComponentStartupParameters& GetStartupParameters() const { return m_startupParameters; }
  inline bool StartupParametersHasBeenSet() const { return m_startupParametersHasBeenSet; }
  template <typename StartupParametersT = InferenceComponentStartupParameters>
  void SetStartupParameters(StartupParametersT&& value) {
    m_startupParametersHasBeenSet = true;
    m_startupParameters = std::forward<StartupParametersT>(value);
  }
  template <typename StartupParametersT = InferenceComponentStartupParameters>
  InferenceComponentSpecification& WithStartupParameters(StartupParametersT&& value) {
    SetStartupParameters(std::forward<StartupParametersT>(value));
    return *this;
  }
 
 
  inline const InferenceComponentComputeResourceRequirements& GetComputeResourceRequirements() const {
    return m_computeResourceRequirements;
  }
  inline bool ComputeResourceRequirementsHasBeenSet() const { return m_computeResourceRequirementsHasBeenSet; }
  template <typename ComputeResourceRequirementsT = InferenceComponentComputeResourceRequirements>
  void SetComputeResourceRequirements(ComputeResourceRequirementsT&& value) {
    m_computeResourceRequirementsHasBeenSet = true;
    m_computeResourceRequirements = std::forward<ComputeResourceRequirementsT>(value);
  }
  template <typename ComputeResourceRequirementsT = InferenceComponentComputeResourceRequirements>
  InferenceComponentSpecification& WithComputeResourceRequirements(ComputeResourceRequirementsT&& value) {
    SetComputeResourceRequirements(std::forward<ComputeResourceRequirementsT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetBaseInferenceComponentName() const { return m_baseInferenceComponentName; }
  inline bool BaseInferenceComponentNameHasBeenSet() const { return m_baseInferenceComponentNameHasBeenSet; }
  template <typename BaseInferenceComponentNameT = Aws::String>
  void SetBaseInferenceComponentName(BaseInferenceComponentNameT&& value) {
    m_baseInferenceComponentNameHasBeenSet = true;
    m_baseInferenceComponentName = std::forward<BaseInferenceComponentNameT>(value);
  }
  template <typename BaseInferenceComponentNameT = Aws::String>
  InferenceComponentSpecification& WithBaseInferenceComponentName(BaseInferenceComponentNameT&& value) {
    SetBaseInferenceComponentName(std::forward<BaseInferenceComponentNameT>(value));
    return *this;
  }
 
 
  inline const InferenceComponentDataCacheConfig& GetDataCacheConfig() const { return m_dataCacheConfig; }
  inline bool DataCacheConfigHasBeenSet() const { return m_dataCacheConfigHasBeenSet; }
  template <typename DataCacheConfigT = InferenceComponentDataCacheConfig>
  void SetDataCacheConfig(DataCacheConfigT&& value) {
    m_dataCacheConfigHasBeenSet = true;
    m_dataCacheConfig = std::forward<DataCacheConfigT>(value);
  }
  template <typename DataCacheConfigT = InferenceComponentDataCacheConfig>
  InferenceComponentSpecification& WithDataCacheConfig(DataCacheConfigT&& value) {
    SetDataCacheConfig(std::forward<DataCacheConfigT>(value));
    return *this;
  }
 
 
  inline const InferenceComponentSchedulingConfig& GetSchedulingConfig() const { return m_schedulingConfig; }
  inline bool SchedulingConfigHasBeenSet() const { return m_schedulingConfigHasBeenSet; }
  template <typename SchedulingConfigT = InferenceComponentSchedulingConfig>
  void SetSchedulingConfig(SchedulingConfigT&& value) {
    m_schedulingConfigHasBeenSet = true;
    m_schedulingConfig = std::forward<SchedulingConfigT>(value);
  }
  template <typename SchedulingConfigT = InferenceComponentSchedulingConfig>
  InferenceComponentSpecification& WithSchedulingConfig(SchedulingConfigT&& value) {
    SetSchedulingConfig(std::forward<SchedulingConfigT>(value));
    return *this;
  }
 private:
  ProductionVariantInstanceType m_instanceType{ProductionVariantInstanceType::NOT_SET};
 
  Aws::String m_modelName;
 
  InferenceComponentContainerSpecification m_container;
 
  InferenceComponentStartupParameters m_startupParameters;
 
  InferenceComponentComputeResourceRequirements m_computeResourceRequirements;
 
  Aws::String m_baseInferenceComponentName;
 
  InferenceComponentDataCacheConfig m_dataCacheConfig;
 
  InferenceComponentSchedulingConfig m_schedulingConfig;
  bool m_instanceTypeHasBeenSet = false;
  bool m_modelNameHasBeenSet = false;
  bool m_containerHasBeenSet = false;
  bool m_startupParametersHasBeenSet = false;
  bool m_computeResourceRequirementsHasBeenSet = false;
  bool m_baseInferenceComponentNameHasBeenSet = false;
  bool m_dataCacheConfigHasBeenSet = false;
  bool m_schedulingConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws