OptimizePromptRequest.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeRequest.h>
#include <aws/bedrock-agent-runtime/model/OptimizePromptHandler.h>
#include <aws/core/utils/event/EventStreamDecoder.h>
#include <aws/bedrock-agent-runtime/model/InputPrompt.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <utility>
 
namespace Aws
{
namespace BedrockAgentRuntime
{
namespace Model
{
 
  class OptimizePromptRequest : public BedrockAgentRuntimeRequest
  {
  public:
    AWS_BEDROCKAGENTRUNTIME_API OptimizePromptRequest() = default;
 
    // Service request name is the Operation name which will send this request out,
    // each operation should has unique request name, so that we can get operation's name from this request.
    // Note: this is not true for response, multiple operations may have the same response name,
    // so we can not get operation's name from response.
    inline virtual const char* GetServiceRequestName() const override { return "OptimizePrompt"; }
 
    inline virtual bool HasEventStreamResponse() const override { return true; }
    AWS_BEDROCKAGENTRUNTIME_API Aws::String SerializePayload() const override;
 
    inline Aws::Utils::Event::EventStreamDecoder& GetEventStreamDecoder() { return m_decoder; }
 
    inline OptimizePromptHandler& GetEventStreamHandler() { return m_handler; }
 
    inline void SetEventStreamHandler(const OptimizePromptHandler& value) { m_handler = value; m_decoder.ResetEventStreamHandler(&m_handler); }
 
    inline OptimizePromptRequest& WithEventStreamHandler(const OptimizePromptHandler& value) { SetEventStreamHandler(value); return *this; }
 
 
 
    inline const InputPrompt& GetInput() const { return m_input; }
    inline bool InputHasBeenSet() const { return m_inputHasBeenSet; }
    template<typename InputT = InputPrompt>
    void SetInput(InputT&& value) { m_inputHasBeenSet = true; m_input = std::forward<InputT>(value); }
    template<typename InputT = InputPrompt>
    OptimizePromptRequest& WithInput(InputT&& value) { SetInput(std::forward<InputT>(value)); return *this;}
 
 
    inline const Aws::String& GetTargetModelId() const { return m_targetModelId; }
    inline bool TargetModelIdHasBeenSet() const { return m_targetModelIdHasBeenSet; }
    template<typename TargetModelIdT = Aws::String>
    void SetTargetModelId(TargetModelIdT&& value) { m_targetModelIdHasBeenSet = true; m_targetModelId = std::forward<TargetModelIdT>(value); }
    template<typename TargetModelIdT = Aws::String>
    OptimizePromptRequest& WithTargetModelId(TargetModelIdT&& value) { SetTargetModelId(std::forward<TargetModelIdT>(value)); return *this;}
  private:
 
    InputPrompt m_input;
    bool m_inputHasBeenSet = false;
 
    Aws::String m_targetModelId;
    bool m_targetModelIdHasBeenSet = false;
    OptimizePromptHandler m_handler;
    Aws::Utils::Event::EventStreamDecoder m_decoder{Utils::Event::EventStreamDecoder(&m_handler)};
 
  };
 
} // namespace Model
} // namespace BedrockAgentRuntime
} // namespace Aws