CountTokensInput.h

 
#pragma once
#include <aws/bedrock-runtime/BedrockRuntime_EXPORTS.h>
#include <aws/bedrock-runtime/model/InvokeModelTokensRequest.h>
#include <aws/bedrock-runtime/model/ConverseTokensRequest.h>
#include <utility>
 
namespace Aws
{
namespace Utils
{
namespace Json
{
  class JsonValue;
  class JsonView;
} // namespace Json
} // namespace Utils
namespace BedrockRuntime
{
namespace Model
{
 
  class CountTokensInput
  {
  public:
    AWS_BEDROCKRUNTIME_API CountTokensInput() = default;
    AWS_BEDROCKRUNTIME_API CountTokensInput(Aws::Utils::Json::JsonView jsonValue);
    AWS_BEDROCKRUNTIME_API CountTokensInput& operator=(Aws::Utils::Json::JsonView jsonValue);
    AWS_BEDROCKRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
 
    inline const InvokeModelTokensRequest& GetInvokeModel() const { return m_invokeModel; }
    inline bool InvokeModelHasBeenSet() const { return m_invokeModelHasBeenSet; }
    template<typename InvokeModelT = InvokeModelTokensRequest>
    void SetInvokeModel(InvokeModelT&& value) { m_invokeModelHasBeenSet = true; m_invokeModel = std::forward<InvokeModelT>(value); }
    template<typename InvokeModelT = InvokeModelTokensRequest>
    CountTokensInput& WithInvokeModel(InvokeModelT&& value) { SetInvokeModel(std::forward<InvokeModelT>(value)); return *this;}
 
 
    inline const ConverseTokensRequest& GetConverse() const { return m_converse; }
    inline bool ConverseHasBeenSet() const { return m_converseHasBeenSet; }
    template<typename ConverseT = ConverseTokensRequest>
    void SetConverse(ConverseT&& value) { m_converseHasBeenSet = true; m_converse = std::forward<ConverseT>(value); }
    template<typename ConverseT = ConverseTokensRequest>
    CountTokensInput& WithConverse(ConverseT&& value) { SetConverse(std::forward<ConverseT>(value)); return *this;}
  private:
 
    InvokeModelTokensRequest m_invokeModel;
    bool m_invokeModelHasBeenSet = false;
 
    ConverseTokensRequest m_converse;
    bool m_converseHasBeenSet = false;
  };
 
} // namespace Model
} // namespace BedrockRuntime
} // namespace Aws