will it blend?

Author: ianardee

src/main/java/com/mindee/CommandLineInterface.java

@@ -7,6 +7,7 @@
 import com.mindee.v2.cli.ExtractionCommand;
 import com.mindee.v2.cli.OcrCommand;
 import com.mindee.v2.cli.SearchModelsCommand;
+import com.mindee.v2.cli.SearchRagDocumentsCommand;
 import com.mindee.v2.cli.SplitCommand;
 import java.lang.reflect.Method;
 import picocli.CommandLine;
@@ -42,6 +43,7 @@ public static void main(String[] args) {
 
     // V2 commands at root
     rootCmd.addSubcommand("search-models", new CommandLine(new SearchModelsCommand()));
+    rootCmd.addSubcommand("search-rag-docs", new CommandLine(new SearchRagDocumentsCommand()));
     rootCmd.addSubcommand("classification", new CommandLine(new ClassificationCommand()));
     rootCmd.addSubcommand("crop", new CommandLine(new CropCommand()));
     rootCmd.addSubcommand("extraction", new CommandLine(new ExtractionCommand()));

src/main/java/com/mindee/v2/MindeeClient.java

@@ -14,6 +14,7 @@
 import com.mindee.v2.parsing.search.BaseSearchResponse;
 import com.mindee.v2.parsing.search.SearchResponse;
 import com.mindee.v2.product.extraction.ExtractionResponse;
+import com.mindee.v2.search.models.ModelSearchParameters;
 import java.io.IOException;
 import java.util.concurrent.CancellationException;
 
@@ -213,7 +214,7 @@ public <TSearchResponse extends BaseSearchResponse> TSearchResponse search(
    */
   @Deprecated
   public SearchResponse searchModels() {
-    return searchModels(null, null);
+    return search(SearchResponse.class, null);
   }
 
   /**
@@ -225,7 +226,7 @@ public SearchResponse searchModels() {
    */
   @Deprecated
   public SearchResponse searchModels(String modelName) {
-    return searchModels(modelName, null);
+    return search(SearchResponse.class, ModelSearchParameters.builder().name(modelName).build());
   }
 
   /**
@@ -238,7 +239,10 @@ public SearchResponse searchModels(String modelName) {
    */
   @Deprecated
   public SearchResponse searchModels(String modelName, String modelType) {
-    return mindeeApi.reqGetSearchModels(modelName, modelType);
+    return search(
+      SearchResponse.class,
+      ModelSearchParameters.builder().name(modelName).modelType(modelType).build()
+    );
   }
 
   /**

src/main/java/com/mindee/v2/cli/BaseCommand.java

@@ -0,0 +1,69 @@
+package com.mindee.v2.cli;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.SerializationFeature;
+import com.mindee.v2.parsing.CommonResponse;
+import java.util.concurrent.Callable;
+import picocli.CommandLine;
+
+/**
+ * Abstract base class for V2 inference CLI commands.
+ * Handles common options (path, model-id, api-key, alias, output) and output formatting.
+ */
+public abstract class BaseCommand implements Callable<Integer> {
+  @CommandLine.Option(names = { "-k", "--api-key" }, description = "Mindee V2 API key.")
+  protected String apiKey;
+
+  /** Output format for the command. */
+  public enum OutputType {
+    summary,
+    full,
+    raw
+  }
+
+  @CommandLine.Option(
+      names = { "-o", "--output" },
+      description = "Specify how to output the data.\n"
+        + "- summary: a basic summary (default)\n"
+        + "- full: detail extraction results, including options\n"
+        + "- raw: full JSON object",
+      defaultValue = "summary"
+  )
+  protected OutputType output;
+
+  /**
+   * Returns the summary string for the given response.
+   * Override in each command.
+   *
+   * @param response the response
+   * @return the summary string
+   */
+  protected abstract String getSummary(CommonResponse response);
+
+  /**
+   * Returns the full string for the given response.
+   *
+   * @param response the product response
+   * @return the full output string
+   */
+  protected abstract String getFullOutput(CommonResponse response);
+
+  /**
+   * Prints the output to the console, taking into account the output type.
+   */
+  protected void printOutput(CommonResponse response) throws Exception {
+    switch (output) {
+      case full:
+        System.out.println(getFullOutput(response));
+        break;
+      case raw:
+        var mapper = new ObjectMapper().enable(SerializationFeature.INDENT_OUTPUT);
+        var jsonNode = mapper.readTree(response.getRawResponse());
+        System.out.println(mapper.writeValueAsString(jsonNode));
+        break;
+      default:
+        System.out.println(getSummary(response));
+        break;
+    }
+  }
+}

src/main/java/com/mindee/v2/cli/BaseInferenceCommand.java

@@ -1,31 +1,25 @@
 package com.mindee.v2.cli;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.fasterxml.jackson.databind.SerializationFeature;
 import com.mindee.input.LocalInputSource;
 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;
 
 /**
  * Abstract base class for V2 inference CLI commands.
  * Handles common options (path, model-id, api-key, alias, output) and output formatting.
  */
-public abstract class BaseInferenceCommand implements Callable<Integer> {
+public abstract class BaseInferenceCommand extends BaseCommand {
 
   @Parameters(index = "0", paramLabel = "<path>", description = "The path of the file to parse")
   protected File file;
 
   @Option(names = { "-m", "--model-id" }, description = "ID of the model to use", required = true)
   protected String modelId;
 
-  @Option(names = { "-k", "--api-key" }, description = "Mindee V2 API key.")
-  protected String apiKey;
-
   @Option(names = { "-a", "--alias" }, description = "Alias for the file")
   protected String alias;
 
@@ -35,23 +29,6 @@ public abstract class BaseInferenceCommand implements Callable<Integer> {
   )
   private List<String> webhookIds;
 
-  /** Output format for the command. */
-  public enum OutputType {
-    summary,
-    full,
-    raw
-  }
-
-  @Option(
-      names = { "-o", "--output" },
-      description = "Specify how to output the data.\n"
-        + "- summary: a basic summary (default)\n"
-        + "- full: detail extraction results, including options\n"
-        + "- raw: full JSON object",
-      defaultValue = "summary"
-  )
-  protected OutputType output;
-
   /**
    * @return The properly formatted webhook IDs.
    */
@@ -72,48 +49,12 @@ protected abstract CommonResponse executeRequest(
       LocalInputSource inputSource
   ) throws Exception;
 
-  /**
-   * Returns the summary (result-only) string for the given response.
-   * Override in each product command.
-   *
-   * @param response the product response
-   * @return the summary string
-   */
-  protected abstract String getSummary(CommonResponse response);
-
-  /**
-   * Returns the full (inference + options + result) string for the given response.
-   * Defaults to the same as {@link #getSummary}; override for richer output.
-   *
-   * @param response the product response
-   * @return the full output string
-   */
-  protected String getFullOutput(CommonResponse response) {
-    return getSummary(response);
-  }
-
   @Override
   public Integer call() throws Exception {
-    var client = new MindeeClient(apiKey != null ? apiKey : "");
+    var client = new MindeeClient(String.valueOf(apiKey));
     var inputSource = new LocalInputSource(file);
     var response = executeRequest(client, inputSource);
     printOutput(response);
     return 0;
   }
-
-  private void printOutput(CommonResponse response) throws Exception {
-    switch (output) {
-      case full:
-        System.out.println(getFullOutput(response));
-        break;
-      case raw:
-        var mapper = new ObjectMapper().enable(SerializationFeature.INDENT_OUTPUT);
-        var jsonNode = mapper.readTree(response.getRawResponse());
-        System.out.println(mapper.writeValueAsString(jsonNode));
-        break;
-      default:
-        System.out.println(getSummary(response));
-        break;
-    }
-  }
 }

src/main/java/com/mindee/v2/cli/SearchModelsCommand.java

@@ -1,9 +1,9 @@
 package com.mindee.v2.cli;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.fasterxml.jackson.databind.SerializationFeature;
 import com.mindee.v2.MindeeClient;
-import java.util.concurrent.Callable;
+import com.mindee.v2.parsing.CommonResponse;
+import com.mindee.v2.search.models.ModelSearchParameters;
+import com.mindee.v2.search.models.ModelSearchResponse;
 import picocli.CommandLine.Command;
 import picocli.CommandLine.Option;
 
@@ -15,41 +15,47 @@
     description = "Search available models.",
     mixinStandardHelpOptions = true
 )
-public class SearchModelsCommand implements Callable<Integer> {
-
-  @Option(names = { "-k", "--api-key" }, description = "Mindee V2 API key.")
-  private String apiKey;
+public class SearchModelsCommand extends BaseCommand {
 
   @Option(
       names = { "-n", "--name" },
       description = "Filter by model name partial match (case insensitive)."
   )
   private String name;
 
-  @Option(
-      names = { "-m", "--model-type" },
-      description = "Filter by exact model type (case sensitive)."
-  )
-  private String modelType;
+  public enum ModelType {
+    extraction,
+    crop,
+    classification,
+    ocr,
+    split
+  }
 
   @Option(
-      names = { "-r", "--raw-json" },
-      description = "Whether to output the raw JSON response.",
-      defaultValue = "false"
+      names = { "-m", "--model-type" },
+      description = "Filter by exact model type.%nAvailable options: ${COMPLETION-CANDIDATES}"
   )
-  private boolean rawJson;
+  private ModelType modelType;
 
   @Override
   public Integer call() throws Exception {
-    var client = new MindeeClient(apiKey != null ? apiKey : "");
-    var response = client.searchModels(name, modelType);
-    if (rawJson) {
-      var mapper = new ObjectMapper().enable(SerializationFeature.INDENT_OUTPUT);
-      var jsonNode = mapper.readTree(response.getRawResponse());
-      System.out.println(mapper.writeValueAsString(jsonNode));
-    } else {
-      System.out.println(response);
-    }
+    var client = new MindeeClient(String.valueOf(apiKey));
+    var response = client
+      .search(
+        ModelSearchResponse.class,
+        ModelSearchParameters.builder().name(name).modelType(String.valueOf(modelType)).build()
+      );
+    printOutput(response);
     return 0;
   }
+
+  @Override
+  protected String getSummary(CommonResponse response) {
+    return ((ModelSearchResponse) response).getModels().toString();
+  }
+
+  @Override
+  protected String getFullOutput(CommonResponse response) {
+    return response.toString();
+  }
 }

src/main/java/com/mindee/v2/cli/SearchRagDocumentsCommand.java

@@ -0,0 +1,50 @@
+package com.mindee.v2.cli;
+
+import com.mindee.v2.MindeeClient;
+import com.mindee.v2.parsing.CommonResponse;
+import com.mindee.v2.search.ragdocuments.RagDocumentSearchParameters;
+import com.mindee.v2.search.ragdocuments.RagDocumentSearchResponse;
+import picocli.CommandLine.Command;
+import picocli.CommandLine.Option;
+
+/**
+ * CLI command for searching available V2 RAG documents.
+ */
+@Command(
+    name = "search-rag-docs",
+    description = "Search available RAG documents for a given model.",
+    mixinStandardHelpOptions = true
+)
+public class SearchRagDocumentsCommand extends BaseCommand {
+
+  @Option(names = { "-m", "--model-id" }, description = "Filter by model ID.", required = true)
+  private String modelId;
+
+  @Option(
+      names = { "-f", "--filename" },
+      description = "Filter by model name partial match (case insensitive)."
+  )
+  private String filename;
+
+  @Override
+  protected String getSummary(CommonResponse response) {
+    return ((RagDocumentSearchResponse) response).getRagDocuments().toString();
+  }
+
+  @Override
+  protected String getFullOutput(CommonResponse response) {
+    return response.toString();
+  }
+
+  @Override
+  public Integer call() throws Exception {
+    var client = new MindeeClient(String.valueOf(apiKey));
+    var response = client
+      .search(
+        RagDocumentSearchResponse.class,
+        RagDocumentSearchParameters.builder(modelId).filename(filename).build()
+      );
+    printOutput(response);
+    return 0;
+  }
+}