diff --git a/src/main/java/com/mindee/parsing/v2/Inference.java b/src/main/java/com/mindee/parsing/v2/Inference.java index 3b50dd6fe..db37e638b 100644 --- a/src/main/java/com/mindee/parsing/v2/Inference.java +++ b/src/main/java/com/mindee/parsing/v2/Inference.java @@ -27,13 +27,13 @@ public class Inference { * Model info. */ @JsonProperty("model") - private InferenceResultModel model; + private InferenceModel model; /** * File info. */ @JsonProperty("file") - private InferenceResultFile file; + private InferenceFile file; /** * Model result values. @@ -51,9 +51,7 @@ public String toString() { .add("=====") .add(":ID: " + (model != null ? model.getId() : "")) .add("") - .add("File") - .add("====") - .add(file != null ? file.toString() : "") + .add(file.toString()) .add("") .add(result != null ? result.toString() : ""); return joiner.toString().trim() + "\n"; diff --git a/src/main/java/com/mindee/parsing/v2/InferenceResultFile.java b/src/main/java/com/mindee/parsing/v2/InferenceFile.java similarity index 52% rename from src/main/java/com/mindee/parsing/v2/InferenceResultFile.java rename to src/main/java/com/mindee/parsing/v2/InferenceFile.java index 5296ae5e9..67e0b3772 100644 --- a/src/main/java/com/mindee/parsing/v2/InferenceResultFile.java +++ b/src/main/java/com/mindee/parsing/v2/InferenceFile.java @@ -2,6 +2,7 @@ import com.fasterxml.jackson.annotation.JsonIgnoreProperties; import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.StringJoiner; import lombok.AllArgsConstructor; import lombok.EqualsAndHashCode; import lombok.Getter; @@ -15,7 +16,7 @@ @JsonIgnoreProperties(ignoreUnknown = true) @AllArgsConstructor @NoArgsConstructor -public class InferenceResultFile { +public class InferenceFile { /** * File name. */ @@ -28,7 +29,29 @@ public class InferenceResultFile { @JsonProperty("alias") private String alias; + /** + * Page Count. + */ + @JsonProperty("page_count") + private int pageCount; + + /** + * Mime type, as read by the server. + */ + @JsonProperty("mime_type") + private String mimeType; + + public String toString() { - return ":Name: " + name + "\n:Alias:" + (alias != null ? " " + alias : ""); + + StringJoiner joiner = new StringJoiner("\n"); + joiner + .add("File") + .add("====") + .add(":Name: " + name) + .add(":Alias:" + (alias != null ? " " + alias : "")) + .add(":Page Count: " + pageCount) + .add(":MIME Type: " + mimeType); + return joiner.toString(); } } diff --git a/src/main/java/com/mindee/parsing/v2/InferenceResultModel.java b/src/main/java/com/mindee/parsing/v2/InferenceModel.java similarity index 93% rename from src/main/java/com/mindee/parsing/v2/InferenceResultModel.java rename to src/main/java/com/mindee/parsing/v2/InferenceModel.java index 70cf0e80d..02c24f74f 100644 --- a/src/main/java/com/mindee/parsing/v2/InferenceResultModel.java +++ b/src/main/java/com/mindee/parsing/v2/InferenceModel.java @@ -15,7 +15,7 @@ @JsonIgnoreProperties(ignoreUnknown = true) @AllArgsConstructor @NoArgsConstructor -public class InferenceResultModel { +public class InferenceModel { /** * The ID of the model. diff --git a/src/main/java/com/mindee/parsing/v2/Job.java b/src/main/java/com/mindee/parsing/v2/Job.java index d08f05e23..db8ee360e 100644 --- a/src/main/java/com/mindee/parsing/v2/Job.java +++ b/src/main/java/com/mindee/parsing/v2/Job.java @@ -80,5 +80,5 @@ public final class Job { * Polling URL. */ @JsonProperty("webhooks") - private List webhooks; + private List webhooks; } diff --git a/src/main/java/com/mindee/parsing/v2/JobResponseWebhook.java b/src/main/java/com/mindee/parsing/v2/JobWebhook.java similarity index 93% rename from src/main/java/com/mindee/parsing/v2/JobResponseWebhook.java rename to src/main/java/com/mindee/parsing/v2/JobWebhook.java index 09ca9f123..1a892eb69 100644 --- a/src/main/java/com/mindee/parsing/v2/JobResponseWebhook.java +++ b/src/main/java/com/mindee/parsing/v2/JobWebhook.java @@ -11,14 +11,14 @@ import lombok.NoArgsConstructor; /** - * JobResponseWebhook info. + * JobWebhook info. */ @Getter @EqualsAndHashCode @JsonIgnoreProperties(ignoreUnknown = true) @AllArgsConstructor @NoArgsConstructor -public final class JobResponseWebhook { +public final class JobWebhook { /** * ID of the webhook. diff --git a/src/test/java/com/mindee/parsing/v2/InferenceTest.java b/src/test/java/com/mindee/parsing/v2/InferenceTest.java index 64b121bad..865427adc 100644 --- a/src/test/java/com/mindee/parsing/v2/InferenceTest.java +++ b/src/test/java/com/mindee/parsing/v2/InferenceTest.java @@ -95,13 +95,15 @@ void asyncPredict_whenComplete_mustExposeAllProperties() throws IOException { assertNotNull(inf, "Inference must not be null"); assertEquals("12345678-1234-1234-1234-123456789abc", inf.getId(), "Inference ID mismatch"); - InferenceResultModel model = inf.getModel(); + InferenceModel model = inf.getModel(); assertNotNull(model, "Model must not be null"); assertEquals("12345678-1234-1234-1234-123456789abc", model.getId(), "Model ID mismatch"); - InferenceResultFile file = inf.getFile(); + InferenceFile file = inf.getFile(); assertNotNull(file, "File must not be null"); assertEquals("complete.jpg", file.getName(), "File name mismatch"); + assertEquals(1, file.getPageCount(), "Page count mismatch"); + assertEquals("image/jpeg", file.getMimeType(), "MIME type mismatch"); assertNull(file.getAlias(), "File alias must be null for this payload"); InferenceFields fields = inf.getResult().getFields(); diff --git a/src/test/resources b/src/test/resources index 02ace39f3..632af76d6 160000 --- a/src/test/resources +++ b/src/test/resources @@ -1 +1 @@ -Subproject commit 02ace39f3b8cdd99dcac4f060d5b24b67ff5f2ab +Subproject commit 632af76d6eefe551cdeb2b7fa2f325cecec1b66f