OptimizePromptRequest.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeRequest.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/InputPrompt.h>
#include <aws/bedrock-agent-runtime/model/OptimizePromptHandler.h>
#include <aws/core/utils/event/EventStreamDecoder.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;
 
  Aws::String m_targetModelId;
  OptimizePromptHandler m_handler;
  Aws::Utils::Event::EventStreamDecoder m_decoder{Utils::Event::EventStreamDecoder(&m_handler)};
 
  bool m_inputHasBeenSet = false;
  bool m_targetModelIdHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws