RetrievalResultLocation.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultConfluenceLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultCustomDocumentLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultGoogleDriveLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultKendraDocumentLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultLocationType.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultOneDriveLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultS3Location.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultSalesforceLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultSharePointLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultSqlLocation.h>
#include <aws/bedrock-agent-runtime/model/RetrievalResultWebLocation.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgentRuntime {
namespace Model {
 
class RetrievalResultLocation {
 public:
  AWS_BEDROCKAGENTRUNTIME_API RetrievalResultLocation() = default;
  AWS_BEDROCKAGENTRUNTIME_API RetrievalResultLocation(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API RetrievalResultLocation& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const RetrievalResultConfluenceLocation& GetConfluenceLocation() const { return m_confluenceLocation; }
  inline bool ConfluenceLocationHasBeenSet() const { return m_confluenceLocationHasBeenSet; }
  template <typename ConfluenceLocationT = RetrievalResultConfluenceLocation>
  void SetConfluenceLocation(ConfluenceLocationT&& value) {
    m_confluenceLocationHasBeenSet = true;
    m_confluenceLocation = std::forward<ConfluenceLocationT>(value);
  }
  template <typename ConfluenceLocationT = RetrievalResultConfluenceLocation>
  RetrievalResultLocation& WithConfluenceLocation(ConfluenceLocationT&& value) {
    SetConfluenceLocation(std::forward<ConfluenceLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultCustomDocumentLocation& GetCustomDocumentLocation() const { return m_customDocumentLocation; }
  inline bool CustomDocumentLocationHasBeenSet() const { return m_customDocumentLocationHasBeenSet; }
  template <typename CustomDocumentLocationT = RetrievalResultCustomDocumentLocation>
  void SetCustomDocumentLocation(CustomDocumentLocationT&& value) {
    m_customDocumentLocationHasBeenSet = true;
    m_customDocumentLocation = std::forward<CustomDocumentLocationT>(value);
  }
  template <typename CustomDocumentLocationT = RetrievalResultCustomDocumentLocation>
  RetrievalResultLocation& WithCustomDocumentLocation(CustomDocumentLocationT&& value) {
    SetCustomDocumentLocation(std::forward<CustomDocumentLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultGoogleDriveLocation& GetGoogleDriveLocation() const { return m_googleDriveLocation; }
  inline bool GoogleDriveLocationHasBeenSet() const { return m_googleDriveLocationHasBeenSet; }
  template <typename GoogleDriveLocationT = RetrievalResultGoogleDriveLocation>
  void SetGoogleDriveLocation(GoogleDriveLocationT&& value) {
    m_googleDriveLocationHasBeenSet = true;
    m_googleDriveLocation = std::forward<GoogleDriveLocationT>(value);
  }
  template <typename GoogleDriveLocationT = RetrievalResultGoogleDriveLocation>
  RetrievalResultLocation& WithGoogleDriveLocation(GoogleDriveLocationT&& value) {
    SetGoogleDriveLocation(std::forward<GoogleDriveLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultKendraDocumentLocation& GetKendraDocumentLocation() const { return m_kendraDocumentLocation; }
  inline bool KendraDocumentLocationHasBeenSet() const { return m_kendraDocumentLocationHasBeenSet; }
  template <typename KendraDocumentLocationT = RetrievalResultKendraDocumentLocation>
  void SetKendraDocumentLocation(KendraDocumentLocationT&& value) {
    m_kendraDocumentLocationHasBeenSet = true;
    m_kendraDocumentLocation = std::forward<KendraDocumentLocationT>(value);
  }
  template <typename KendraDocumentLocationT = RetrievalResultKendraDocumentLocation>
  RetrievalResultLocation& WithKendraDocumentLocation(KendraDocumentLocationT&& value) {
    SetKendraDocumentLocation(std::forward<KendraDocumentLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultOneDriveLocation& GetOneDriveLocation() const { return m_oneDriveLocation; }
  inline bool OneDriveLocationHasBeenSet() const { return m_oneDriveLocationHasBeenSet; }
  template <typename OneDriveLocationT = RetrievalResultOneDriveLocation>
  void SetOneDriveLocation(OneDriveLocationT&& value) {
    m_oneDriveLocationHasBeenSet = true;
    m_oneDriveLocation = std::forward<OneDriveLocationT>(value);
  }
  template <typename OneDriveLocationT = RetrievalResultOneDriveLocation>
  RetrievalResultLocation& WithOneDriveLocation(OneDriveLocationT&& value) {
    SetOneDriveLocation(std::forward<OneDriveLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultS3Location& GetS3Location() const { return m_s3Location; }
  inline bool S3LocationHasBeenSet() const { return m_s3LocationHasBeenSet; }
  template <typename S3LocationT = RetrievalResultS3Location>
  void SetS3Location(S3LocationT&& value) {
    m_s3LocationHasBeenSet = true;
    m_s3Location = std::forward<S3LocationT>(value);
  }
  template <typename S3LocationT = RetrievalResultS3Location>
  RetrievalResultLocation& WithS3Location(S3LocationT&& value) {
    SetS3Location(std::forward<S3LocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultSalesforceLocation& GetSalesforceLocation() const { return m_salesforceLocation; }
  inline bool SalesforceLocationHasBeenSet() const { return m_salesforceLocationHasBeenSet; }
  template <typename SalesforceLocationT = RetrievalResultSalesforceLocation>
  void SetSalesforceLocation(SalesforceLocationT&& value) {
    m_salesforceLocationHasBeenSet = true;
    m_salesforceLocation = std::forward<SalesforceLocationT>(value);
  }
  template <typename SalesforceLocationT = RetrievalResultSalesforceLocation>
  RetrievalResultLocation& WithSalesforceLocation(SalesforceLocationT&& value) {
    SetSalesforceLocation(std::forward<SalesforceLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultSharePointLocation& GetSharePointLocation() const { return m_sharePointLocation; }
  inline bool SharePointLocationHasBeenSet() const { return m_sharePointLocationHasBeenSet; }
  template <typename SharePointLocationT = RetrievalResultSharePointLocation>
  void SetSharePointLocation(SharePointLocationT&& value) {
    m_sharePointLocationHasBeenSet = true;
    m_sharePointLocation = std::forward<SharePointLocationT>(value);
  }
  template <typename SharePointLocationT = RetrievalResultSharePointLocation>
  RetrievalResultLocation& WithSharePointLocation(SharePointLocationT&& value) {
    SetSharePointLocation(std::forward<SharePointLocationT>(value));
    return *this;
  }
 
 
  inline const RetrievalResultSqlLocation& GetSqlLocation() const { return m_sqlLocation; }
  inline bool SqlLocationHasBeenSet() const { return m_sqlLocationHasBeenSet; }
  template <typename SqlLocationT = RetrievalResultSqlLocation>
  void SetSqlLocation(SqlLocationT&& value) {
    m_sqlLocationHasBeenSet = true;
    m_sqlLocation = std::forward<SqlLocationT>(value);
  }
  template <typename SqlLocationT = RetrievalResultSqlLocation>
  RetrievalResultLocation& WithSqlLocation(SqlLocationT&& value) {
    SetSqlLocation(std::forward<SqlLocationT>(value));
    return *this;
  }
 
 
  inline RetrievalResultLocationType GetType() const { return m_type; }
  inline bool TypeHasBeenSet() const { return m_typeHasBeenSet; }
  inline void SetType(RetrievalResultLocationType value) {
    m_typeHasBeenSet = true;
    m_type = value;
  }
  inline RetrievalResultLocation& WithType(RetrievalResultLocationType value) {
    SetType(value);
    return *this;
  }
 
 
  inline const RetrievalResultWebLocation& GetWebLocation() const { return m_webLocation; }
  inline bool WebLocationHasBeenSet() const { return m_webLocationHasBeenSet; }
  template <typename WebLocationT = RetrievalResultWebLocation>
  void SetWebLocation(WebLocationT&& value) {
    m_webLocationHasBeenSet = true;
    m_webLocation = std::forward<WebLocationT>(value);
  }
  template <typename WebLocationT = RetrievalResultWebLocation>
  RetrievalResultLocation& WithWebLocation(WebLocationT&& value) {
    SetWebLocation(std::forward<WebLocationT>(value));
    return *this;
  }
 private:
  RetrievalResultConfluenceLocation m_confluenceLocation;
 
  RetrievalResultCustomDocumentLocation m_customDocumentLocation;
 
  RetrievalResultGoogleDriveLocation m_googleDriveLocation;
 
  RetrievalResultKendraDocumentLocation m_kendraDocumentLocation;
 
  RetrievalResultOneDriveLocation m_oneDriveLocation;
 
  RetrievalResultS3Location m_s3Location;
 
  RetrievalResultSalesforceLocation m_salesforceLocation;
 
  RetrievalResultSharePointLocation m_sharePointLocation;
 
  RetrievalResultSqlLocation m_sqlLocation;
 
  RetrievalResultLocationType m_type{RetrievalResultLocationType::NOT_SET};
 
  RetrievalResultWebLocation m_webLocation;
  bool m_confluenceLocationHasBeenSet = false;
  bool m_customDocumentLocationHasBeenSet = false;
  bool m_googleDriveLocationHasBeenSet = false;
  bool m_kendraDocumentLocationHasBeenSet = false;
  bool m_oneDriveLocationHasBeenSet = false;
  bool m_s3LocationHasBeenSet = false;
  bool m_salesforceLocationHasBeenSet = false;
  bool m_sharePointLocationHasBeenSet = false;
  bool m_sqlLocationHasBeenSet = false;
  bool m_typeHasBeenSet = false;
  bool m_webLocationHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws