QueryGenerationConfiguration.h

 
#pragma once
#include <aws/bedrock-agent/BedrockAgent_EXPORTS.h>
#include <aws/bedrock-agent/model/QueryGenerationContext.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgent {
namespace Model {
 
class QueryGenerationConfiguration {
 public:
  AWS_BEDROCKAGENT_API QueryGenerationConfiguration() = default;
  AWS_BEDROCKAGENT_API QueryGenerationConfiguration(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENT_API QueryGenerationConfiguration& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENT_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline int GetExecutionTimeoutSeconds() const { return m_executionTimeoutSeconds; }
  inline bool ExecutionTimeoutSecondsHasBeenSet() const { return m_executionTimeoutSecondsHasBeenSet; }
  inline void SetExecutionTimeoutSeconds(int value) {
    m_executionTimeoutSecondsHasBeenSet = true;
    m_executionTimeoutSeconds = value;
  }
  inline QueryGenerationConfiguration& WithExecutionTimeoutSeconds(int value) {
    SetExecutionTimeoutSeconds(value);
    return *this;
  }
 
 
  inline const QueryGenerationContext& GetGenerationContext() const { return m_generationContext; }
  inline bool GenerationContextHasBeenSet() const { return m_generationContextHasBeenSet; }
  template <typename GenerationContextT = QueryGenerationContext>
  void SetGenerationContext(GenerationContextT&& value) {
    m_generationContextHasBeenSet = true;
    m_generationContext = std::forward<GenerationContextT>(value);
  }
  template <typename GenerationContextT = QueryGenerationContext>
  QueryGenerationConfiguration& WithGenerationContext(GenerationContextT&& value) {
    SetGenerationContext(std::forward<GenerationContextT>(value));
    return *this;
  }
 private:
  int m_executionTimeoutSeconds{0};
 
  QueryGenerationContext m_generationContext;
  bool m_executionTimeoutSecondsHasBeenSet = false;
  bool m_generationContextHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgent
}  // namespace Aws