SessionState.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/ConversationHistory.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/bedrock-agent-runtime/model/InputFile.h>
#include <aws/bedrock-agent-runtime/model/KnowledgeBaseConfiguration.h>
#include <aws/bedrock-agent-runtime/model/InvocationResultMember.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;
    bool m_conversationHistoryHasBeenSet = false;
 
    Aws::Vector<InputFile> m_files;
    bool m_filesHasBeenSet = false;
 
    Aws::String m_invocationId;
    bool m_invocationIdHasBeenSet = false;
 
    Aws::Vector<KnowledgeBaseConfiguration> m_knowledgeBaseConfigurations;
    bool m_knowledgeBaseConfigurationsHasBeenSet = false;
 
    Aws::Map<Aws::String, Aws::String> m_promptSessionAttributes;
    bool m_promptSessionAttributesHasBeenSet = false;
 
    Aws::Vector<InvocationResultMember> m_returnControlInvocationResults;
    bool m_returnControlInvocationResultsHasBeenSet = false;
 
    Aws::Map<Aws::String, Aws::String> m_sessionAttributes;
    bool m_sessionAttributesHasBeenSet = false;
  };
 
} // namespace Model
} // namespace BedrockAgentRuntime
} // namespace Aws