Observation.h

 
#pragma once
#include <aws/bedrock-agent-runtime/BedrockAgentRuntime_EXPORTS.h>
#include <aws/bedrock-agent-runtime/model/ActionGroupInvocationOutput.h>
#include <aws/bedrock-agent-runtime/model/AgentCollaboratorInvocationOutput.h>
#include <aws/bedrock-agent-runtime/model/CodeInterpreterInvocationOutput.h>
#include <aws/bedrock-agent-runtime/model/FinalResponse.h>
#include <aws/bedrock-agent-runtime/model/KnowledgeBaseLookupOutput.h>
#include <aws/bedrock-agent-runtime/model/RepromptResponse.h>
#include <aws/bedrock-agent-runtime/model/Type.h>
#include <aws/core/utils/memory/stl/AWSString.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace BedrockAgentRuntime {
namespace Model {
 
class Observation {
 public:
  AWS_BEDROCKAGENTRUNTIME_API Observation() = default;
  AWS_BEDROCKAGENTRUNTIME_API Observation(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Observation& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKAGENTRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const ActionGroupInvocationOutput& GetActionGroupInvocationOutput() const { return m_actionGroupInvocationOutput; }
  inline bool ActionGroupInvocationOutputHasBeenSet() const { return m_actionGroupInvocationOutputHasBeenSet; }
  template <typename ActionGroupInvocationOutputT = ActionGroupInvocationOutput>
  void SetActionGroupInvocationOutput(ActionGroupInvocationOutputT&& value) {
    m_actionGroupInvocationOutputHasBeenSet = true;
    m_actionGroupInvocationOutput = std::forward<ActionGroupInvocationOutputT>(value);
  }
  template <typename ActionGroupInvocationOutputT = ActionGroupInvocationOutput>
  Observation& WithActionGroupInvocationOutput(ActionGroupInvocationOutputT&& value) {
    SetActionGroupInvocationOutput(std::forward<ActionGroupInvocationOutputT>(value));
    return *this;
  }
 
 
  inline const AgentCollaboratorInvocationOutput& GetAgentCollaboratorInvocationOutput() const {
    return m_agentCollaboratorInvocationOutput;
  }
  inline bool AgentCollaboratorInvocationOutputHasBeenSet() const { return m_agentCollaboratorInvocationOutputHasBeenSet; }
  template <typename AgentCollaboratorInvocationOutputT = AgentCollaboratorInvocationOutput>
  void SetAgentCollaboratorInvocationOutput(AgentCollaboratorInvocationOutputT&& value) {
    m_agentCollaboratorInvocationOutputHasBeenSet = true;
    m_agentCollaboratorInvocationOutput = std::forward<AgentCollaboratorInvocationOutputT>(value);
  }
  template <typename AgentCollaboratorInvocationOutputT = AgentCollaboratorInvocationOutput>
  Observation& WithAgentCollaboratorInvocationOutput(AgentCollaboratorInvocationOutputT&& value) {
    SetAgentCollaboratorInvocationOutput(std::forward<AgentCollaboratorInvocationOutputT>(value));
    return *this;
  }
 
 
  inline const CodeInterpreterInvocationOutput& GetCodeInterpreterInvocationOutput() const { return m_codeInterpreterInvocationOutput; }
  inline bool CodeInterpreterInvocationOutputHasBeenSet() const { return m_codeInterpreterInvocationOutputHasBeenSet; }
  template <typename CodeInterpreterInvocationOutputT = CodeInterpreterInvocationOutput>
  void SetCodeInterpreterInvocationOutput(CodeInterpreterInvocationOutputT&& value) {
    m_codeInterpreterInvocationOutputHasBeenSet = true;
    m_codeInterpreterInvocationOutput = std::forward<CodeInterpreterInvocationOutputT>(value);
  }
  template <typename CodeInterpreterInvocationOutputT = CodeInterpreterInvocationOutput>
  Observation& WithCodeInterpreterInvocationOutput(CodeInterpreterInvocationOutputT&& value) {
    SetCodeInterpreterInvocationOutput(std::forward<CodeInterpreterInvocationOutputT>(value));
    return *this;
  }
 
 
  inline const FinalResponse& GetFinalResponse() const { return m_finalResponse; }
  inline bool FinalResponseHasBeenSet() const { return m_finalResponseHasBeenSet; }
  template <typename FinalResponseT = FinalResponse>
  void SetFinalResponse(FinalResponseT&& value) {
    m_finalResponseHasBeenSet = true;
    m_finalResponse = std::forward<FinalResponseT>(value);
  }
  template <typename FinalResponseT = FinalResponse>
  Observation& WithFinalResponse(FinalResponseT&& value) {
    SetFinalResponse(std::forward<FinalResponseT>(value));
    return *this;
  }
 
 
  inline const KnowledgeBaseLookupOutput& GetKnowledgeBaseLookupOutput() const { return m_knowledgeBaseLookupOutput; }
  inline bool KnowledgeBaseLookupOutputHasBeenSet() const { return m_knowledgeBaseLookupOutputHasBeenSet; }
  template <typename KnowledgeBaseLookupOutputT = KnowledgeBaseLookupOutput>
  void SetKnowledgeBaseLookupOutput(KnowledgeBaseLookupOutputT&& value) {
    m_knowledgeBaseLookupOutputHasBeenSet = true;
    m_knowledgeBaseLookupOutput = std::forward<KnowledgeBaseLookupOutputT>(value);
  }
  template <typename KnowledgeBaseLookupOutputT = KnowledgeBaseLookupOutput>
  Observation& WithKnowledgeBaseLookupOutput(KnowledgeBaseLookupOutputT&& value) {
    SetKnowledgeBaseLookupOutput(std::forward<KnowledgeBaseLookupOutputT>(value));
    return *this;
  }
 
 
  inline const RepromptResponse& GetRepromptResponse() const { return m_repromptResponse; }
  inline bool RepromptResponseHasBeenSet() const { return m_repromptResponseHasBeenSet; }
  template <typename RepromptResponseT = RepromptResponse>
  void SetRepromptResponse(RepromptResponseT&& value) {
    m_repromptResponseHasBeenSet = true;
    m_repromptResponse = std::forward<RepromptResponseT>(value);
  }
  template <typename RepromptResponseT = RepromptResponse>
  Observation& WithRepromptResponse(RepromptResponseT&& value) {
    SetRepromptResponse(std::forward<RepromptResponseT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTraceId() const { return m_traceId; }
  inline bool TraceIdHasBeenSet() const { return m_traceIdHasBeenSet; }
  template <typename TraceIdT = Aws::String>
  void SetTraceId(TraceIdT&& value) {
    m_traceIdHasBeenSet = true;
    m_traceId = std::forward<TraceIdT>(value);
  }
  template <typename TraceIdT = Aws::String>
  Observation& WithTraceId(TraceIdT&& value) {
    SetTraceId(std::forward<TraceIdT>(value));
    return *this;
  }
 
 
  inline Type GetType() const { return m_type; }
  inline bool TypeHasBeenSet() const { return m_typeHasBeenSet; }
  inline void SetType(Type value) {
    m_typeHasBeenSet = true;
    m_type = value;
  }
  inline Observation& WithType(Type value) {
    SetType(value);
    return *this;
  }
 private:
  ActionGroupInvocationOutput m_actionGroupInvocationOutput;
 
  AgentCollaboratorInvocationOutput m_agentCollaboratorInvocationOutput;
 
  CodeInterpreterInvocationOutput m_codeInterpreterInvocationOutput;
 
  FinalResponse m_finalResponse;
 
  KnowledgeBaseLookupOutput m_knowledgeBaseLookupOutput;
 
  RepromptResponse m_repromptResponse;
 
  Aws::String m_traceId;
 
  Type m_type{Type::NOT_SET};
  bool m_actionGroupInvocationOutputHasBeenSet = false;
  bool m_agentCollaboratorInvocationOutputHasBeenSet = false;
  bool m_codeInterpreterInvocationOutputHasBeenSet = false;
  bool m_finalResponseHasBeenSet = false;
  bool m_knowledgeBaseLookupOutputHasBeenSet = false;
  bool m_repromptResponseHasBeenSet = false;
  bool m_traceIdHasBeenSet = false;
  bool m_typeHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockAgentRuntime
}  // namespace Aws