TransformationConfiguration.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/QueryTransformationMode.h>
#include <aws/bedrock-agent-runtime/model/TextToSqlConfiguration.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgentRuntime {
namespace Model {
 
class TransformationConfiguration {
 public:
  AWS_BEDROCKAGENTRUNTIME_API TransformationConfiguration() = default;
  AWS_BEDROCKAGENTRUNTIME_API TransformationConfiguration(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API TransformationConfiguration& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline QueryTransformationMode GetMode() const { return m_mode; }
  inline bool ModeHasBeenSet() const { return m_modeHasBeenSet; }
  inline void SetMode(QueryTransformationMode value) {
    m_modeHasBeenSet = true;
    m_mode = value;
  }
  inline TransformationConfiguration& WithMode(QueryTransformationMode value) {
    SetMode(value);
    return *this;
  }
 
 
  inline const TextToSqlConfiguration& GetTextToSqlConfiguration() const { return m_textToSqlConfiguration; }
  inline bool TextToSqlConfigurationHasBeenSet() const { return m_textToSqlConfigurationHasBeenSet; }
  template <typename TextToSqlConfigurationT = TextToSqlConfiguration>
  void SetTextToSqlConfiguration(TextToSqlConfigurationT&& value) {
    m_textToSqlConfigurationHasBeenSet = true;
    m_textToSqlConfiguration = std::forward<TextToSqlConfigurationT>(value);
  }
  template <typename TextToSqlConfigurationT = TextToSqlConfiguration>
  TransformationConfiguration& WithTextToSqlConfiguration(TextToSqlConfigurationT&& value) {
    SetTextToSqlConfiguration(std::forward<TextToSqlConfigurationT>(value));
    return *this;
  }
 private:
  QueryTransformationMode m_mode{QueryTransformationMode::NOT_SET};
 
  TextToSqlConfiguration m_textToSqlConfiguration;
  bool m_modeHasBeenSet = false;
  bool m_textToSqlConfigurationHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws