ManagedSearchConfiguration.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/ManagedSearchRerankingConfiguration.h>
#include <aws/bedrock-agent-runtime/model/RerankingModelType.h>
#include <aws/bedrock-agent-runtime/model/RetrievalFilter.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgentRuntime {
namespace Model {
 
class ManagedSearchConfiguration {
 public:
  AWS_BEDROCKAGENTRUNTIME_API ManagedSearchConfiguration() = default;
  AWS_BEDROCKAGENTRUNTIME_API ManagedSearchConfiguration(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API ManagedSearchConfiguration& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const RetrievalFilter& GetFilter() const { return m_filter; }
  inline bool FilterHasBeenSet() const { return m_filterHasBeenSet; }
  template <typename FilterT = RetrievalFilter>
  void SetFilter(FilterT&& value) {
    m_filterHasBeenSet = true;
    m_filter = std::forward<FilterT>(value);
  }
  template <typename FilterT = RetrievalFilter>
  ManagedSearchConfiguration& WithFilter(FilterT&& value) {
    SetFilter(std::forward<FilterT>(value));
    return *this;
  }
 
 
  inline int GetNumberOfResults() const { return m_numberOfResults; }
  inline bool NumberOfResultsHasBeenSet() const { return m_numberOfResultsHasBeenSet; }
  inline void SetNumberOfResults(int value) {
    m_numberOfResultsHasBeenSet = true;
    m_numberOfResults = value;
  }
  inline ManagedSearchConfiguration& WithNumberOfResults(int value) {
    SetNumberOfResults(value);
    return *this;
  }
 
 
  inline const ManagedSearchRerankingConfiguration& GetRerankingConfiguration() const { return m_rerankingConfiguration; }
  inline bool RerankingConfigurationHasBeenSet() const { return m_rerankingConfigurationHasBeenSet; }
  template <typename RerankingConfigurationT = ManagedSearchRerankingConfiguration>
  void SetRerankingConfiguration(RerankingConfigurationT&& value) {
    m_rerankingConfigurationHasBeenSet = true;
    m_rerankingConfiguration = std::forward<RerankingConfigurationT>(value);
  }
  template <typename RerankingConfigurationT = ManagedSearchRerankingConfiguration>
  ManagedSearchConfiguration& WithRerankingConfiguration(RerankingConfigurationT&& value) {
    SetRerankingConfiguration(std::forward<RerankingConfigurationT>(value));
    return *this;
  }
 
 
  inline RerankingModelType GetRerankingModelType() const { return m_rerankingModelType; }
  inline bool RerankingModelTypeHasBeenSet() const { return m_rerankingModelTypeHasBeenSet; }
  inline void SetRerankingModelType(RerankingModelType value) {
    m_rerankingModelTypeHasBeenSet = true;
    m_rerankingModelType = value;
  }
  inline ManagedSearchConfiguration& WithRerankingModelType(RerankingModelType value) {
    SetRerankingModelType(value);
    return *this;
  }
 private:
  RetrievalFilter m_filter;
 
  int m_numberOfResults{0};
 
  ManagedSearchRerankingConfiguration m_rerankingConfiguration;
 
  RerankingModelType m_rerankingModelType{RerankingModelType::NOT_SET};
  bool m_filterHasBeenSet = false;
  bool m_numberOfResultsHasBeenSet = false;
  bool m_rerankingConfigurationHasBeenSet = false;
  bool m_rerankingModelTypeHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws