SessionState.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/ConversationHistory.h>
#include <aws/bedrock-agent-runtime/model/InputFile.h>
#include <aws/bedrock-agent-runtime/model/InvocationResultMember.h>
#include <aws/bedrock-agent-runtime/model/KnowledgeBaseConfiguration.h>
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgentRuntime {
namespace Model {
 
class SessionState {
 public:
  AWS_BEDROCKAGENTRUNTIME_API SessionState() = default;
  AWS_BEDROCKAGENTRUNTIME_API SessionState(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API SessionState& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const ConversationHistory& GetConversationHistory() const { return m_conversationHistory; }
  inline bool ConversationHistoryHasBeenSet() const { return m_conversationHistoryHasBeenSet; }
  template <typename ConversationHistoryT = ConversationHistory>
  void SetConversationHistory(ConversationHistoryT&& value) {
    m_conversationHistoryHasBeenSet = true;
    m_conversationHistory = std::forward<ConversationHistoryT>(value);
  }
  template <typename ConversationHistoryT = ConversationHistory>
  SessionState& WithConversationHistory(ConversationHistoryT&& value) {
    SetConversationHistory(std::forward<ConversationHistoryT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<InputFile>& GetFiles() const { return m_files; }
  inline bool FilesHasBeenSet() const { return m_filesHasBeenSet; }
  template <typename FilesT = Aws::Vector<InputFile>>
  void SetFiles(FilesT&& value) {
    m_filesHasBeenSet = true;
    m_files = std::forward<FilesT>(value);
  }
  template <typename FilesT = Aws::Vector<InputFile>>
  SessionState& WithFiles(FilesT&& value) {
    SetFiles(std::forward<FilesT>(value));
    return *this;
  }
  template <typename FilesT = InputFile>
  SessionState& AddFiles(FilesT&& value) {
    m_filesHasBeenSet = true;
    m_files.emplace_back(std::forward<FilesT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetInvocationId() const { return m_invocationId; }
  inline bool InvocationIdHasBeenSet() const { return m_invocationIdHasBeenSet; }
  template <typename InvocationIdT = Aws::String>
  void SetInvocationId(InvocationIdT&& value) {
    m_invocationIdHasBeenSet = true;
    m_invocationId = std::forward<InvocationIdT>(value);
  }
  template <typename InvocationIdT = Aws::String>
  SessionState& WithInvocationId(InvocationIdT&& value) {
    SetInvocationId(std::forward<InvocationIdT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<KnowledgeBaseConfiguration>& GetKnowledgeBaseConfigurations() const { return m_knowledgeBaseConfigurations; }
  inline bool KnowledgeBaseConfigurationsHasBeenSet() const { return m_knowledgeBaseConfigurationsHasBeenSet; }
  template <typename KnowledgeBaseConfigurationsT = Aws::Vector<KnowledgeBaseConfiguration>>
  void SetKnowledgeBaseConfigurations(KnowledgeBaseConfigurationsT&& value) {
    m_knowledgeBaseConfigurationsHasBeenSet = true;
    m_knowledgeBaseConfigurations = std::forward<KnowledgeBaseConfigurationsT>(value);
  }
  template <typename KnowledgeBaseConfigurationsT = Aws::Vector<KnowledgeBaseConfiguration>>
  SessionState& WithKnowledgeBaseConfigurations(KnowledgeBaseConfigurationsT&& value) {
    SetKnowledgeBaseConfigurations(std::forward<KnowledgeBaseConfigurationsT>(value));
    return *this;
  }
  template <typename KnowledgeBaseConfigurationsT = KnowledgeBaseConfiguration>
  SessionState& AddKnowledgeBaseConfigurations(KnowledgeBaseConfigurationsT&& value) {
    m_knowledgeBaseConfigurationsHasBeenSet = true;
    m_knowledgeBaseConfigurations.emplace_back(std::forward<KnowledgeBaseConfigurationsT>(value));
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, Aws::String>& GetPromptSessionAttributes() const { return m_promptSessionAttributes; }
  inline bool PromptSessionAttributesHasBeenSet() const { return m_promptSessionAttributesHasBeenSet; }
  template <typename PromptSessionAttributesT = Aws::Map<Aws::String, Aws::String>>
  void SetPromptSessionAttributes(PromptSessionAttributesT&& value) {
    m_promptSessionAttributesHasBeenSet = true;
    m_promptSessionAttributes = std::forward<PromptSessionAttributesT>(value);
  }
  template <typename PromptSessionAttributesT = Aws::Map<Aws::String, Aws::String>>
  SessionState& WithPromptSessionAttributes(PromptSessionAttributesT&& value) {
    SetPromptSessionAttributes(std::forward<PromptSessionAttributesT>(value));
    return *this;
  }
  template <typename PromptSessionAttributesKeyT = Aws::String, typename PromptSessionAttributesValueT = Aws::String>
  SessionState& AddPromptSessionAttributes(PromptSessionAttributesKeyT&& key, PromptSessionAttributesValueT&& value) {
    m_promptSessionAttributesHasBeenSet = true;
    m_promptSessionAttributes.emplace(std::forward<PromptSessionAttributesKeyT>(key), std::forward<PromptSessionAttributesValueT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<InvocationResultMember>& GetReturnControlInvocationResults() const { return m_returnControlInvocationResults; }
  inline bool ReturnControlInvocationResultsHasBeenSet() const { return m_returnControlInvocationResultsHasBeenSet; }
  template <typename ReturnControlInvocationResultsT = Aws::Vector<InvocationResultMember>>
  void SetReturnControlInvocationResults(ReturnControlInvocationResultsT&& value) {
    m_returnControlInvocationResultsHasBeenSet = true;
    m_returnControlInvocationResults = std::forward<ReturnControlInvocationResultsT>(value);
  }
  template <typename ReturnControlInvocationResultsT = Aws::Vector<InvocationResultMember>>
  SessionState& WithReturnControlInvocationResults(ReturnControlInvocationResultsT&& value) {
    SetReturnControlInvocationResults(std::forward<ReturnControlInvocationResultsT>(value));
    return *this;
  }
  template <typename ReturnControlInvocationResultsT = InvocationResultMember>
  SessionState& AddReturnControlInvocationResults(ReturnControlInvocationResultsT&& value) {
    m_returnControlInvocationResultsHasBeenSet = true;
    m_returnControlInvocationResults.emplace_back(std::forward<ReturnControlInvocationResultsT>(value));
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, Aws::String>& GetSessionAttributes() const { return m_sessionAttributes; }
  inline bool SessionAttributesHasBeenSet() const { return m_sessionAttributesHasBeenSet; }
  template <typename SessionAttributesT = Aws::Map<Aws::String, Aws::String>>
  void SetSessionAttributes(SessionAttributesT&& value) {
    m_sessionAttributesHasBeenSet = true;
    m_sessionAttributes = std::forward<SessionAttributesT>(value);
  }
  template <typename SessionAttributesT = Aws::Map<Aws::String, Aws::String>>
  SessionState& WithSessionAttributes(SessionAttributesT&& value) {
    SetSessionAttributes(std::forward<SessionAttributesT>(value));
    return *this;
  }
  template <typename SessionAttributesKeyT = Aws::String, typename SessionAttributesValueT = Aws::String>
  SessionState& AddSessionAttributes(SessionAttributesKeyT&& key, SessionAttributesValueT&& value) {
    m_sessionAttributesHasBeenSet = true;
    m_sessionAttributes.emplace(std::forward<SessionAttributesKeyT>(key), std::forward<SessionAttributesValueT>(value));
    return *this;
  }
 private:
  ConversationHistory m_conversationHistory;
 
  Aws::Vector<InputFile> m_files;
 
  Aws::String m_invocationId;
 
  Aws::Vector<KnowledgeBaseConfiguration> m_knowledgeBaseConfigurations;
 
  Aws::Map<Aws::String, Aws::String> m_promptSessionAttributes;
 
  Aws::Vector<InvocationResultMember> m_returnControlInvocationResults;
 
  Aws::Map<Aws::String, Aws::String> m_sessionAttributes;
  bool m_conversationHistoryHasBeenSet = false;
  bool m_filesHasBeenSet = false;
  bool m_invocationIdHasBeenSet = false;
  bool m_knowledgeBaseConfigurationsHasBeenSet = false;
  bool m_promptSessionAttributesHasBeenSet = false;
  bool m_returnControlInvocationResultsHasBeenSet = false;
  bool m_sessionAttributesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws