TokenUsage.h

 
#pragma once
#include <aws/bedrock-runtime/BedrockRuntime_EXPORTS.h>
#include <aws/bedrock-runtime/model/CacheDetail.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 BedrockRuntime {
namespace Model {
 
class TokenUsage {
 public:
  AWS_BEDROCKRUNTIME_API TokenUsage() = default;
  AWS_BEDROCKRUNTIME_API TokenUsage(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKRUNTIME_API TokenUsage& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCKRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline int GetInputTokens() const { return m_inputTokens; }
  inline bool InputTokensHasBeenSet() const { return m_inputTokensHasBeenSet; }
  inline void SetInputTokens(int value) {
    m_inputTokensHasBeenSet = true;
    m_inputTokens = value;
  }
  inline TokenUsage& WithInputTokens(int value) {
    SetInputTokens(value);
    return *this;
  }
 
 
  inline int GetOutputTokens() const { return m_outputTokens; }
  inline bool OutputTokensHasBeenSet() const { return m_outputTokensHasBeenSet; }
  inline void SetOutputTokens(int value) {
    m_outputTokensHasBeenSet = true;
    m_outputTokens = value;
  }
  inline TokenUsage& WithOutputTokens(int value) {
    SetOutputTokens(value);
    return *this;
  }
 
 
  inline int GetTotalTokens() const { return m_totalTokens; }
  inline bool TotalTokensHasBeenSet() const { return m_totalTokensHasBeenSet; }
  inline void SetTotalTokens(int value) {
    m_totalTokensHasBeenSet = true;
    m_totalTokens = value;
  }
  inline TokenUsage& WithTotalTokens(int value) {
    SetTotalTokens(value);
    return *this;
  }
 
 
  inline int GetCacheReadInputTokens() const { return m_cacheReadInputTokens; }
  inline bool CacheReadInputTokensHasBeenSet() const { return m_cacheReadInputTokensHasBeenSet; }
  inline void SetCacheReadInputTokens(int value) {
    m_cacheReadInputTokensHasBeenSet = true;
    m_cacheReadInputTokens = value;
  }
  inline TokenUsage& WithCacheReadInputTokens(int value) {
    SetCacheReadInputTokens(value);
    return *this;
  }
 
 
  inline int GetCacheWriteInputTokens() const { return m_cacheWriteInputTokens; }
  inline bool CacheWriteInputTokensHasBeenSet() const { return m_cacheWriteInputTokensHasBeenSet; }
  inline void SetCacheWriteInputTokens(int value) {
    m_cacheWriteInputTokensHasBeenSet = true;
    m_cacheWriteInputTokens = value;
  }
  inline TokenUsage& WithCacheWriteInputTokens(int value) {
    SetCacheWriteInputTokens(value);
    return *this;
  }
 
 
  inline const Aws::Vector<CacheDetail>& GetCacheDetails() const { return m_cacheDetails; }
  inline bool CacheDetailsHasBeenSet() const { return m_cacheDetailsHasBeenSet; }
  template <typename CacheDetailsT = Aws::Vector<CacheDetail>>
  void SetCacheDetails(CacheDetailsT&& value) {
    m_cacheDetailsHasBeenSet = true;
    m_cacheDetails = std::forward<CacheDetailsT>(value);
  }
  template <typename CacheDetailsT = Aws::Vector<CacheDetail>>
  TokenUsage& WithCacheDetails(CacheDetailsT&& value) {
    SetCacheDetails(std::forward<CacheDetailsT>(value));
    return *this;
  }
  template <typename CacheDetailsT = CacheDetail>
  TokenUsage& AddCacheDetails(CacheDetailsT&& value) {
    m_cacheDetailsHasBeenSet = true;
    m_cacheDetails.emplace_back(std::forward<CacheDetailsT>(value));
    return *this;
  }
 private:
  int m_inputTokens{0};
 
  int m_outputTokens{0};
 
  int m_totalTokens{0};
 
  int m_cacheReadInputTokens{0};
 
  int m_cacheWriteInputTokens{0};
 
  Aws::Vector<CacheDetail> m_cacheDetails;
  bool m_inputTokensHasBeenSet = false;
  bool m_outputTokensHasBeenSet = false;
  bool m_totalTokensHasBeenSet = false;
  bool m_cacheReadInputTokensHasBeenSet = false;
  bool m_cacheWriteInputTokensHasBeenSet = false;
  bool m_cacheDetailsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace BedrockRuntime
}  // namespace Aws