InvokeEndpointRequest.h

 
#pragma once
#include <aws/core/utils/Array.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker-runtime/SageMakerRuntimeRequest.h>
#include <aws/sagemaker-runtime/SageMakerRuntime_EXPORTS.h>
 
#include <utility>
 
namespace Aws {
namespace SageMakerRuntime {
namespace Model {
 
class InvokeEndpointRequest : public StreamingSageMakerRuntimeRequest {
 public:
  AWS_SAGEMAKERRUNTIME_API InvokeEndpointRequest() = 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 "InvokeEndpoint"; }
 
  AWS_SAGEMAKERRUNTIME_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetEndpointName() const { return m_endpointName; }
  inline bool EndpointNameHasBeenSet() const { return m_endpointNameHasBeenSet; }
  template <typename EndpointNameT = Aws::String>
  void SetEndpointName(EndpointNameT&& value) {
    m_endpointNameHasBeenSet = true;
    m_endpointName = std::forward<EndpointNameT>(value);
  }
  template <typename EndpointNameT = Aws::String>
  InvokeEndpointRequest& WithEndpointName(EndpointNameT&& value) {
    SetEndpointName(std::forward<EndpointNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetAccept() const { return m_accept; }
  inline bool AcceptHasBeenSet() const { return m_acceptHasBeenSet; }
  template <typename AcceptT = Aws::String>
  void SetAccept(AcceptT&& value) {
    m_acceptHasBeenSet = true;
    m_accept = std::forward<AcceptT>(value);
  }
  template <typename AcceptT = Aws::String>
  InvokeEndpointRequest& WithAccept(AcceptT&& value) {
    SetAccept(std::forward<AcceptT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetCustomAttributes() const { return m_customAttributes; }
  inline bool CustomAttributesHasBeenSet() const { return m_customAttributesHasBeenSet; }
  template <typename CustomAttributesT = Aws::String>
  void SetCustomAttributes(CustomAttributesT&& value) {
    m_customAttributesHasBeenSet = true;
    m_customAttributes = std::forward<CustomAttributesT>(value);
  }
  template <typename CustomAttributesT = Aws::String>
  InvokeEndpointRequest& WithCustomAttributes(CustomAttributesT&& value) {
    SetCustomAttributes(std::forward<CustomAttributesT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTargetModel() const { return m_targetModel; }
  inline bool TargetModelHasBeenSet() const { return m_targetModelHasBeenSet; }
  template <typename TargetModelT = Aws::String>
  void SetTargetModel(TargetModelT&& value) {
    m_targetModelHasBeenSet = true;
    m_targetModel = std::forward<TargetModelT>(value);
  }
  template <typename TargetModelT = Aws::String>
  InvokeEndpointRequest& WithTargetModel(TargetModelT&& value) {
    SetTargetModel(std::forward<TargetModelT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTargetVariant() const { return m_targetVariant; }
  inline bool TargetVariantHasBeenSet() const { return m_targetVariantHasBeenSet; }
  template <typename TargetVariantT = Aws::String>
  void SetTargetVariant(TargetVariantT&& value) {
    m_targetVariantHasBeenSet = true;
    m_targetVariant = std::forward<TargetVariantT>(value);
  }
  template <typename TargetVariantT = Aws::String>
  InvokeEndpointRequest& WithTargetVariant(TargetVariantT&& value) {
    SetTargetVariant(std::forward<TargetVariantT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTargetContainerHostname() const { return m_targetContainerHostname; }
  inline bool TargetContainerHostnameHasBeenSet() const { return m_targetContainerHostnameHasBeenSet; }
  template <typename TargetContainerHostnameT = Aws::String>
  void SetTargetContainerHostname(TargetContainerHostnameT&& value) {
    m_targetContainerHostnameHasBeenSet = true;
    m_targetContainerHostname = std::forward<TargetContainerHostnameT>(value);
  }
  template <typename TargetContainerHostnameT = Aws::String>
  InvokeEndpointRequest& WithTargetContainerHostname(TargetContainerHostnameT&& value) {
    SetTargetContainerHostname(std::forward<TargetContainerHostnameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetInferenceId() const { return m_inferenceId; }
  inline bool InferenceIdHasBeenSet() const { return m_inferenceIdHasBeenSet; }
  template <typename InferenceIdT = Aws::String>
  void SetInferenceId(InferenceIdT&& value) {
    m_inferenceIdHasBeenSet = true;
    m_inferenceId = std::forward<InferenceIdT>(value);
  }
  template <typename InferenceIdT = Aws::String>
  InvokeEndpointRequest& WithInferenceId(InferenceIdT&& value) {
    SetInferenceId(std::forward<InferenceIdT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetEnableExplanations() const { return m_enableExplanations; }
  inline bool EnableExplanationsHasBeenSet() const { return m_enableExplanationsHasBeenSet; }
  template <typename EnableExplanationsT = Aws::String>
  void SetEnableExplanations(EnableExplanationsT&& value) {
    m_enableExplanationsHasBeenSet = true;
    m_enableExplanations = std::forward<EnableExplanationsT>(value);
  }
  template <typename EnableExplanationsT = Aws::String>
  InvokeEndpointRequest& WithEnableExplanations(EnableExplanationsT&& value) {
    SetEnableExplanations(std::forward<EnableExplanationsT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetInferenceComponentName() const { return m_inferenceComponentName; }
  inline bool InferenceComponentNameHasBeenSet() const { return m_inferenceComponentNameHasBeenSet; }
  template <typename InferenceComponentNameT = Aws::String>
  void SetInferenceComponentName(InferenceComponentNameT&& value) {
    m_inferenceComponentNameHasBeenSet = true;
    m_inferenceComponentName = std::forward<InferenceComponentNameT>(value);
  }
  template <typename InferenceComponentNameT = Aws::String>
  InvokeEndpointRequest& WithInferenceComponentName(InferenceComponentNameT&& value) {
    SetInferenceComponentName(std::forward<InferenceComponentNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetSessionId() const { return m_sessionId; }
  inline bool SessionIdHasBeenSet() const { return m_sessionIdHasBeenSet; }
  template <typename SessionIdT = Aws::String>
  void SetSessionId(SessionIdT&& value) {
    m_sessionIdHasBeenSet = true;
    m_sessionId = std::forward<SessionIdT>(value);
  }
  template <typename SessionIdT = Aws::String>
  InvokeEndpointRequest& WithSessionId(SessionIdT&& value) {
    SetSessionId(std::forward<SessionIdT>(value));
    return *this;
  }
 private:
  Aws::String m_endpointName;
 
  Aws::String m_accept;
 
  Aws::String m_customAttributes;
 
  Aws::String m_targetModel;
 
  Aws::String m_targetVariant;
 
  Aws::String m_targetContainerHostname;
 
  Aws::String m_inferenceId;
 
  Aws::String m_enableExplanations;
 
  Aws::String m_inferenceComponentName;
 
  Aws::String m_sessionId;
  bool m_endpointNameHasBeenSet = false;
  bool m_acceptHasBeenSet = false;
  bool m_customAttributesHasBeenSet = false;
  bool m_targetModelHasBeenSet = false;
  bool m_targetVariantHasBeenSet = false;
  bool m_targetContainerHostnameHasBeenSet = false;
  bool m_inferenceIdHasBeenSet = false;
  bool m_enableExplanationsHasBeenSet = false;
  bool m_inferenceComponentNameHasBeenSet = false;
  bool m_sessionIdHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMakerRuntime
}  // namespace Aws