Updated ModelsProcessResponse class to support ps() API

Signed-off-by: Amith Koujalgi <koujalgi.amith@gmail.com>
This commit is contained in:
Amith Koujalgi 2024-08-06 11:11:15 +05:30
parent 75e6576a13
commit 422601c0fc

View File

@ -1,6 +1,7 @@
package io.github.ollama4j.models.ps; package io.github.ollama4j.models.ps;
import com.fasterxml.jackson.annotation.JsonIgnoreProperties; import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data; import lombok.Data;
import lombok.NoArgsConstructor; import lombok.NoArgsConstructor;
@ -10,28 +11,53 @@ import java.util.List;
@NoArgsConstructor @NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true) @JsonIgnoreProperties(ignoreUnknown = true)
public class ModelsProcessResponse { public class ModelsProcessResponse {
@JsonProperty("models")
private List<ModelProcess> models; private List<ModelProcess> models;
@Data @Data
@NoArgsConstructor @NoArgsConstructor
public static class ModelProcess { public static class ModelProcess {
@JsonProperty("name")
private String name; private String name;
@JsonProperty("model")
private String model; private String model;
@JsonProperty("size")
private long size; private long size;
@JsonProperty("digest")
private String digest; private String digest;
@JsonProperty("details")
private ModelDetails details; private ModelDetails details;
private String expiresAt;
@JsonProperty("expires_at")
private String expiresAt; // Consider using LocalDateTime if you need to process date/time
@JsonProperty("size_vram")
private long sizeVram; private long sizeVram;
} }
@Data @Data
@NoArgsConstructor @NoArgsConstructor
public static class ModelDetails { public static class ModelDetails {
@JsonProperty("parent_model")
private String parentModel; private String parentModel;
@JsonProperty("format")
private String format; private String format;
@JsonProperty("family")
private String family; private String family;
@JsonProperty("families")
private List<String> families; private List<String> families;
@JsonProperty("parameter_size")
private String parameterSize; private String parameterSize;
@JsonProperty("quantization_level")
private String quantizationLevel; private String quantizationLevel;
} }
} }