InvokeModelResult.h

 
#pragma once
#include <aws/bedrock-runtime/BedrockRuntime_EXPORTS.h>
#include <aws/bedrock-runtime/model/PerformanceConfigLatency.h>
#include <aws/bedrock-runtime/model/ServiceTierType.h>
#include <aws/core/http/HttpResponse.h>
#include <aws/core/utils/Array.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/stream/ResponseStream.h>
 
#include <utility>
 
namespace Aws {
template <typename RESULT_TYPE>
class AmazonWebServiceResult;
 
namespace BedrockRuntime {
namespace Model {
class InvokeModelResult {
 public:
  AWS_BEDROCKRUNTIME_API InvokeModelResult() = default;
  AWS_BEDROCKRUNTIME_API InvokeModelResult(InvokeModelResult&&) = default;
  AWS_BEDROCKRUNTIME_API InvokeModelResult& operator=(InvokeModelResult&&) = default;
  // we delete these because Microsoft doesn't handle move generation correctly
  // and we therefore don't trust them to get it right here either.
  InvokeModelResult(const InvokeModelResult&) = delete;
  InvokeModelResult& operator=(const InvokeModelResult&) = delete;
 
  AWS_BEDROCKRUNTIME_API InvokeModelResult(Aws::AmazonWebServiceResult<Aws::Utils::Stream::ResponseStream>&& result);
  AWS_BEDROCKRUNTIME_API InvokeModelResult& operator=(Aws::AmazonWebServiceResult<Aws::Utils::Stream::ResponseStream>&& result);
 
 
  inline Aws::IOStream& GetBody() const { return m_body.GetUnderlyingStream(); }
  inline void ReplaceBody(Aws::IOStream* body) { m_body = Aws::Utils::Stream::ResponseStream(body); }
 
 
 
  inline const Aws::String& GetContentType() const { return m_contentType; }
  template <typename ContentTypeT = Aws::String>
  void SetContentType(ContentTypeT&& value) {
    m_contentTypeHasBeenSet = true;
    m_contentType = std::forward<ContentTypeT>(value);
  }
  template <typename ContentTypeT = Aws::String>
  InvokeModelResult& WithContentType(ContentTypeT&& value) {
    SetContentType(std::forward<ContentTypeT>(value));
    return *this;
  }
 
 
  inline PerformanceConfigLatency GetPerformanceConfigLatency() const { return m_performanceConfigLatency; }
  inline void SetPerformanceConfigLatency(PerformanceConfigLatency value) {
    m_performanceConfigLatencyHasBeenSet = true;
    m_performanceConfigLatency = value;
  }
  inline InvokeModelResult& WithPerformanceConfigLatency(PerformanceConfigLatency value) {
    SetPerformanceConfigLatency(value);
    return *this;
  }
 
 
  inline ServiceTierType GetServiceTier() const { return m_serviceTier; }
  inline void SetServiceTier(ServiceTierType value) {
    m_serviceTierHasBeenSet = true;
    m_serviceTier = value;
  }
  inline InvokeModelResult& WithServiceTier(ServiceTierType value) {
    SetServiceTier(value);
    return *this;
  }
 
 
  inline const Aws::String& GetRequestId() const { return m_requestId; }
  template <typename RequestIdT = Aws::String>
  void SetRequestId(RequestIdT&& value) {
    m_requestIdHasBeenSet = true;
    m_requestId = std::forward<RequestIdT>(value);
  }
  template <typename RequestIdT = Aws::String>
  InvokeModelResult& WithRequestId(RequestIdT&& value) {
    SetRequestId(std::forward<RequestIdT>(value));
    return *this;
  }
  inline Aws::Http::HttpResponseCode GetHttpResponseCode() const { return m_HttpResponseCode; }
 
 private:
  Aws::Utils::Stream::ResponseStream m_body{};
 
  Aws::String m_contentType;
 
  PerformanceConfigLatency m_performanceConfigLatency{PerformanceConfigLatency::NOT_SET};
 
  ServiceTierType m_serviceTier{ServiceTierType::NOT_SET};
 
  Aws::String m_requestId;
  Aws::Http::HttpResponseCode m_HttpResponseCode;
  bool m_bodyHasBeenSet = false;
  bool m_contentTypeHasBeenSet = false;
  bool m_performanceConfigLatencyHasBeenSet = false;
  bool m_serviceTierHasBeenSet = false;
  bool m_requestIdHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockRuntime
}  // namespace Aws