RetrieveRequest.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/kendra/KendraRequest.h>
#include <aws/kendra/Kendra_EXPORTS.h>
#include <aws/kendra/model/AttributeFilter.h>
#include <aws/kendra/model/DocumentRelevanceConfiguration.h>
#include <aws/kendra/model/UserContext.h>
 
#include <utility>
 
namespace Aws {
namespace kendra {
namespace Model {
 
class RetrieveRequest : public KendraRequest {
 public:
  AWS_KENDRA_API RetrieveRequest() = 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 "Retrieve"; }
 
  AWS_KENDRA_API Aws::String SerializePayload() const override;
 
  AWS_KENDRA_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetIndexId() const { return m_indexId; }
  inline bool IndexIdHasBeenSet() const { return m_indexIdHasBeenSet; }
  template <typename IndexIdT = Aws::String>
  void SetIndexId(IndexIdT&& value) {
    m_indexIdHasBeenSet = true;
    m_indexId = std::forward<IndexIdT>(value);
  }
  template <typename IndexIdT = Aws::String>
  RetrieveRequest& WithIndexId(IndexIdT&& value) {
    SetIndexId(std::forward<IndexIdT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetQueryText() const { return m_queryText; }
  inline bool QueryTextHasBeenSet() const { return m_queryTextHasBeenSet; }
  template <typename QueryTextT = Aws::String>
  void SetQueryText(QueryTextT&& value) {
    m_queryTextHasBeenSet = true;
    m_queryText = std::forward<QueryTextT>(value);
  }
  template <typename QueryTextT = Aws::String>
  RetrieveRequest& WithQueryText(QueryTextT&& value) {
    SetQueryText(std::forward<QueryTextT>(value));
    return *this;
  }
 
 
  inline const AttributeFilter& GetAttributeFilter() const { return m_attributeFilter; }
  inline bool AttributeFilterHasBeenSet() const { return m_attributeFilterHasBeenSet; }
  template <typename AttributeFilterT = AttributeFilter>
  void SetAttributeFilter(AttributeFilterT&& value) {
    m_attributeFilterHasBeenSet = true;
    m_attributeFilter = std::forward<AttributeFilterT>(value);
  }
  template <typename AttributeFilterT = AttributeFilter>
  RetrieveRequest& WithAttributeFilter(AttributeFilterT&& value) {
    SetAttributeFilter(std::forward<AttributeFilterT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetRequestedDocumentAttributes() const { return m_requestedDocumentAttributes; }
  inline bool RequestedDocumentAttributesHasBeenSet() const { return m_requestedDocumentAttributesHasBeenSet; }
  template <typename RequestedDocumentAttributesT = Aws::Vector<Aws::String>>
  void SetRequestedDocumentAttributes(RequestedDocumentAttributesT&& value) {
    m_requestedDocumentAttributesHasBeenSet = true;
    m_requestedDocumentAttributes = std::forward<RequestedDocumentAttributesT>(value);
  }
  template <typename RequestedDocumentAttributesT = Aws::Vector<Aws::String>>
  RetrieveRequest& WithRequestedDocumentAttributes(RequestedDocumentAttributesT&& value) {
    SetRequestedDocumentAttributes(std::forward<RequestedDocumentAttributesT>(value));
    return *this;
  }
  template <typename RequestedDocumentAttributesT = Aws::String>
  RetrieveRequest& AddRequestedDocumentAttributes(RequestedDocumentAttributesT&& value) {
    m_requestedDocumentAttributesHasBeenSet = true;
    m_requestedDocumentAttributes.emplace_back(std::forward<RequestedDocumentAttributesT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<DocumentRelevanceConfiguration>& GetDocumentRelevanceOverrideConfigurations() const {
    return m_documentRelevanceOverrideConfigurations;
  }
  inline bool DocumentRelevanceOverrideConfigurationsHasBeenSet() const { return m_documentRelevanceOverrideConfigurationsHasBeenSet; }
  template <typename DocumentRelevanceOverrideConfigurationsT = Aws::Vector<DocumentRelevanceConfiguration>>
  void SetDocumentRelevanceOverrideConfigurations(DocumentRelevanceOverrideConfigurationsT&& value) {
    m_documentRelevanceOverrideConfigurationsHasBeenSet = true;
    m_documentRelevanceOverrideConfigurations = std::forward<DocumentRelevanceOverrideConfigurationsT>(value);
  }
  template <typename DocumentRelevanceOverrideConfigurationsT = Aws::Vector<DocumentRelevanceConfiguration>>
  RetrieveRequest& WithDocumentRelevanceOverrideConfigurations(DocumentRelevanceOverrideConfigurationsT&& value) {
    SetDocumentRelevanceOverrideConfigurations(std::forward<DocumentRelevanceOverrideConfigurationsT>(value));
    return *this;
  }
  template <typename DocumentRelevanceOverrideConfigurationsT = DocumentRelevanceConfiguration>
  RetrieveRequest& AddDocumentRelevanceOverrideConfigurations(DocumentRelevanceOverrideConfigurationsT&& value) {
    m_documentRelevanceOverrideConfigurationsHasBeenSet = true;
    m_documentRelevanceOverrideConfigurations.emplace_back(std::forward<DocumentRelevanceOverrideConfigurationsT>(value));
    return *this;
  }
 
 
  inline int GetPageNumber() const { return m_pageNumber; }
  inline bool PageNumberHasBeenSet() const { return m_pageNumberHasBeenSet; }
  inline void SetPageNumber(int value) {
    m_pageNumberHasBeenSet = true;
    m_pageNumber = value;
  }
  inline RetrieveRequest& WithPageNumber(int value) {
    SetPageNumber(value);
    return *this;
  }
 
 
  inline int GetPageSize() const { return m_pageSize; }
  inline bool PageSizeHasBeenSet() const { return m_pageSizeHasBeenSet; }
  inline void SetPageSize(int value) {
    m_pageSizeHasBeenSet = true;
    m_pageSize = value;
  }
  inline RetrieveRequest& WithPageSize(int value) {
    SetPageSize(value);
    return *this;
  }
 
 
  inline const UserContext& GetUserContext() const { return m_userContext; }
  inline bool UserContextHasBeenSet() const { return m_userContextHasBeenSet; }
  template <typename UserContextT = UserContext>
  void SetUserContext(UserContextT&& value) {
    m_userContextHasBeenSet = true;
    m_userContext = std::forward<UserContextT>(value);
  }
  template <typename UserContextT = UserContext>
  RetrieveRequest& WithUserContext(UserContextT&& value) {
    SetUserContext(std::forward<UserContextT>(value));
    return *this;
  }
 private:
  Aws::String m_indexId;
  bool m_indexIdHasBeenSet = false;
 
  Aws::String m_queryText;
  bool m_queryTextHasBeenSet = false;
 
  AttributeFilter m_attributeFilter;
  bool m_attributeFilterHasBeenSet = false;
 
  Aws::Vector<Aws::String> m_requestedDocumentAttributes;
  bool m_requestedDocumentAttributesHasBeenSet = false;
 
  Aws::Vector<DocumentRelevanceConfiguration> m_documentRelevanceOverrideConfigurations;
  bool m_documentRelevanceOverrideConfigurationsHasBeenSet = false;
 
  int m_pageNumber{0};
  bool m_pageNumberHasBeenSet = false;
 
  int m_pageSize{0};
  bool m_pageSizeHasBeenSet = false;
 
  UserContext m_userContext;
  bool m_userContextHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace kendra
}  // namespace Aws