diff --git a/src/main/java/com/mindee/v2/cli/BaseInferenceCommand.java b/src/main/java/com/mindee/v2/cli/BaseInferenceCommand.java index 411774590..fbf08d084 100644 --- a/src/main/java/com/mindee/v2/cli/BaseInferenceCommand.java +++ b/src/main/java/com/mindee/v2/cli/BaseInferenceCommand.java @@ -6,6 +6,7 @@ import com.mindee.v2.MindeeClient; import com.mindee.v2.parsing.CommonResponse; import java.io.File; +import java.util.List; import java.util.concurrent.Callable; import picocli.CommandLine.Option; import picocli.CommandLine.Parameters; @@ -28,6 +29,12 @@ public abstract class BaseInferenceCommand implements Callable { @Option(names = { "-a", "--alias" }, description = "Alias for the file") protected String alias; + @Option( + names = { "-w", "--webhook-id" }, + description = "Specify a webhook by ID. May be used multiple times." + ) + private List webhookIds; + /** Output format for the command. */ public enum OutputType { summary, @@ -45,6 +52,13 @@ public enum OutputType { ) protected OutputType output; + /** + * @return The properly formatted webhook IDs. + */ + protected String[] getWebhookIds() { + return (webhookIds != null ? webhookIds.toArray(new String[0]) : new String[] {}); + } + /** * Executes the inference request and returns the product response. * diff --git a/src/main/java/com/mindee/v2/cli/ClassificationCommand.java b/src/main/java/com/mindee/v2/cli/ClassificationCommand.java index 1f7844389..12b9ea55e 100644 --- a/src/main/java/com/mindee/v2/cli/ClassificationCommand.java +++ b/src/main/java/com/mindee/v2/cli/ClassificationCommand.java @@ -26,7 +26,7 @@ protected CommonResponse executeRequest( .enqueueAndGetResult( ClassificationResponse.class, inputSource, - ClassificationParameters.builder(modelId).alias(alias).build() + ClassificationParameters.builder(modelId).alias(alias).webhookIds(getWebhookIds()).build() ); } diff --git a/src/main/java/com/mindee/v2/cli/CropCommand.java b/src/main/java/com/mindee/v2/cli/CropCommand.java index 2431f7bb5..4ccabf5d5 100644 --- a/src/main/java/com/mindee/v2/cli/CropCommand.java +++ b/src/main/java/com/mindee/v2/cli/CropCommand.java @@ -22,7 +22,7 @@ protected CommonResponse executeRequest( .enqueueAndGetResult( CropResponse.class, inputSource, - CropParameters.builder(modelId).alias(alias).build() + CropParameters.builder(modelId).alias(alias).webhookIds(getWebhookIds()).build() ); } diff --git a/src/main/java/com/mindee/v2/cli/ExtractionCommand.java b/src/main/java/com/mindee/v2/cli/ExtractionCommand.java index 8b11fc949..464504372 100644 --- a/src/main/java/com/mindee/v2/cli/ExtractionCommand.java +++ b/src/main/java/com/mindee/v2/cli/ExtractionCommand.java @@ -66,6 +66,7 @@ protected CommonResponse executeRequest( ExtractionParameters .builder(modelId) .alias(alias) + .webhookIds(getWebhookIds()) .rag(rag) .rawText(rawText) .confidence(confidence) diff --git a/src/main/java/com/mindee/v2/cli/OcrCommand.java b/src/main/java/com/mindee/v2/cli/OcrCommand.java index 14664b281..771b887f1 100644 --- a/src/main/java/com/mindee/v2/cli/OcrCommand.java +++ b/src/main/java/com/mindee/v2/cli/OcrCommand.java @@ -22,7 +22,7 @@ protected CommonResponse executeRequest( .enqueueAndGetResult( OcrResponse.class, inputSource, - OcrParameters.builder(modelId).alias(alias).build() + OcrParameters.builder(modelId).alias(alias).webhookIds(getWebhookIds()).build() ); } diff --git a/src/main/java/com/mindee/v2/cli/SplitCommand.java b/src/main/java/com/mindee/v2/cli/SplitCommand.java index a8185f962..aca7106b0 100644 --- a/src/main/java/com/mindee/v2/cli/SplitCommand.java +++ b/src/main/java/com/mindee/v2/cli/SplitCommand.java @@ -22,7 +22,7 @@ protected CommonResponse executeRequest( .enqueueAndGetResult( SplitResponse.class, inputSource, - SplitParameters.builder(modelId).alias(alias).build() + SplitParameters.builder(modelId).alias(alias).webhookIds(getWebhookIds()).build() ); } diff --git a/tests/test_v2_cli.sh b/tests/test_v2_cli.sh index fed84061e..545a2a473 100755 --- a/tests/test_v2_cli.sh +++ b/tests/test_v2_cli.sh @@ -22,7 +22,7 @@ run_test() { model_type="$2" echo "--- Test $model_type ID: $model_id" - SUMMARY_OUTPUT=$(./cli.sh "$model_type" -m "$model_id" "$TEST_FILE") + SUMMARY_OUTPUT=$(./cli.sh "$model_type" -m "$model_id" -a "java-sdk-cli-${model_type}" "$TEST_FILE") echo "$SUMMARY_OUTPUT" echo "" echo ""