RetrieveAndGenerateStreamRequest.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeRequest.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/RetrieveAndGenerateConfiguration.h>
#include <aws/bedrock-agent-runtime/model/RetrieveAndGenerateInput.h>
#include <aws/bedrock-agent-runtime/model/RetrieveAndGenerateSessionConfiguration.h>
#include <aws/bedrock-agent-runtime/model/RetrieveAndGenerateStreamHandler.h>
#include <aws/bedrock-agent-runtime/model/UserContext.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 RetrieveAndGenerateStreamRequest : public BedrockAgentRuntimeRequest {
 public:
  AWS_BEDROCKAGENTRUNTIME_API RetrieveAndGenerateStreamRequest() = 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 "RetrieveAndGenerateStream"; }
 
  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 RetrieveAndGenerateStreamHandler& GetEventStreamHandler() { return m_handler; }
 
  inline void SetEventStreamHandler(const RetrieveAndGenerateStreamHandler& value) {
    m_handler = value;
    m_decoder.ResetEventStreamHandler(&m_handler);
  }
 
  inline RetrieveAndGenerateStreamRequest& WithEventStreamHandler(const RetrieveAndGenerateStreamHandler& value) {
    SetEventStreamHandler(value);
    return *this;
  }
 
 
  inline const RetrieveAndGenerateInput& GetInput() const { return m_input; }
  inline bool InputHasBeenSet() const { return m_inputHasBeenSet; }
  template <typename InputT = RetrieveAndGenerateInput>
  void SetInput(InputT&& value) {
    m_inputHasBeenSet = true;
    m_input = std::forward<InputT>(value);
  }
  template <typename InputT = RetrieveAndGenerateInput>
  RetrieveAndGenerateStreamRequest& WithInput(InputT&& value) {
    SetInput(std::forward<InputT>(value));
    return *this;
  }
 
 
  inline const RetrieveAndGenerateConfiguration& GetRetrieveAndGenerateConfiguration() const { return m_retrieveAndGenerateConfiguration; }
  inline bool RetrieveAndGenerateConfigurationHasBeenSet() const { return m_retrieveAndGenerateConfigurationHasBeenSet; }
  template <typename RetrieveAndGenerateConfigurationT = RetrieveAndGenerateConfiguration>
  void SetRetrieveAndGenerateConfiguration(RetrieveAndGenerateConfigurationT&& value) {
    m_retrieveAndGenerateConfigurationHasBeenSet = true;
    m_retrieveAndGenerateConfiguration = std::forward<RetrieveAndGenerateConfigurationT>(value);
  }
  template <typename RetrieveAndGenerateConfigurationT = RetrieveAndGenerateConfiguration>
  RetrieveAndGenerateStreamRequest& WithRetrieveAndGenerateConfiguration(RetrieveAndGenerateConfigurationT&& value) {
    SetRetrieveAndGenerateConfiguration(std::forward<RetrieveAndGenerateConfigurationT>(value));
    return *this;
  }
 
 
  inline const RetrieveAndGenerateSessionConfiguration& GetSessionConfiguration() const { return m_sessionConfiguration; }
  inline bool SessionConfigurationHasBeenSet() const { return m_sessionConfigurationHasBeenSet; }
  template <typename SessionConfigurationT = RetrieveAndGenerateSessionConfiguration>
  void SetSessionConfiguration(SessionConfigurationT&& value) {
    m_sessionConfigurationHasBeenSet = true;
    m_sessionConfiguration = std::forward<SessionConfigurationT>(value);
  }
  template <typename SessionConfigurationT = RetrieveAndGenerateSessionConfiguration>
  RetrieveAndGenerateStreamRequest& WithSessionConfiguration(SessionConfigurationT&& value) {
    SetSessionConfiguration(std::forward<SessionConfigurationT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetSessionId() const { return m_sessionId; }
  inline bool SessionIdHasBeenSet() const { return m_sessionIdHasBeenSet; }
  template <typename SessionIdT = Aws::String>
  void SetSessionId(SessionIdT&& value) {
    m_sessionIdHasBeenSet = true;
    m_sessionId = std::forward<SessionIdT>(value);
  }
  template <typename SessionIdT = Aws::String>
  RetrieveAndGenerateStreamRequest& WithSessionId(SessionIdT&& value) {
    SetSessionId(std::forward<SessionIdT>(value));
    return *this;
  }
 
 
  inline const UserContext& GetUserContext() const { return m_userContext; }
  inline bool UserContextHasBeenSet() const { return m_userContextHasBeenSet; }
  template <typename UserContextT = UserContext>
  void SetUserContext(UserContextT&& value) {
    m_userContextHasBeenSet = true;
    m_userContext = std::forward<UserContextT>(value);
  }
  template <typename UserContextT = UserContext>
  RetrieveAndGenerateStreamRequest& WithUserContext(UserContextT&& value) {
    SetUserContext(std::forward<UserContextT>(value));
    return *this;
  }
 private:
  RetrieveAndGenerateInput m_input;
 
  RetrieveAndGenerateConfiguration m_retrieveAndGenerateConfiguration;
 
  RetrieveAndGenerateSessionConfiguration m_sessionConfiguration;
 
  Aws::String m_sessionId;
 
  UserContext m_userContext;
  RetrieveAndGenerateStreamHandler m_handler;
  Aws::Utils::Event::EventStreamDecoder m_decoder{Utils::Event::EventStreamDecoder(&m_handler)};
 
  bool m_inputHasBeenSet = false;
  bool m_retrieveAndGenerateConfigurationHasBeenSet = false;
  bool m_sessionConfigurationHasBeenSet = false;
  bool m_sessionIdHasBeenSet = false;
  bool m_userContextHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws