✨ allow passing webhooks to the CLI (#342)

Author: ianardee

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<Integer> {
   @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<String> 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.
    *

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()
       );
   }
 

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()
       );
   }
 

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)

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()
       );
   }
 

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()
       );
   }
 

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 ""