InvokeAgentRequest.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeRequest.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/BedrockModelConfigurations.h>
#include <aws/bedrock-agent-runtime/model/InvokeAgentHandler.h>
#include <aws/bedrock-agent-runtime/model/PromptCreationConfigurations.h>
#include <aws/bedrock-agent-runtime/model/SessionState.h>
#include <aws/bedrock-agent-runtime/model/StreamingConfigurations.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 InvokeAgentRequest : public BedrockAgentRuntimeRequest {
 public:
  AWS_BEDROCKAGENTRUNTIME_API InvokeAgentRequest() = 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 "InvokeAgent"; }
 
  inline virtual bool HasEventStreamResponse() const override { return true; }
  AWS_BEDROCKAGENTRUNTIME_API Aws::String SerializePayload() const override;
 
  AWS_BEDROCKAGENTRUNTIME_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
  inline Aws::Utils::Event::EventStreamDecoder& GetEventStreamDecoder() { return m_decoder; }
 
  inline InvokeAgentHandler& GetEventStreamHandler() { return m_handler; }
 
  inline void SetEventStreamHandler(const InvokeAgentHandler& value) {
    m_handler = value;
    m_decoder.ResetEventStreamHandler(&m_handler);
  }
 
  inline InvokeAgentRequest& WithEventStreamHandler(const InvokeAgentHandler& value) {
    SetEventStreamHandler(value);
    return *this;
  }
 
 
  inline const Aws::String& GetAgentAliasId() const { return m_agentAliasId; }
  inline bool AgentAliasIdHasBeenSet() const { return m_agentAliasIdHasBeenSet; }
  template <typename AgentAliasIdT = Aws::String>
  void SetAgentAliasId(AgentAliasIdT&& value) {
    m_agentAliasIdHasBeenSet = true;
    m_agentAliasId = std::forward<AgentAliasIdT>(value);
  }
  template <typename AgentAliasIdT = Aws::String>
  InvokeAgentRequest& WithAgentAliasId(AgentAliasIdT&& value) {
    SetAgentAliasId(std::forward<AgentAliasIdT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetAgentId() const { return m_agentId; }
  inline bool AgentIdHasBeenSet() const { return m_agentIdHasBeenSet; }
  template <typename AgentIdT = Aws::String>
  void SetAgentId(AgentIdT&& value) {
    m_agentIdHasBeenSet = true;
    m_agentId = std::forward<AgentIdT>(value);
  }
  template <typename AgentIdT = Aws::String>
  InvokeAgentRequest& WithAgentId(AgentIdT&& value) {
    SetAgentId(std::forward<AgentIdT>(value));
    return *this;
  }
 
 
  inline const BedrockModelConfigurations& GetBedrockModelConfigurations() const { return m_bedrockModelConfigurations; }
  inline bool BedrockModelConfigurationsHasBeenSet() const { return m_bedrockModelConfigurationsHasBeenSet; }
  template <typename BedrockModelConfigurationsT = BedrockModelConfigurations>
  void SetBedrockModelConfigurations(BedrockModelConfigurationsT&& value) {
    m_bedrockModelConfigurationsHasBeenSet = true;
    m_bedrockModelConfigurations = std::forward<BedrockModelConfigurationsT>(value);
  }
  template <typename BedrockModelConfigurationsT = BedrockModelConfigurations>
  InvokeAgentRequest& WithBedrockModelConfigurations(BedrockModelConfigurationsT&& value) {
    SetBedrockModelConfigurations(std::forward<BedrockModelConfigurationsT>(value));
    return *this;
  }
 
 
  inline bool GetEnableTrace() const { return m_enableTrace; }
  inline bool EnableTraceHasBeenSet() const { return m_enableTraceHasBeenSet; }
  inline void SetEnableTrace(bool value) {
    m_enableTraceHasBeenSet = true;
    m_enableTrace = value;
  }
  inline InvokeAgentRequest& WithEnableTrace(bool value) {
    SetEnableTrace(value);
    return *this;
  }
 
 
  inline bool GetEndSession() const { return m_endSession; }
  inline bool EndSessionHasBeenSet() const { return m_endSessionHasBeenSet; }
  inline void SetEndSession(bool value) {
    m_endSessionHasBeenSet = true;
    m_endSession = value;
  }
  inline InvokeAgentRequest& WithEndSession(bool value) {
    SetEndSession(value);
    return *this;
  }
 
 
  inline const Aws::String& GetInputText() const { return m_inputText; }
  inline bool InputTextHasBeenSet() const { return m_inputTextHasBeenSet; }
  template <typename InputTextT = Aws::String>
  void SetInputText(InputTextT&& value) {
    m_inputTextHasBeenSet = true;
    m_inputText = std::forward<InputTextT>(value);
  }
  template <typename InputTextT = Aws::String>
  InvokeAgentRequest& WithInputText(InputTextT&& value) {
    SetInputText(std::forward<InputTextT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetMemoryId() const { return m_memoryId; }
  inline bool MemoryIdHasBeenSet() const { return m_memoryIdHasBeenSet; }
  template <typename MemoryIdT = Aws::String>
  void SetMemoryId(MemoryIdT&& value) {
    m_memoryIdHasBeenSet = true;
    m_memoryId = std::forward<MemoryIdT>(value);
  }
  template <typename MemoryIdT = Aws::String>
  InvokeAgentRequest& WithMemoryId(MemoryIdT&& value) {
    SetMemoryId(std::forward<MemoryIdT>(value));
    return *this;
  }
 
 
  inline const PromptCreationConfigurations& GetPromptCreationConfigurations() const { return m_promptCreationConfigurations; }
  inline bool PromptCreationConfigurationsHasBeenSet() const { return m_promptCreationConfigurationsHasBeenSet; }
  template <typename PromptCreationConfigurationsT = PromptCreationConfigurations>
  void SetPromptCreationConfigurations(PromptCreationConfigurationsT&& value) {
    m_promptCreationConfigurationsHasBeenSet = true;
    m_promptCreationConfigurations = std::forward<PromptCreationConfigurationsT>(value);
  }
  template <typename PromptCreationConfigurationsT = PromptCreationConfigurations>
  InvokeAgentRequest& WithPromptCreationConfigurations(PromptCreationConfigurationsT&& value) {
    SetPromptCreationConfigurations(std::forward<PromptCreationConfigurationsT>(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>
  InvokeAgentRequest& WithSessionId(SessionIdT&& value) {
    SetSessionId(std::forward<SessionIdT>(value));
    return *this;
  }
 
 
  inline const SessionState& GetSessionState() const { return m_sessionState; }
  inline bool SessionStateHasBeenSet() const { return m_sessionStateHasBeenSet; }
  template <typename SessionStateT = SessionState>
  void SetSessionState(SessionStateT&& value) {
    m_sessionStateHasBeenSet = true;
    m_sessionState = std::forward<SessionStateT>(value);
  }
  template <typename SessionStateT = SessionState>
  InvokeAgentRequest& WithSessionState(SessionStateT&& value) {
    SetSessionState(std::forward<SessionStateT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetSourceArn() const { return m_sourceArn; }
  inline bool SourceArnHasBeenSet() const { return m_sourceArnHasBeenSet; }
  template <typename SourceArnT = Aws::String>
  void SetSourceArn(SourceArnT&& value) {
    m_sourceArnHasBeenSet = true;
    m_sourceArn = std::forward<SourceArnT>(value);
  }
  template <typename SourceArnT = Aws::String>
  InvokeAgentRequest& WithSourceArn(SourceArnT&& value) {
    SetSourceArn(std::forward<SourceArnT>(value));
    return *this;
  }
 
 
  inline const StreamingConfigurations& GetStreamingConfigurations() const { return m_streamingConfigurations; }
  inline bool StreamingConfigurationsHasBeenSet() const { return m_streamingConfigurationsHasBeenSet; }
  template <typename StreamingConfigurationsT = StreamingConfigurations>
  void SetStreamingConfigurations(StreamingConfigurationsT&& value) {
    m_streamingConfigurationsHasBeenSet = true;
    m_streamingConfigurations = std::forward<StreamingConfigurationsT>(value);
  }
  template <typename StreamingConfigurationsT = StreamingConfigurations>
  InvokeAgentRequest& WithStreamingConfigurations(StreamingConfigurationsT&& value) {
    SetStreamingConfigurations(std::forward<StreamingConfigurationsT>(value));
    return *this;
  }
 private:
  Aws::String m_agentAliasId;
 
  Aws::String m_agentId;
 
  BedrockModelConfigurations m_bedrockModelConfigurations;
 
  bool m_enableTrace{false};
 
  bool m_endSession{false};
 
  Aws::String m_inputText;
 
  Aws::String m_memoryId;
 
  PromptCreationConfigurations m_promptCreationConfigurations;
 
  Aws::String m_sessionId;
 
  SessionState m_sessionState;
 
  Aws::String m_sourceArn;
 
  StreamingConfigurations m_streamingConfigurations;
  InvokeAgentHandler m_handler;
  Aws::Utils::Event::EventStreamDecoder m_decoder{Utils::Event::EventStreamDecoder(&m_handler)};
 
  bool m_agentAliasIdHasBeenSet = false;
  bool m_agentIdHasBeenSet = false;
  bool m_bedrockModelConfigurationsHasBeenSet = false;
  bool m_enableTraceHasBeenSet = false;
  bool m_endSessionHasBeenSet = false;
  bool m_inputTextHasBeenSet = false;
  bool m_memoryIdHasBeenSet = false;
  bool m_promptCreationConfigurationsHasBeenSet = false;
  bool m_sessionIdHasBeenSet = false;
  bool m_sessionStateHasBeenSet = false;
  bool m_sourceArnHasBeenSet = false;
  bool m_streamingConfigurationsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws