PostProcessingModelInvocationOutput.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/Metadata.h>
#include <aws/bedrock-agent-runtime/model/PostProcessingParsedResponse.h>
#include <aws/bedrock-agent-runtime/model/RawResponse.h>
#include <aws/bedrock-agent-runtime/model/ReasoningContentBlock.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 BedrockAgentRuntime {
namespace Model {
 
class PostProcessingModelInvocationOutput {
 public:
  AWS_BEDROCKAGENTRUNTIME_API PostProcessingModelInvocationOutput() = default;
  AWS_BEDROCKAGENTRUNTIME_API PostProcessingModelInvocationOutput(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API PostProcessingModelInvocationOutput& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Metadata& GetMetadata() const { return m_metadata; }
  inline bool MetadataHasBeenSet() const { return m_metadataHasBeenSet; }
  template <typename MetadataT = Metadata>
  void SetMetadata(MetadataT&& value) {
    m_metadataHasBeenSet = true;
    m_metadata = std::forward<MetadataT>(value);
  }
  template <typename MetadataT = Metadata>
  PostProcessingModelInvocationOutput& WithMetadata(MetadataT&& value) {
    SetMetadata(std::forward<MetadataT>(value));
    return *this;
  }
 
 
  inline const PostProcessingParsedResponse& GetParsedResponse() const { return m_parsedResponse; }
  inline bool ParsedResponseHasBeenSet() const { return m_parsedResponseHasBeenSet; }
  template <typename ParsedResponseT = PostProcessingParsedResponse>
  void SetParsedResponse(ParsedResponseT&& value) {
    m_parsedResponseHasBeenSet = true;
    m_parsedResponse = std::forward<ParsedResponseT>(value);
  }
  template <typename ParsedResponseT = PostProcessingParsedResponse>
  PostProcessingModelInvocationOutput& WithParsedResponse(ParsedResponseT&& value) {
    SetParsedResponse(std::forward<ParsedResponseT>(value));
    return *this;
  }
 
 
  inline const RawResponse& GetRawResponse() const { return m_rawResponse; }
  inline bool RawResponseHasBeenSet() const { return m_rawResponseHasBeenSet; }
  template <typename RawResponseT = RawResponse>
  void SetRawResponse(RawResponseT&& value) {
    m_rawResponseHasBeenSet = true;
    m_rawResponse = std::forward<RawResponseT>(value);
  }
  template <typename RawResponseT = RawResponse>
  PostProcessingModelInvocationOutput& WithRawResponse(RawResponseT&& value) {
    SetRawResponse(std::forward<RawResponseT>(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>
  PostProcessingModelInvocationOutput& WithReasoningContent(ReasoningContentT&& value) {
    SetReasoningContent(std::forward<ReasoningContentT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTraceId() const { return m_traceId; }
  inline bool TraceIdHasBeenSet() const { return m_traceIdHasBeenSet; }
  template <typename TraceIdT = Aws::String>
  void SetTraceId(TraceIdT&& value) {
    m_traceIdHasBeenSet = true;
    m_traceId = std::forward<TraceIdT>(value);
  }
  template <typename TraceIdT = Aws::String>
  PostProcessingModelInvocationOutput& WithTraceId(TraceIdT&& value) {
    SetTraceId(std::forward<TraceIdT>(value));
    return *this;
  }
 private:
  Metadata m_metadata;
 
  PostProcessingParsedResponse m_parsedResponse;
 
  RawResponse m_rawResponse;
 
  ReasoningContentBlock m_reasoningContent;
 
  Aws::String m_traceId;
  bool m_metadataHasBeenSet = false;
  bool m_parsedResponseHasBeenSet = false;
  bool m_rawResponseHasBeenSet = false;
  bool m_reasoningContentHasBeenSet = false;
  bool m_traceIdHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws