OllamaStructuredResult.java

Go to the documentation of this file.

package io.github.ollama4j.models.response;
 
import static io.github.ollama4j.utils.Utils.getObjectMapper;
 
import java.util.Map;
 
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
 
import lombok.Data;
import lombok.Getter;
import lombok.NoArgsConstructor;
 
@Getter
@SuppressWarnings("unused")
@Data
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public class OllamaStructuredResult {
  private String response;
 
  private int httpStatusCode;
 
  private long responseTime = 0;
 
  private String model;
 
  public OllamaStructuredResult(String response, long responseTime, int httpStatusCode) {
    this.response = response;
    this.responseTime = responseTime;
    this.httpStatusCode = httpStatusCode;
  }
 
  @Override
  public String toString() {
    try {
      return getObjectMapper().writerWithDefaultPrettyPrinter().writeValueAsString(this);
    } catch (JsonProcessingException e) {
      throw new RuntimeException(e);
    }
  }

  public Map<String, Object> getStructuredResponse() {
    try {
      Map<String, Object> response = getObjectMapper().readValue(this.getResponse(),
          new TypeReference<Map<String, Object>>() {
          });
      return response;
    } catch (JsonProcessingException e) {
      throw new RuntimeException(e);
    }
  }

  public <T> T getStructuredResponse(Class<T> clazz) {
    try {
      return getObjectMapper().readValue(this.getResponse(), clazz);
    } catch (JsonProcessingException e) {
      throw new RuntimeException(e);
    }
  }
}