KnowledgeBaseLookupOutput.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/RetrievedReference.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgentRuntime {
namespace Model {
 
class KnowledgeBaseLookupOutput {
 public:
  AWS_BEDROCKAGENTRUNTIME_API KnowledgeBaseLookupOutput() = default;
  AWS_BEDROCKAGENTRUNTIME_API KnowledgeBaseLookupOutput(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API KnowledgeBaseLookupOutput& 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>
  KnowledgeBaseLookupOutput& WithMetadata(MetadataT&& value) {
    SetMetadata(std::forward<MetadataT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<RetrievedReference>& GetRetrievedReferences() const { return m_retrievedReferences; }
  inline bool RetrievedReferencesHasBeenSet() const { return m_retrievedReferencesHasBeenSet; }
  template <typename RetrievedReferencesT = Aws::Vector<RetrievedReference>>
  void SetRetrievedReferences(RetrievedReferencesT&& value) {
    m_retrievedReferencesHasBeenSet = true;
    m_retrievedReferences = std::forward<RetrievedReferencesT>(value);
  }
  template <typename RetrievedReferencesT = Aws::Vector<RetrievedReference>>
  KnowledgeBaseLookupOutput& WithRetrievedReferences(RetrievedReferencesT&& value) {
    SetRetrievedReferences(std::forward<RetrievedReferencesT>(value));
    return *this;
  }
  template <typename RetrievedReferencesT = RetrievedReference>
  KnowledgeBaseLookupOutput& AddRetrievedReferences(RetrievedReferencesT&& value) {
    m_retrievedReferencesHasBeenSet = true;
    m_retrievedReferences.emplace_back(std::forward<RetrievedReferencesT>(value));
    return *this;
  }
 private:
  Metadata m_metadata;
 
  Aws::Vector<RetrievedReference> m_retrievedReferences;
  bool m_metadataHasBeenSet = false;
  bool m_retrievedReferencesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws