ContentBlock.h

 
#pragma once
#include <aws/bedrock-runtime/BedrockRuntime_EXPORTS.h>
#include <aws/bedrock-runtime/model/AudioBlock.h>
#include <aws/bedrock-runtime/model/CachePointBlock.h>
#include <aws/bedrock-runtime/model/CitationsContentBlock.h>
#include <aws/bedrock-runtime/model/DocumentBlock.h>
#include <aws/bedrock-runtime/model/GuardrailConverseContentBlock.h>
#include <aws/bedrock-runtime/model/ImageBlock.h>
#include <aws/bedrock-runtime/model/ReasoningContentBlock.h>
#include <aws/bedrock-runtime/model/SearchResultBlock.h>
#include <aws/bedrock-runtime/model/ToolAdditionBlock.h>
#include <aws/bedrock-runtime/model/ToolRemovalBlock.h>
#include <aws/bedrock-runtime/model/ToolResultBlock.h>
#include <aws/bedrock-runtime/model/ToolUseBlock.h>
#include <aws/bedrock-runtime/model/VideoBlock.h>
#include <aws/core/utils/memory/stl/AWSString.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockRuntime {
namespace Model {
 
class ContentBlock {
 public:
  AWS_BEDROCKRUNTIME_API ContentBlock() = default;
  AWS_BEDROCKRUNTIME_API ContentBlock(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKRUNTIME_API ContentBlock& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Aws::String& GetText() const { return m_text; }
  inline bool TextHasBeenSet() const { return m_textHasBeenSet; }
  template <typename TextT = Aws::String>
  void SetText(TextT&& value) {
    m_textHasBeenSet = true;
    m_text = std::forward<TextT>(value);
  }
  template <typename TextT = Aws::String>
  ContentBlock& WithText(TextT&& value) {
    SetText(std::forward<TextT>(value));
    return *this;
  }
 
 
  inline const ImageBlock& GetImage() const { return m_image; }
  inline bool ImageHasBeenSet() const { return m_imageHasBeenSet; }
  template <typename ImageT = ImageBlock>
  void SetImage(ImageT&& value) {
    m_imageHasBeenSet = true;
    m_image = std::forward<ImageT>(value);
  }
  template <typename ImageT = ImageBlock>
  ContentBlock& WithImage(ImageT&& value) {
    SetImage(std::forward<ImageT>(value));
    return *this;
  }
 
 
  inline const DocumentBlock& GetDocument() const { return m_document; }
  inline bool DocumentHasBeenSet() const { return m_documentHasBeenSet; }
  template <typename DocumentT = DocumentBlock>
  void SetDocument(DocumentT&& value) {
    m_documentHasBeenSet = true;
    m_document = std::forward<DocumentT>(value);
  }
  template <typename DocumentT = DocumentBlock>
  ContentBlock& WithDocument(DocumentT&& value) {
    SetDocument(std::forward<DocumentT>(value));
    return *this;
  }
 
 
  inline const VideoBlock& GetVideo() const { return m_video; }
  inline bool VideoHasBeenSet() const { return m_videoHasBeenSet; }
  template <typename VideoT = VideoBlock>
  void SetVideo(VideoT&& value) {
    m_videoHasBeenSet = true;
    m_video = std::forward<VideoT>(value);
  }
  template <typename VideoT = VideoBlock>
  ContentBlock& WithVideo(VideoT&& value) {
    SetVideo(std::forward<VideoT>(value));
    return *this;
  }
 
 
  inline const AudioBlock& GetAudio() const { return m_audio; }
  inline bool AudioHasBeenSet() const { return m_audioHasBeenSet; }
  template <typename AudioT = AudioBlock>
  void SetAudio(AudioT&& value) {
    m_audioHasBeenSet = true;
    m_audio = std::forward<AudioT>(value);
  }
  template <typename AudioT = AudioBlock>
  ContentBlock& WithAudio(AudioT&& value) {
    SetAudio(std::forward<AudioT>(value));
    return *this;
  }
 
 
  inline const ToolUseBlock& GetToolUse() const { return m_toolUse; }
  inline bool ToolUseHasBeenSet() const { return m_toolUseHasBeenSet; }
  template <typename ToolUseT = ToolUseBlock>
  void SetToolUse(ToolUseT&& value) {
    m_toolUseHasBeenSet = true;
    m_toolUse = std::forward<ToolUseT>(value);
  }
  template <typename ToolUseT = ToolUseBlock>
  ContentBlock& WithToolUse(ToolUseT&& value) {
    SetToolUse(std::forward<ToolUseT>(value));
    return *this;
  }
 
 
  inline const ToolResultBlock& GetToolResult() const { return m_toolResult; }
  inline bool ToolResultHasBeenSet() const { return m_toolResultHasBeenSet; }
  template <typename ToolResultT = ToolResultBlock>
  void SetToolResult(ToolResultT&& value) {
    m_toolResultHasBeenSet = true;
    m_toolResult = std::forward<ToolResultT>(value);
  }
  template <typename ToolResultT = ToolResultBlock>
  ContentBlock& WithToolResult(ToolResultT&& value) {
    SetToolResult(std::forward<ToolResultT>(value));
    return *this;
  }
 
 
  inline const GuardrailConverseContentBlock& GetGuardContent() const { return m_guardContent; }
  inline bool GuardContentHasBeenSet() const { return m_guardContentHasBeenSet; }
  template <typename GuardContentT = GuardrailConverseContentBlock>
  void SetGuardContent(GuardContentT&& value) {
    m_guardContentHasBeenSet = true;
    m_guardContent = std::forward<GuardContentT>(value);
  }
  template <typename GuardContentT = GuardrailConverseContentBlock>
  ContentBlock& WithGuardContent(GuardContentT&& value) {
    SetGuardContent(std::forward<GuardContentT>(value));
    return *this;
  }
 
 
  inline const CachePointBlock& GetCachePoint() const { return m_cachePoint; }
  inline bool CachePointHasBeenSet() const { return m_cachePointHasBeenSet; }
  template <typename CachePointT = CachePointBlock>
  void SetCachePoint(CachePointT&& value) {
    m_cachePointHasBeenSet = true;
    m_cachePoint = std::forward<CachePointT>(value);
  }
  template <typename CachePointT = CachePointBlock>
  ContentBlock& WithCachePoint(CachePointT&& value) {
    SetCachePoint(std::forward<CachePointT>(value));
    return *this;
  }
 
 
  inline const ReasoningContentBlock& GetReasoningContent() const { return m_reasoningContent; }
  inline bool ReasoningContentHasBeenSet() const { return m_reasoningContentHasBeenSet; }
  template <typename ReasoningContentT = ReasoningContentBlock>
  void SetReasoningContent(ReasoningContentT&& value) {
    m_reasoningContentHasBeenSet = true;
    m_reasoningContent = std::forward<ReasoningContentT>(value);
  }
  template <typename ReasoningContentT = ReasoningContentBlock>
  ContentBlock& WithReasoningContent(ReasoningContentT&& value) {
    SetReasoningContent(std::forward<ReasoningContentT>(value));
    return *this;
  }
 
 
  inline const CitationsContentBlock& GetCitationsContent() const { return m_citationsContent; }
  inline bool CitationsContentHasBeenSet() const { return m_citationsContentHasBeenSet; }
  template <typename CitationsContentT = CitationsContentBlock>
  void SetCitationsContent(CitationsContentT&& value) {
    m_citationsContentHasBeenSet = true;
    m_citationsContent = std::forward<CitationsContentT>(value);
  }
  template <typename CitationsContentT = CitationsContentBlock>
  ContentBlock& WithCitationsContent(CitationsContentT&& value) {
    SetCitationsContent(std::forward<CitationsContentT>(value));
    return *this;
  }
 
 
  inline const SearchResultBlock& GetSearchResult() const { return m_searchResult; }
  inline bool SearchResultHasBeenSet() const { return m_searchResultHasBeenSet; }
  template <typename SearchResultT = SearchResultBlock>
  void SetSearchResult(SearchResultT&& value) {
    m_searchResultHasBeenSet = true;
    m_searchResult = std::forward<SearchResultT>(value);
  }
  template <typename SearchResultT = SearchResultBlock>
  ContentBlock& WithSearchResult(SearchResultT&& value) {
    SetSearchResult(std::forward<SearchResultT>(value));
    return *this;
  }
 
 
  inline const ToolAdditionBlock& GetToolAddition() const { return m_toolAddition; }
  inline bool ToolAdditionHasBeenSet() const { return m_toolAdditionHasBeenSet; }
  template <typename ToolAdditionT = ToolAdditionBlock>
  void SetToolAddition(ToolAdditionT&& value) {
    m_toolAdditionHasBeenSet = true;
    m_toolAddition = std::forward<ToolAdditionT>(value);
  }
  template <typename ToolAdditionT = ToolAdditionBlock>
  ContentBlock& WithToolAddition(ToolAdditionT&& value) {
    SetToolAddition(std::forward<ToolAdditionT>(value));
    return *this;
  }
 
 
  inline const ToolRemovalBlock& GetToolRemoval() const { return m_toolRemoval; }
  inline bool ToolRemovalHasBeenSet() const { return m_toolRemovalHasBeenSet; }
  template <typename ToolRemovalT = ToolRemovalBlock>
  void SetToolRemoval(ToolRemovalT&& value) {
    m_toolRemovalHasBeenSet = true;
    m_toolRemoval = std::forward<ToolRemovalT>(value);
  }
  template <typename ToolRemovalT = ToolRemovalBlock>
  ContentBlock& WithToolRemoval(ToolRemovalT&& value) {
    SetToolRemoval(std::forward<ToolRemovalT>(value));
    return *this;
  }
 private:
  Aws::String m_text;
 
  ImageBlock m_image;
 
  DocumentBlock m_document;
 
  VideoBlock m_video;
 
  AudioBlock m_audio;
 
  ToolUseBlock m_toolUse;
 
  ToolResultBlock m_toolResult;
 
  GuardrailConverseContentBlock m_guardContent;
 
  CachePointBlock m_cachePoint;
 
  ReasoningContentBlock m_reasoningContent;
 
  CitationsContentBlock m_citationsContent;
 
  SearchResultBlock m_searchResult;
 
  ToolAdditionBlock m_toolAddition;
 
  ToolRemovalBlock m_toolRemoval;
  bool m_textHasBeenSet = false;
  bool m_imageHasBeenSet = false;
  bool m_documentHasBeenSet = false;
  bool m_videoHasBeenSet = false;
  bool m_audioHasBeenSet = false;
  bool m_toolUseHasBeenSet = false;
  bool m_toolResultHasBeenSet = false;
  bool m_guardContentHasBeenSet = false;
  bool m_cachePointHasBeenSet = false;
  bool m_reasoningContentHasBeenSet = false;
  bool m_citationsContentHasBeenSet = false;
  bool m_searchResultHasBeenSet = false;
  bool m_toolAdditionHasBeenSet = false;
  bool m_toolRemovalHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockRuntime
}  // namespace Aws