OllamaResult.java

Go to the documentation of this file.

package io.github.ollama4j.models.response;
 
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
 
import lombok.Data;
import lombok.Getter;
 
import static io.github.ollama4j.utils.Utils.getObjectMapper;
 
import java.util.HashMap;
import java.util.Map;

@Getter
@SuppressWarnings("unused")
@Data
@JsonIgnoreProperties(ignoreUnknown = true)
public class OllamaResult {
  private final String response;

  private int httpStatusCode;

  private long responseTime = 0;
 
  public OllamaResult(String response, long responseTime, int httpStatusCode) {
    this.response = response;
    this.responseTime = responseTime;
    this.httpStatusCode = httpStatusCode;
  }
 
  @Override
  public String toString() {
    try {
      Map<String, Object> responseMap = new HashMap<>();
      responseMap.put("response", this.response);
      responseMap.put("httpStatusCode", this.httpStatusCode);
      responseMap.put("responseTime", this.responseTime);
      return getObjectMapper().writerWithDefaultPrettyPrinter().writeValueAsString(responseMap);
    } catch (JsonProcessingException e) {
      throw new RuntimeException(e);
    }
  }

  public Map<String, Object> getStructuredResponse() {
    String responseStr = this.getResponse();
    if (responseStr == null || responseStr.trim().isEmpty()) {
      throw new IllegalArgumentException("Response is empty or null");
    }
 
    try {
      // Check if the response is a valid JSON
      if ((!responseStr.trim().startsWith("{") && !responseStr.trim().startsWith("[")) ||
          (!responseStr.trim().endsWith("}") && !responseStr.trim().endsWith("]"))) {
        throw new IllegalArgumentException("Response is not a valid JSON object");
      }
 
      Map<String, Object> response = getObjectMapper().readValue(responseStr,
          new TypeReference<Map<String, Object>>() {
          });
      return response;
    } catch (JsonProcessingException e) {
      throw new IllegalArgumentException("Failed to parse response as JSON: " + e.getMessage(), e);
    }
  }

  public <T> T as(Class<T> clazz) {
    String responseStr = this.getResponse();
    if (responseStr == null || responseStr.trim().isEmpty()) {
      throw new IllegalArgumentException("Response is empty or null");
    }
 
    try {
      // Check if the response is a valid JSON
      if ((!responseStr.trim().startsWith("{") && !responseStr.trim().startsWith("[")) ||
          (!responseStr.trim().endsWith("}") && !responseStr.trim().endsWith("]"))) {
        throw new IllegalArgumentException("Response is not a valid JSON object");
      }
      return getObjectMapper().readValue(responseStr, clazz);
    } catch (JsonProcessingException e) {
      throw new IllegalArgumentException("Failed to parse response as JSON: " + e.getMessage(), e);
    }
  }
}