BedrockModelSpecification.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/lexv2-models/LexModelsV2_EXPORTS.h>
#include <aws/lexv2-models/model/BedrockGuardrailConfiguration.h>
#include <aws/lexv2-models/model/BedrockTraceStatus.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace LexModelsV2 {
namespace Model {
 
class BedrockModelSpecification {
 public:
  AWS_LEXMODELSV2_API BedrockModelSpecification() = default;
  AWS_LEXMODELSV2_API BedrockModelSpecification(Aws::Utils::Json::JsonView jsonValue);
  AWS_LEXMODELSV2_API BedrockModelSpecification& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_LEXMODELSV2_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Aws::String& GetModelArn() const { return m_modelArn; }
  inline bool ModelArnHasBeenSet() const { return m_modelArnHasBeenSet; }
  template <typename ModelArnT = Aws::String>
  void SetModelArn(ModelArnT&& value) {
    m_modelArnHasBeenSet = true;
    m_modelArn = std::forward<ModelArnT>(value);
  }
  template <typename ModelArnT = Aws::String>
  BedrockModelSpecification& WithModelArn(ModelArnT&& value) {
    SetModelArn(std::forward<ModelArnT>(value));
    return *this;
  }
 
 
  inline const BedrockGuardrailConfiguration& GetGuardrail() const { return m_guardrail; }
  inline bool GuardrailHasBeenSet() const { return m_guardrailHasBeenSet; }
  template <typename GuardrailT = BedrockGuardrailConfiguration>
  void SetGuardrail(GuardrailT&& value) {
    m_guardrailHasBeenSet = true;
    m_guardrail = std::forward<GuardrailT>(value);
  }
  template <typename GuardrailT = BedrockGuardrailConfiguration>
  BedrockModelSpecification& WithGuardrail(GuardrailT&& value) {
    SetGuardrail(std::forward<GuardrailT>(value));
    return *this;
  }
 
 
  inline BedrockTraceStatus GetTraceStatus() const { return m_traceStatus; }
  inline bool TraceStatusHasBeenSet() const { return m_traceStatusHasBeenSet; }
  inline void SetTraceStatus(BedrockTraceStatus value) {
    m_traceStatusHasBeenSet = true;
    m_traceStatus = value;
  }
  inline BedrockModelSpecification& WithTraceStatus(BedrockTraceStatus value) {
    SetTraceStatus(value);
    return *this;
  }
 
 
  inline const Aws::String& GetCustomPrompt() const { return m_customPrompt; }
  inline bool CustomPromptHasBeenSet() const { return m_customPromptHasBeenSet; }
  template <typename CustomPromptT = Aws::String>
  void SetCustomPrompt(CustomPromptT&& value) {
    m_customPromptHasBeenSet = true;
    m_customPrompt = std::forward<CustomPromptT>(value);
  }
  template <typename CustomPromptT = Aws::String>
  BedrockModelSpecification& WithCustomPrompt(CustomPromptT&& value) {
    SetCustomPrompt(std::forward<CustomPromptT>(value));
    return *this;
  }
 private:
  Aws::String m_modelArn;
 
  BedrockGuardrailConfiguration m_guardrail;
 
  BedrockTraceStatus m_traceStatus{BedrockTraceStatus::NOT_SET};
 
  Aws::String m_customPrompt;
  bool m_modelArnHasBeenSet = false;
  bool m_guardrailHasBeenSet = false;
  bool m_traceStatusHasBeenSet = false;
  bool m_customPromptHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace LexModelsV2
}  // namespace Aws