InvokeInlineAgentRequest.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntimeRequest.h>
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/AgentActionGroup.h>
#include <aws/bedrock-agent-runtime/model/AgentCollaboration.h>
#include <aws/bedrock-agent-runtime/model/Collaborator.h>
#include <aws/bedrock-agent-runtime/model/CollaboratorConfiguration.h>
#include <aws/bedrock-agent-runtime/model/CustomOrchestration.h>
#include <aws/bedrock-agent-runtime/model/GuardrailConfigurationWithArn.h>
#include <aws/bedrock-agent-runtime/model/InlineBedrockModelConfigurations.h>
#include <aws/bedrock-agent-runtime/model/InlineSessionState.h>
#include <aws/bedrock-agent-runtime/model/InvokeInlineAgentHandler.h>
#include <aws/bedrock-agent-runtime/model/KnowledgeBase.h>
#include <aws/bedrock-agent-runtime/model/OrchestrationType.h>
#include <aws/bedrock-agent-runtime/model/PromptCreationConfigurations.h>
#include <aws/bedrock-agent-runtime/model/PromptOverrideConfiguration.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 <aws/core/utils/memory/stl/AWSVector.h>
 
#include <utility>
 
namespace Aws {
namespace BedrockAgentRuntime {
namespace Model {
 
class InvokeInlineAgentRequest : public BedrockAgentRuntimeRequest {
 public:
  AWS_BEDROCKAGENTRUNTIME_API InvokeInlineAgentRequest() = 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 "InvokeInlineAgent"; }
 
  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 InvokeInlineAgentHandler& GetEventStreamHandler() { return m_handler; }
 
  inline void SetEventStreamHandler(const InvokeInlineAgentHandler& value) {
    m_handler = value;
    m_decoder.ResetEventStreamHandler(&m_handler);
  }
 
