REVAI-3919:Split NlpModel to have separate enums for Translation and Summarization (#63)

https://revinc.atlassian.net/browse/REVAI-3919

Author: kirillatrev

README.md

@@ -207,7 +207,7 @@ int channelId = 1;
 InputStream inputStream = apiClient.getCaptions(revAiJob.getJobId(), RevAiCaptionType.VTT, channelId);
 
 // or if you requested transcript translation(s)
-InputStream inputStream = apiClient.getTranslatedCaptions(revAiJob.getJobId(), "es", RevAiCaptionType.VTT, channelId);
+InputStream inputStream = apiClient.getTranslatedCaptions(revAiJob.getJobId(), "es", RevAiCaptionType.VTT);
 ```
 
 ### Getting transcript summary

examples/AsyncSummarizeMediaUrl.java

@@ -28,7 +28,7 @@ public static void main(String[] args) {
     revAiJobOptions.setSourceConfig(mediaUrl, null);
     revAiJobOptions.setDeleteAfterSeconds(2592000); // 30 days in seconds
     revAiJobOptions.setLanguage("en");
-    revAiJobOptions.setSummarizationOptions(new SummarizationOptions().setModel(NlpModel.STANDARD));
+    revAiJobOptions.setSummarizationOptions(new SummarizationOptions().setModel(SummarizationModel.STANDARD));
 
     RevAiJob submittedJob;
 

examples/AsyncTranslateMediaUrl.java

@@ -30,7 +30,7 @@ public static void main(String[] args) {
     revAiJobOptions.setLanguage("en");
     revAiJobOptions.setTranslationOptions(new TranslationOptions(Arrays.asList(
             new TranslationLanguageOptions("es")
-                    .setModel(NlpModel.PREMIUM),
+                    .setModel(TranslationModel.PREMIUM),
             new TranslationLanguageOptions("de"))
     ));
 

pom.xml

@@ -5,7 +5,7 @@
     <!-- groupId, artifactId, these namespaces should not be changed -->
     <groupId>ai.rev</groupId>
     <artifactId>revai-java-sdk</artifactId>
-    <version>2.4.0</version>
+    <version>2.4.1</version>
     <name>Rev AI SDK for Java</name>
     <description>Java SDK for Rev AI API</description>
     <url>https://docs.rev.ai/</url>

src/main/java/ai/rev/speechtotext/ApiClient.java

@@ -438,23 +438,18 @@ public InputStream getCaptions(String id, RevAiCaptionType captionType, Integer
    * @param id The ID of the job to return captions for.
    * @param language requested translation language.
    * @param captionType An enumeration of the desired caption type. Default is SRT.
-   * @param channelId Identifies the audio channel of the file to output captions for. Default is
-   *     null.
    * @return InputStream A stream of bytes that represents the caption output.
    * @throws IOException If the response has a status code > 399.
    * @throws IllegalArgumentException If the job ID provided is null.
    * @see <a
    *     href="https://docs.rev.ai/api/asynchronous/reference/#operation/GetCaptions">https://docs.rev.ai/api/asynchronous/reference/#operation/GetCaptions</a>
    */
-  public InputStream getTranslatedCaptions(String id, String language, RevAiCaptionType captionType, Integer channelId)
+  public InputStream getTranslatedCaptions(String id, String language, RevAiCaptionType captionType)
           throws IOException {
     if (id == null) {
       throw new IllegalArgumentException("Job ID must be provided");
     }
     Map<String, String> query = new HashMap<>();
-    if (channelId != null) {
-      query.put("speaker_channel", channelId.toString());
-    }
     if (captionType == null) {
       captionType = RevAiCaptionType.SRT;
     }

src/main/java/ai/rev/speechtotext/models/asynchronous/Summarization.java

@@ -1,6 +1,5 @@
 package ai.rev.speechtotext.models.asynchronous;
 
-import ai.rev.speechtotext.models.NlpModel;
 import com.google.gson.annotations.SerializedName;
 
 /**
@@ -15,10 +14,10 @@ public class Summarization {
   /**
    * Summarization model.
    *
-   * @see NlpModel
+   * @see TranslationModel
    */
   @SerializedName("model")
-  private NlpModel model;
+  private TranslationModel model;
 
   /** Formatting options. Default is Paragraph. */
   @SerializedName("type")
@@ -56,9 +55,9 @@ public String getPrompt() {
    * Returns backend model used for the summarization job.
    *
    * @return Backend model used for the summarization job.
-   * @see NlpModel
+   * @see TranslationModel
    */
-  public NlpModel getModel() {
+  public TranslationModel getModel() {
     return model;
   }
 

src/main/java/ai/rev/speechtotext/models/asynchronous/SummarizationModel.java

@@ -0,0 +1,30 @@
+package ai.rev.speechtotext.models.asynchronous;
+
+import com.google.gson.annotations.SerializedName;
+
+/** Supported model types for summarization. */
+public enum SummarizationModel {
+
+    @SerializedName("standard")
+    STANDARD("standard"),
+    @SerializedName("premium")
+    PREMIUM("premium");
+
+    private final String model;
+
+    SummarizationModel(String model) {
+        this.model = model;
+    }
+
+    /**
+     * Returns the String value of the enumeration.
+     *
+     * @return The String value of the enumeration.
+     */
+    public String getModel() { return model; }
+
+    @Override
+    public String toString() {
+        return "{" + "model='" + model + '\'' + '}';
+    }
+}

src/main/java/ai/rev/speechtotext/models/asynchronous/SummarizationOptions.java

@@ -1,6 +1,5 @@
 package ai.rev.speechtotext.models.asynchronous;
 
-import ai.rev.speechtotext.models.NlpModel;
 import com.google.gson.annotations.SerializedName;
 
 /**
@@ -18,7 +17,7 @@ public class SummarizationOptions {
 
   /** Standard or Premium AI backend. */
   @SerializedName("model")
-  private NlpModel model;
+  private SummarizationModel model;
 
   /** Formatting options. Default is Paragraph. */
   @SerializedName("type")
@@ -47,19 +46,19 @@ public SummarizationOptions setPrompt(String prompt) {
    * Returns backend model used for the summarization job.
    *
    * @return Backend model used for the summarization job.
-   * @see NlpModel
+   * @see TranslationModel
    */
-  public NlpModel getModel() {
+  public SummarizationModel getModel() {
     return model;
   }
 
   /**
    * Sets backend model used for the summarization job.
    *
    * @param model Backend model used for the summarization job.
-   * @see NlpModel
+   * @see SummarizationModel
    */
-  public SummarizationOptions setModel(NlpModel model) {
+  public SummarizationOptions setModel(SummarizationModel model) {
     this.model = model;
     return this;
   }

src/main/java/ai/rev/speechtotext/models/asynchronous/TranslationLanguage.java

@@ -1,6 +1,5 @@
 package ai.rev.speechtotext.models.asynchronous;
 
-import ai.rev.speechtotext.models.NlpModel;
 import com.google.gson.annotations.SerializedName;
 
 /**
@@ -11,7 +10,7 @@
 public class TranslationLanguage {
   /** Standard or Premium AI backend. */
   @SerializedName("model")
-  private NlpModel model;
+  private TranslationModel model;
 
   @SerializedName("language")
   private String language;
@@ -26,9 +25,9 @@ public class TranslationLanguage {
    * Returns backend model used for the summarization job.
    *
    * @return Backend model used for the summarization job.
-   * @see NlpModel
+   * @see TranslationModel
    */
-  public NlpModel getModel() {
+  public TranslationModel getModel() {
     return model;
   }
 

src/main/java/ai/rev/speechtotext/models/asynchronous/TranslationLanguageOptions.java

@@ -1,6 +1,5 @@
 package ai.rev.speechtotext.models.asynchronous;
 
-import ai.rev.speechtotext.models.NlpModel;
 import com.google.gson.annotations.SerializedName;
 
 /**
@@ -13,7 +12,7 @@
 public class TranslationLanguageOptions {
   /** Standard or Premium AI backend. */
   @SerializedName("model")
-  private NlpModel model;
+  private TranslationModel model;
 
   @SerializedName("language")
   private final String language;
@@ -22,19 +21,19 @@ public class TranslationLanguageOptions {
    * Returns backend model used for the summarization job.
    *
    * @return Backend model used for the summarization job.
-   * @see NlpModel
+   * @see TranslationModel
    */
-  public NlpModel getModel() {
+  public TranslationModel getModel() {
     return model;
   }
 
   /**
    * Sets backend model to use for the summarization job.
    *
    * @param model Backend model to use for the summarization job
-   * @see NlpModel
+   * @see TranslationModel
    */
-  public TranslationLanguageOptions setModel(NlpModel model) {
+  public TranslationLanguageOptions setModel(TranslationModel model) {
     this.model = model;
     return this;
   }