KnowledgeBaseConfiguration.h

 
#pragma once
#include <aws/bedrock-agent/BedrockAgent_EXPORTS.h>
#include <aws/bedrock-agent/model/KendraKnowledgeBaseConfiguration.h>
#include <aws/bedrock-agent/model/KnowledgeBaseType.h>
#include <aws/bedrock-agent/model/ManagedKnowledgeBaseConfiguration.h>
#include <aws/bedrock-agent/model/SqlKnowledgeBaseConfiguration.h>
#include <aws/bedrock-agent/model/VectorKnowledgeBaseConfiguration.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgent {
namespace Model {
 
class KnowledgeBaseConfiguration {
 public:
  AWS_BEDROCKAGENT_API KnowledgeBaseConfiguration() = default;
  AWS_BEDROCKAGENT_API KnowledgeBaseConfiguration(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENT_API KnowledgeBaseConfiguration& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENT_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline KnowledgeBaseType GetType() const { return m_type; }
  inline bool TypeHasBeenSet() const { return m_typeHasBeenSet; }
  inline void SetType(KnowledgeBaseType value) {
    m_typeHasBeenSet = true;
    m_type = value;
  }
  inline KnowledgeBaseConfiguration& WithType(KnowledgeBaseType value) {
    SetType(value);
    return *this;
  }
 
 
  inline const VectorKnowledgeBaseConfiguration& GetVectorKnowledgeBaseConfiguration() const { return m_vectorKnowledgeBaseConfiguration; }
  inline bool VectorKnowledgeBaseConfigurationHasBeenSet() const { return m_vectorKnowledgeBaseConfigurationHasBeenSet; }
  template <typename VectorKnowledgeBaseConfigurationT = VectorKnowledgeBaseConfiguration>
  void SetVectorKnowledgeBaseConfiguration(VectorKnowledgeBaseConfigurationT&& value) {
    m_vectorKnowledgeBaseConfigurationHasBeenSet = true;
    m_vectorKnowledgeBaseConfiguration = std::forward<VectorKnowledgeBaseConfigurationT>(value);
  }
  template <typename VectorKnowledgeBaseConfigurationT = VectorKnowledgeBaseConfiguration>
  KnowledgeBaseConfiguration& WithVectorKnowledgeBaseConfiguration(VectorKnowledgeBaseConfigurationT&& value) {
    SetVectorKnowledgeBaseConfiguration(std::forward<VectorKnowledgeBaseConfigurationT>(value));
    return *this;
  }
 
 
  inline const ManagedKnowledgeBaseConfiguration& GetManagedKnowledgeBaseConfiguration() const {
    return m_managedKnowledgeBaseConfiguration;
  }
  inline bool ManagedKnowledgeBaseConfigurationHasBeenSet() const { return m_managedKnowledgeBaseConfigurationHasBeenSet; }
  template <typename ManagedKnowledgeBaseConfigurationT = ManagedKnowledgeBaseConfiguration>
  void SetManagedKnowledgeBaseConfiguration(ManagedKnowledgeBaseConfigurationT&& value) {
    m_managedKnowledgeBaseConfigurationHasBeenSet = true;
    m_managedKnowledgeBaseConfiguration = std::forward<ManagedKnowledgeBaseConfigurationT>(value);
  }
  template <typename ManagedKnowledgeBaseConfigurationT = ManagedKnowledgeBaseConfiguration>
  KnowledgeBaseConfiguration& WithManagedKnowledgeBaseConfiguration(ManagedKnowledgeBaseConfigurationT&& value) {
    SetManagedKnowledgeBaseConfiguration(std::forward<ManagedKnowledgeBaseConfigurationT>(value));
    return *this;
  }
 
 
  inline const KendraKnowledgeBaseConfiguration& GetKendraKnowledgeBaseConfiguration() const { return m_kendraKnowledgeBaseConfiguration; }
  inline bool KendraKnowledgeBaseConfigurationHasBeenSet() const { return m_kendraKnowledgeBaseConfigurationHasBeenSet; }
  template <typename KendraKnowledgeBaseConfigurationT = KendraKnowledgeBaseConfiguration>
  void SetKendraKnowledgeBaseConfiguration(KendraKnowledgeBaseConfigurationT&& value) {
    m_kendraKnowledgeBaseConfigurationHasBeenSet = true;
    m_kendraKnowledgeBaseConfiguration = std::forward<KendraKnowledgeBaseConfigurationT>(value);
  }
  template <typename KendraKnowledgeBaseConfigurationT = KendraKnowledgeBaseConfiguration>
  KnowledgeBaseConfiguration& WithKendraKnowledgeBaseConfiguration(KendraKnowledgeBaseConfigurationT&& value) {
    SetKendraKnowledgeBaseConfiguration(std::forward<KendraKnowledgeBaseConfigurationT>(value));
    return *this;
  }
 
 
  inline const SqlKnowledgeBaseConfiguration& GetSqlKnowledgeBaseConfiguration() const { return m_sqlKnowledgeBaseConfiguration; }
  inline bool SqlKnowledgeBaseConfigurationHasBeenSet() const { return m_sqlKnowledgeBaseConfigurationHasBeenSet; }
  template <typename SqlKnowledgeBaseConfigurationT = SqlKnowledgeBaseConfiguration>
  void SetSqlKnowledgeBaseConfiguration(SqlKnowledgeBaseConfigurationT&& value) {
    m_sqlKnowledgeBaseConfigurationHasBeenSet = true;
    m_sqlKnowledgeBaseConfiguration = std::forward<SqlKnowledgeBaseConfigurationT>(value);
  }
  template <typename SqlKnowledgeBaseConfigurationT = SqlKnowledgeBaseConfiguration>
  KnowledgeBaseConfiguration& WithSqlKnowledgeBaseConfiguration(SqlKnowledgeBaseConfigurationT&& value) {
    SetSqlKnowledgeBaseConfiguration(std::forward<SqlKnowledgeBaseConfigurationT>(value));
    return *this;
  }
 private:
  KnowledgeBaseType m_type{KnowledgeBaseType::NOT_SET};
 
  VectorKnowledgeBaseConfiguration m_vectorKnowledgeBaseConfiguration;
 
  ManagedKnowledgeBaseConfiguration m_managedKnowledgeBaseConfiguration;
 
  KendraKnowledgeBaseConfiguration m_kendraKnowledgeBaseConfiguration;
 
  SqlKnowledgeBaseConfiguration m_sqlKnowledgeBaseConfiguration;
  bool m_typeHasBeenSet = false;
  bool m_vectorKnowledgeBaseConfigurationHasBeenSet = false;
  bool m_managedKnowledgeBaseConfigurationHasBeenSet = false;
  bool m_kendraKnowledgeBaseConfigurationHasBeenSet = false;
  bool m_sqlKnowledgeBaseConfigurationHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgent
}  // namespace Aws