SageMakerEndpoint.h

 
#pragma once
#include <aws/bedrock/Bedrock_EXPORTS.h>
#include <aws/bedrock/model/VpcConfig.h>
#include <aws/core/utils/memory/stl/AWSString.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace Bedrock {
namespace Model {
 
class SageMakerEndpoint {
 public:
  AWS_BEDROCK_API SageMakerEndpoint() = default;
  AWS_BEDROCK_API SageMakerEndpoint(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCK_API SageMakerEndpoint& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCK_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline int GetInitialInstanceCount() const { return m_initialInstanceCount; }
  inline bool InitialInstanceCountHasBeenSet() const { return m_initialInstanceCountHasBeenSet; }
  inline void SetInitialInstanceCount(int value) {
    m_initialInstanceCountHasBeenSet = true;
    m_initialInstanceCount = value;
  }
  inline SageMakerEndpoint& WithInitialInstanceCount(int value) {
    SetInitialInstanceCount(value);
    return *this;
  }
 
 
  inline const Aws::String& GetInstanceType() const { return m_instanceType; }
  inline bool InstanceTypeHasBeenSet() const { return m_instanceTypeHasBeenSet; }
  template <typename InstanceTypeT = Aws::String>
  void SetInstanceType(InstanceTypeT&& value) {
    m_instanceTypeHasBeenSet = true;
    m_instanceType = std::forward<InstanceTypeT>(value);
  }
  template <typename InstanceTypeT = Aws::String>
  SageMakerEndpoint& WithInstanceType(InstanceTypeT&& value) {
    SetInstanceType(std::forward<InstanceTypeT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetExecutionRole() const { return m_executionRole; }
  inline bool ExecutionRoleHasBeenSet() const { return m_executionRoleHasBeenSet; }
  template <typename ExecutionRoleT = Aws::String>
  void SetExecutionRole(ExecutionRoleT&& value) {
    m_executionRoleHasBeenSet = true;
    m_executionRole = std::forward<ExecutionRoleT>(value);
  }
  template <typename ExecutionRoleT = Aws::String>
  SageMakerEndpoint& WithExecutionRole(ExecutionRoleT&& value) {
    SetExecutionRole(std::forward<ExecutionRoleT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetKmsEncryptionKey() const { return m_kmsEncryptionKey; }
  inline bool KmsEncryptionKeyHasBeenSet() const { return m_kmsEncryptionKeyHasBeenSet; }
  template <typename KmsEncryptionKeyT = Aws::String>
  void SetKmsEncryptionKey(KmsEncryptionKeyT&& value) {
    m_kmsEncryptionKeyHasBeenSet = true;
    m_kmsEncryptionKey = std::forward<KmsEncryptionKeyT>(value);
  }
  template <typename KmsEncryptionKeyT = Aws::String>
  SageMakerEndpoint& WithKmsEncryptionKey(KmsEncryptionKeyT&& value) {
    SetKmsEncryptionKey(std::forward<KmsEncryptionKeyT>(value));
    return *this;
  }
 
 
  inline const VpcConfig& GetVpc() const { return m_vpc; }
  inline bool VpcHasBeenSet() const { return m_vpcHasBeenSet; }
  template <typename VpcT = VpcConfig>
  void SetVpc(VpcT&& value) {
    m_vpcHasBeenSet = true;
    m_vpc = std::forward<VpcT>(value);
  }
  template <typename VpcT = VpcConfig>
  SageMakerEndpoint& WithVpc(VpcT&& value) {
    SetVpc(std::forward<VpcT>(value));
    return *this;
  }
 private:
  int m_initialInstanceCount{0};
 
  Aws::String m_instanceType;
 
  Aws::String m_executionRole;
 
  Aws::String m_kmsEncryptionKey;
 
  VpcConfig m_vpc;
  bool m_initialInstanceCountHasBeenSet = false;
  bool m_instanceTypeHasBeenSet = false;
  bool m_executionRoleHasBeenSet = false;
  bool m_kmsEncryptionKeyHasBeenSet = false;
  bool m_vpcHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace Bedrock
}  // namespace Aws