com.databricks.sdk.service.serving.ExternalModelUsageElement Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of databricks-sdk-java Show documentation

There is a newer version: 0.35.0

// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.

package com.databricks.sdk.service.serving;

import com.databricks.sdk.support.Generated;
import com.databricks.sdk.support.ToStringer;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.Objects;

@Generated
public class ExternalModelUsageElement {
  /** The number of tokens in the chat/completions response. */
  @JsonProperty("completion_tokens")
  private Long completionTokens;

  /** The number of tokens in the prompt. */
  @JsonProperty("prompt_tokens")
  private Long promptTokens;

  /** The total number of tokens in the prompt and response. */
  @JsonProperty("total_tokens")
  private Long totalTokens;

  public ExternalModelUsageElement setCompletionTokens(Long completionTokens) {
    this.completionTokens = completionTokens;
    return this;
  }

  public Long getCompletionTokens() {
    return completionTokens;
  }

  public ExternalModelUsageElement setPromptTokens(Long promptTokens) {
    this.promptTokens = promptTokens;
    return this;
  }

  public Long getPromptTokens() {
    return promptTokens;
  }

  public ExternalModelUsageElement setTotalTokens(Long totalTokens) {
    this.totalTokens = totalTokens;
    return this;
  }

  public Long getTotalTokens() {
    return totalTokens;
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) return true;
    if (o == null || getClass() != o.getClass()) return false;
    ExternalModelUsageElement that = (ExternalModelUsageElement) o;
    return Objects.equals(completionTokens, that.completionTokens)
        && Objects.equals(promptTokens, that.promptTokens)
        && Objects.equals(totalTokens, that.totalTokens);
  }

  @Override
  public int hashCode() {
    return Objects.hash(completionTokens, promptTokens, totalTokens);
  }

  @Override
  public String toString() {
    return new ToStringer(ExternalModelUsageElement.class)
        .add("completionTokens", completionTokens)
        .add("promptTokens", promptTokens)
        .add("totalTokens", totalTokens)
        .toString();
  }
}