TextGenerationJobConfig.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/AutoMLJobCompletionCriteria.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/sagemaker/model/ModelAccessConfig.h>
#include <utility>
 
namespace Aws
{
namespace Utils
{
namespace Json
{
  class JsonValue;
  class JsonView;
} // namespace Json
} // namespace Utils
namespace SageMaker
{
namespace Model
{
 
  class TextGenerationJobConfig
  {
  public:
    AWS_SAGEMAKER_API TextGenerationJobConfig();
    AWS_SAGEMAKER_API TextGenerationJobConfig(Aws::Utils::Json::JsonView jsonValue);
    AWS_SAGEMAKER_API TextGenerationJobConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
    AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
 
    inline const AutoMLJobCompletionCriteria& GetCompletionCriteria() const{ return m_completionCriteria; }
    inline bool CompletionCriteriaHasBeenSet() const { return m_completionCriteriaHasBeenSet; }
    inline void SetCompletionCriteria(const AutoMLJobCompletionCriteria& value) { m_completionCriteriaHasBeenSet = true; m_completionCriteria = value; }
    inline void SetCompletionCriteria(AutoMLJobCompletionCriteria&& value) { m_completionCriteriaHasBeenSet = true; m_completionCriteria = std::move(value); }
    inline TextGenerationJobConfig& WithCompletionCriteria(const AutoMLJobCompletionCriteria& value) { SetCompletionCriteria(value); return *this;}
    inline TextGenerationJobConfig& WithCompletionCriteria(AutoMLJobCompletionCriteria&& value) { SetCompletionCriteria(std::move(value)); return *this;}
 
 
    inline const Aws::String& GetBaseModelName() const{ return m_baseModelName; }
    inline bool BaseModelNameHasBeenSet() const { return m_baseModelNameHasBeenSet; }
    inline void SetBaseModelName(const Aws::String& value) { m_baseModelNameHasBeenSet = true; m_baseModelName = value; }
    inline void SetBaseModelName(Aws::String&& value) { m_baseModelNameHasBeenSet = true; m_baseModelName = std::move(value); }
    inline void SetBaseModelName(const char* value) { m_baseModelNameHasBeenSet = true; m_baseModelName.assign(value); }
    inline TextGenerationJobConfig& WithBaseModelName(const Aws::String& value) { SetBaseModelName(value); return *this;}
    inline TextGenerationJobConfig& WithBaseModelName(Aws::String&& value) { SetBaseModelName(std::move(value)); return *this;}
    inline TextGenerationJobConfig& WithBaseModelName(const char* value) { SetBaseModelName(value); return *this;}
 
 
    inline const Aws::Map<Aws::String, Aws::String>& GetTextGenerationHyperParameters() const{ return m_textGenerationHyperParameters; }
    inline bool TextGenerationHyperParametersHasBeenSet() const { return m_textGenerationHyperParametersHasBeenSet; }
    inline void SetTextGenerationHyperParameters(const Aws::Map<Aws::String, Aws::String>& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters = value; }
    inline void SetTextGenerationHyperParameters(Aws::Map<Aws::String, Aws::String>&& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters = std::move(value); }
    inline TextGenerationJobConfig& WithTextGenerationHyperParameters(const Aws::Map<Aws::String, Aws::String>& value) { SetTextGenerationHyperParameters(value); return *this;}
    inline TextGenerationJobConfig& WithTextGenerationHyperParameters(Aws::Map<Aws::String, Aws::String>&& value) { SetTextGenerationHyperParameters(std::move(value)); return *this;}
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(const Aws::String& key, const Aws::String& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(key, value); return *this; }
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(Aws::String&& key, const Aws::String& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(std::move(key), value); return *this; }
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(const Aws::String& key, Aws::String&& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(key, std::move(value)); return *this; }
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(Aws::String&& key, Aws::String&& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(std::move(key), std::move(value)); return *this; }
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(const char* key, Aws::String&& value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(key, std::move(value)); return *this; }
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(Aws::String&& key, const char* value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(std::move(key), value); return *this; }
    inline TextGenerationJobConfig& AddTextGenerationHyperParameters(const char* key, const char* value) { m_textGenerationHyperParametersHasBeenSet = true; m_textGenerationHyperParameters.emplace(key, value); return *this; }
 
    
    inline const ModelAccessConfig& GetModelAccessConfig() const{ return m_modelAccessConfig; }
    inline bool ModelAccessConfigHasBeenSet() const { return m_modelAccessConfigHasBeenSet; }
    inline void SetModelAccessConfig(const ModelAccessConfig& value) { m_modelAccessConfigHasBeenSet = true; m_modelAccessConfig = value; }
    inline void SetModelAccessConfig(ModelAccessConfig&& value) { m_modelAccessConfigHasBeenSet = true; m_modelAccessConfig = std::move(value); }
    inline TextGenerationJobConfig& WithModelAccessConfig(const ModelAccessConfig& value) { SetModelAccessConfig(value); return *this;}
    inline TextGenerationJobConfig& WithModelAccessConfig(ModelAccessConfig&& value) { SetModelAccessConfig(std::move(value)); return *this;}
  private:
 
    AutoMLJobCompletionCriteria m_completionCriteria;
    bool m_completionCriteriaHasBeenSet = false;
 
    Aws::String m_baseModelName;
    bool m_baseModelNameHasBeenSet = false;
 
    Aws::Map<Aws::String, Aws::String> m_textGenerationHyperParameters;
    bool m_textGenerationHyperParametersHasBeenSet = false;
 
    ModelAccessConfig m_modelAccessConfig;
    bool m_modelAccessConfigHasBeenSet = false;
  };
 
} // namespace Model
} // namespace SageMaker
} // namespace Aws