  inline InvokeInlineAgentRequest& WithEventStreamHandler(const InvokeInlineAgentHandler& value) {
    SetEventStreamHandler(value);
    return *this;
  }
 
 
  inline const Aws::Vector<AgentActionGroup>& GetActionGroups() const { return m_actionGroups; }
  inline bool ActionGroupsHasBeenSet() const { return m_actionGroupsHasBeenSet; }
  template <typename ActionGroupsT = Aws::Vector<AgentActionGroup>>
  void SetActionGroups(ActionGroupsT&& value) {
    m_actionGroupsHasBeenSet = true;
    m_actionGroups = std::forward<ActionGroupsT>(value);
  }
  template <typename ActionGroupsT = Aws::Vector<AgentActionGroup>>
  InvokeInlineAgentRequest& WithActionGroups(ActionGroupsT&& value) {
    SetActionGroups(std::forward<ActionGroupsT>(value));
    return *this;
  }
  template <typename ActionGroupsT = AgentActionGroup>
  InvokeInlineAgentRequest& AddActionGroups(ActionGroupsT&& value) {
    m_actionGroupsHasBeenSet = true;
    m_actionGroups.emplace_back(std::forward<ActionGroupsT>(value));
    return *this;
  }
 
 
  inline AgentCollaboration GetAgentCollaboration() const { return m_agentCollaboration; }
  inline bool AgentCollaborationHasBeenSet() const { return m_agentCollaborationHasBeenSet; }
  inline void SetAgentCollaboration(AgentCollaboration value) {
    m_agentCollaborationHasBeenSet = true;
    m_agentCollaboration = value;
  }
  inline InvokeInlineAgentRequest& WithAgentCollaboration(AgentCollaboration value) {
    SetAgentCollaboration(value);
    return *this;
  }
 
 
  inline const Aws::String& GetAgentName() const { return m_agentName; }
  inline bool AgentNameHasBeenSet() const { return m_agentNameHasBeenSet; }
  template <typename AgentNameT = Aws::String>
  void SetAgentName(AgentNameT&& value) {
    m_agentNameHasBeenSet = true;
    m_agentName = std::forward<AgentNameT>(value);
  }
  template <typename AgentNameT = Aws::String>
  InvokeInlineAgentRequest& WithAgentName(AgentNameT&& value) {
    SetAgentName(std::forward<AgentNameT>(value));
    return *this;
  }
 
 
  inline const InlineBedrockModelConfigurations& GetBedrockModelConfigurations() const { return m_bedrockModelConfigurations; }
  inline bool BedrockModelConfigurationsHasBeenSet() const { return m_bedrockModelConfigurationsHasBeenSet; }
  template <typename BedrockModelConfigurationsT = InlineBedrockModelConfigurations>
  void SetBedrockModelConfigurations(BedrockModelConfigurationsT&& value) {
    m_bedrockModelConfigurationsHasBeenSet = true;
    m_bedrockModelConfigurations = std::forward<BedrockModelConfigurationsT>(value);
  }
  template <typename BedrockModelConfigurationsT = InlineBedrockModelConfigurations>
  InvokeInlineAgentRequest& WithBedrockModelConfigurations(BedrockModelConfigurationsT&& value) {
    SetBedrockModelConfigurations(std::forward<BedrockModelConfigurationsT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<CollaboratorConfiguration>& GetCollaboratorConfigurations() const { return m_collaboratorConfigurations; }
  inline bool CollaboratorConfigurationsHasBeenSet() const { return m_collaboratorConfigurationsHasBeenSet; }
  template <typename CollaboratorConfigurationsT = Aws::Vector<CollaboratorConfiguration>>
  void SetCollaboratorConfigurations(CollaboratorConfigurationsT&& value) {
    m_collaboratorConfigurationsHasBeenSet = true;
    m_collaboratorConfigurations = std::forward<CollaboratorConfigurationsT>(value);
  }
  template <typename CollaboratorConfigurationsT = Aws::Vector<CollaboratorConfiguration>>
  InvokeInlineAgentRequest& WithCollaboratorConfigurations(CollaboratorConfigurationsT&& value) {
    SetCollaboratorConfigurations(std::forward<CollaboratorConfigurationsT>(value));
    return *this;
  }
  template <typename CollaboratorConfigurationsT = CollaboratorConfiguration>
  InvokeInlineAgentRequest& AddCollaboratorConfigurations(CollaboratorConfigurationsT&& value) {
    m_collaboratorConfigurationsHasBeenSet = true;
    m_collaboratorConfigurations.emplace_back(std::forward<CollaboratorConfigurationsT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Collaborator>& GetCollaborators() const { return m_collaborators; }
  inline bool CollaboratorsHasBeenSet() const { return m_collaboratorsHasBeenSet; }
  template <typename CollaboratorsT = Aws::Vector<Collaborator>>
  void SetCollaborators(CollaboratorsT&& value) {
    m_collaboratorsHasBeenSet = true;
    m_collaborators = std::forward<CollaboratorsT>(value);
  }
  template <typename CollaboratorsT = Aws::Vector<Collaborator>>
  InvokeInlineAgentRequest& WithCollaborators(CollaboratorsT&& value) {
    SetCollaborators(std::forward<CollaboratorsT>(value));
    return *this;
  }
  template <typename CollaboratorsT = Collaborator>
  InvokeInlineAgentRequest& AddCollaborators(CollaboratorsT&& value) {
    m_collaboratorsHasBeenSet = true;
    m_collaborators.emplace_back(std::forward<CollaboratorsT>(value));
    return *this;
  }
 
 
  inline const CustomOrchestration& GetCustomOrchestration() const { return m_customOrchestration; }
  inline bool CustomOrchestrationHasBeenSet() const { return m_customOrchestrationHasBeenSet; }
  template <typename CustomOrchestrationT = CustomOrchestration>
  void SetCustomOrchestration(CustomOrchestrationT&& value) {
    m_customOrchestrationHasBeenSet = true;
    m_customOrchestration = std::forward<CustomOrchestrationT>(value);
  }
  template <typename CustomOrchestrationT = CustomOrchestration>
  InvokeInlineAgentRequest& WithCustomOrchestration(CustomOrchestrationT&& value) {
    SetCustomOrchestration(std::forward<CustomOrchestrationT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetCustomerEncryptionKeyArn() const { return m_customerEncryptionKeyArn; }
  inline bool CustomerEncryptionKeyArnHasBeenSet() const { return m_customerEncryptionKeyArnHasBeenSet; }
  template <typename CustomerEncryptionKeyArnT = Aws::String>
  void SetCustomerEncryptionKeyArn(CustomerEncryptionKeyArnT&& value) {
    m_customerEncryptionKeyArnHasBeenSet = true;
    m_customerEncryptionKeyArn = std::forward<CustomerEncryptionKeyArnT>(value);
  }
  template <typename CustomerEncryptionKeyArnT = Aws::String>
  InvokeInlineAgentRequest& WithCustomerEncryptionKeyArn(CustomerEncryptionKeyArnT&& value) {
    SetCustomerEncryptionKeyArn(std::forward<CustomerEncryptionKeyArnT>(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 InvokeInlineAgentRequest& 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 InvokeInlineAgentRequest& WithEndSession(bool value) {
    SetEndSession(value);
    return *this;
  }
 
 
  inline const Aws::String& GetFoundationModel() const { return m_foundationModel; }
  inline bool FoundationModelHasBeenSet() const { return m_foundationModelHasBeenSet; }
  template <typename FoundationModelT = Aws::String>
  void SetFoundationModel(FoundationModelT&& value) {
    m_foundationModelHasBeenSet = true;
    m_foundationModel = std::forward<FoundationModelT>(value);
  }
  template <typename FoundationModelT = Aws::String>
  InvokeInlineAgentRequest& WithFoundationModel(FoundationModelT&& value) {
    SetFoundationModel(std::forward<FoundationModelT>(value));
    return *this;
  }
 
 
  inline const GuardrailConfigurationWithArn& GetGuardrailConfiguration() const { return m_guardrailConfiguration; }
  inline bool GuardrailConfigurationHasBeenSet() const { return m_guardrailConfigurationHasBeenSet; }
  template <typename GuardrailConfigurationT = GuardrailConfigurationWithArn>
  void SetGuardrailConfiguration(GuardrailConfigurationT&& value) {
    m_guardrailConfigurationHasBeenSet = true;
    m_guardrailConfiguration = std::forward<GuardrailConfigurationT>(value);
  }
  template <typename GuardrailConfigurationT = GuardrailConfigurationWithArn>
  InvokeInlineAgentRequest& WithGuardrailConfiguration(GuardrailConfigurationT&& value) {
    SetGuardrailConfiguration(std::forward<GuardrailConfigurationT>(value));
    return *this;
  }
 
 
  inline int GetIdleSessionTTLInSeconds() const { return m_idleSessionTTLInSeconds; }
  inline bool IdleSessionTTLInSecondsHasBeenSet() const { return m_idleSessionTTLInSecondsHasBeenSet; }
  inline void SetIdleSessionTTLInSeconds(int value) {
    m_idleSessionTTLInSecondsHasBeenSet = true;
    m_idleSessionTTLInSeconds = value;
  }
  inline InvokeInlineAgentRequest& WithIdleSessionTTLInSeconds(int value) {
    SetIdleSessionTTLInSeconds(value);
    return *this;
  }
 
 
  inline const InlineSessionState& GetInlineSessionState() const { return m_inlineSessionState; }
  inline bool InlineSessionStateHasBeenSet() const { return m_inlineSessionStateHasBeenSet; }
  template <typename InlineSessionStateT = InlineSessionState>
  void SetInlineSessionState(InlineSessionStateT&& value) {
    m_inlineSessionStateHasBeenSet = true;
    m_inlineSessionState = std::forward<InlineSessionStateT>(value);
  }
  template <typename InlineSessionStateT = InlineSessionState>
  InvokeInlineAgentRequest& WithInlineSessionState(InlineSessionStateT&& value) {
    SetInlineSessionState(std::forward<InlineSessionStateT>(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>
  InvokeInlineAgentRequest& WithInputText(InputTextT&& value) {
    SetInputText(std::forward<InputTextT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetInstruction() const { return m_instruction; }
  inline bool InstructionHasBeenSet() const { return m_instructionHasBeenSet; }
  template <typename InstructionT = Aws::String>
  void SetInstruction(InstructionT&& value) {
    m_instructionHasBeenSet = true;
    m_instruction = std::forward<InstructionT>(value);
  }
  template <typename InstructionT = Aws::String>
  InvokeInlineAgentRequest& WithInstruction(InstructionT&& value) {
    SetInstruction(std::forward<InstructionT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<KnowledgeBase>& GetKnowledgeBases() const { return m_knowledgeBases; }
  inline bool KnowledgeBasesHasBeenSet() const { return m_knowledgeBasesHasBeenSet; }
  template <typename KnowledgeBasesT = Aws::Vector<KnowledgeBase>>
  void SetKnowledgeBases(KnowledgeBasesT&& value) {
    m_knowledgeBasesHasBeenSet = true;
    m_knowledgeBases = std::forward<KnowledgeBasesT>(value);
  }
  template <typename KnowledgeBasesT = Aws::Vector<KnowledgeBase>>
  InvokeInlineAgentRequest& WithKnowledgeBases(KnowledgeBasesT&& value) {
    SetKnowledgeBases(std::forward<KnowledgeBasesT>(value));
    return *this;
  }
  template <typename KnowledgeBasesT = KnowledgeBase>
  InvokeInlineAgentRequest& AddKnowledgeBases(KnowledgeBasesT&& value) {
    m_knowledgeBasesHasBeenSet = true;
    m_knowledgeBases.emplace_back(std::forward<KnowledgeBasesT>(value));
    return *this;
  }
 
 
  inline OrchestrationType GetOrchestrationType() const { return m_orchestrationType; }
  inline bool OrchestrationTypeHasBeenSet() const { return m_orchestrationTypeHasBeenSet; }
  inline void SetOrchestrationType(OrchestrationType value) {
    m_orchestrationTypeHasBeenSet = true;
    m_orchestrationType = value;
  }
  inline InvokeInlineAgentRequest& WithOrchestrationType(OrchestrationType value) {
    SetOrchestrationType(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>
  InvokeInlineAgentRequest& WithPromptCreationConfigurations(PromptCreationConfigurationsT&& value) {
    SetPromptCreationConfigurations(std::forward<PromptCreationConfigurationsT>(value));
    return *this;
  }
 
 
  inline const PromptOverrideConfiguration& GetPromptOverrideConfiguration() const { return m_promptOverrideConfiguration; }
  inline bool PromptOverrideConfigurationHasBeenSet() const { return m_promptOverrideConfigurationHasBeenSet; }
  template <typename PromptOverrideConfigurationT = PromptOverrideConfiguration>
  void SetPromptOverrideConfiguration(PromptOverrideConfigurationT&& value) {
    m_promptOverrideConfigurationHasBeenSet = true;
    m_promptOverrideConfiguration = std::forward<PromptOverrideConfigurationT>(value);
  }
  template <typename PromptOverrideConfigurationT = PromptOverrideConfiguration>
  InvokeInlineAgentRequest& WithPromptOverrideConfiguration(PromptOverrideConfigurationT&& value) {
    SetPromptOverrideConfiguration(std::forward<PromptOverrideConfigurationT>(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>
  InvokeInlineAgentRequest& WithSessionId(SessionIdT&& value) {
    SetSessionId(std::forward<SessionIdT>(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>
  InvokeInlineAgentRequest& WithStreamingConfigurations(StreamingConfigurationsT&& value) {
    SetStreamingConfigurations(std::forward<StreamingConfigurationsT>(value));
    return *this;
  }
 private:
  Aws::Vector<AgentActionGroup> m_actionGroups;
 
  AgentCollaboration m_agentCollaboration{AgentCollaboration::NOT_SET};
 
  Aws::String m_agentName;
 
  InlineBedrockModelConfigurations m_bedrockModelConfigurations;
 
  Aws::Vector<CollaboratorConfiguration> m_collaboratorConfigurations;
 
  Aws::Vector<Collaborator> m_collaborators;
 
  CustomOrchestration m_customOrchestration;
 
  Aws::String m_customerEncryptionKeyArn;
 
  bool m_enableTrace{false};
 
  bool m_endSession{false};
 
  Aws::String m_foundationModel;
 
  GuardrailConfigurationWithArn m_guardrailConfiguration;
 
  int m_idleSessionTTLInSeconds{0};
 
  InlineSessionState m_inlineSessionState;
 
  Aws::String m_inputText;
 
  Aws::String m_instruction;
 
  Aws::Vector<KnowledgeBase> m_knowledgeBases;
 
  OrchestrationType m_orchestrationType{OrchestrationType::NOT_SET};
 
  PromptCreationConfigurations m_promptCreationConfigurations;
 
  PromptOverrideConfiguration m_promptOverrideConfiguration;
 
  Aws::String m_sessionId;
 
  StreamingConfigurations m_streamingConfigurations;
  InvokeInlineAgentHandler m_handler;
  Aws::Utils::Event::EventStreamDecoder m_decoder{Utils::Event::EventStreamDecoder(&m_handler)};
 
  bool m_actionGroupsHasBeenSet = false;
  bool m_agentCollaborationHasBeenSet = false;
  bool m_agentNameHasBeenSet = false;
  bool m_bedrockModelConfigurationsHasBeenSet = false;
  bool m_collaboratorConfigurationsHasBeenSet = false;
  bool m_collaboratorsHasBeenSet = false;
  bool m_customOrchestrationHasBeenSet = false;
  bool m_customerEncryptionKeyArnHasBeenSet = false;
  bool m_enableTraceHasBeenSet = false;
  bool m_endSessionHasBeenSet = false;
  bool m_foundationModelHasBeenSet = false;
  bool m_guardrailConfigurationHasBeenSet = false;
  bool m_idleSessionTTLInSecondsHasBeenSet = false;
  bool m_inlineSessionStateHasBeenSet = false;
  bool m_inputTextHasBeenSet = false;
  bool m_instructionHasBeenSet = false;
  bool m_knowledgeBasesHasBeenSet = false;
  bool m_orchestrationTypeHasBeenSet = false;
  bool m_promptCreationConfigurationsHasBeenSet = false;
  bool m_promptOverrideConfigurationHasBeenSet = false;
  bool m_sessionIdHasBeenSet = false;
  bool m_streamingConfigurationsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws