Skip to content

Commit c7c941c

Browse files
committed
update test resources
1 parent 2c5c6b8 commit c7c941c

File tree

5 files changed

+10
-12
lines changed

5 files changed

+10
-12
lines changed

docs/code_samples/v2_extraction.txt

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ import com.mindee.MindeeClientV2;
22
import com.mindee.InferenceParameters;
33
import com.mindee.input.LocalInputSource;
44
import com.mindee.parsing.v2.InferenceResponse;
5-
import java.io.File;
65
import java.io.IOException;
76

87
public class SimpleMindeeClientV2 {
@@ -19,7 +18,7 @@ public class SimpleMindeeClientV2 {
1918

2019
// Set inference parameters
2120
// Note: modelId is mandatory.
22-
InferenceParameters inferenceParams = InferenceParameters
21+
InferenceParameters extractionParams = InferenceParameters
2322
// ID of the model, required.
2423
.builder(modelId)
2524

@@ -38,14 +37,13 @@ public class SimpleMindeeClientV2 {
3837
.build();
3938

4039
// Load a file from disk
41-
LocalInputSource inputSource = new LocalInputSource(
42-
new File(filePath)
43-
);
40+
LocalInputSource inputSource = new LocalInputSource(filePath);
4441

4542
// Send for processing
46-
InferenceResponse response = mindeeClient.enqueueAndGetInference(
43+
InferenceResponse response = mindeeClient.enqueueAndGetResult(
44+
InferenceResponse.class,
4745
inputSource,
48-
inferenceParams
46+
extractionParams
4947
);
5048

5149
// Print a summary of the response

src/main/java/com/mindee/MindeeClientV2.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ public JobResponse enqueueInference(
4848
*/
4949
public JobResponse enqueueInference(
5050
URLInputSource inputSource,
51-
BaseParameters params
51+
InferenceParameters params
5252
) throws IOException {
5353
return enqueue(inputSource, params);
5454
}

src/test/java/com/mindee/v2/product/CropTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ void mustHaveValidProperties() throws IOException {
6868
assertEquals(0, crop1.getLocation().getPage());
6969

7070
CropItem crop2 = crops.get(1);
71-
assertEquals("invoice", crop2.getObjectType());
71+
assertEquals("receipt", crop2.getObjectType());
7272
assertNotNull(crop2.getLocation().getPolygon());
7373
assertEquals(0, crop2.getLocation().getPage());
7474
}

src/test/java/com/mindee/v2/product/SplitTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,15 +49,15 @@ void mustHaveValidProperties() throws IOException {
4949
assertEquals(3, splits.size());
5050

5151
SplitRange split1 = splits.get(0);
52-
assertEquals("invoice", split1.getDocumentType());
52+
assertEquals("passport", split1.getDocumentType());
5353
assertEquals(0, split1.getPageRange().get(0));
5454

5555
SplitRange split2 = splits.get(1);
5656
assertEquals("invoice", split2.getDocumentType());
5757
assertEquals(1, split2.getPageRange().get(0));
5858

5959
SplitRange split3 = splits.get(2);
60-
assertEquals("invoice", split3.getDocumentType());
60+
assertEquals("receipt", split3.getDocumentType());
6161
assertEquals(4, split3.getPageRange().get(0));
6262
}
6363
}

0 commit comments

Comments
 (0)