fix(Visual Recognition): Ensure file extensions are sent with names f…

…or createClassifier
watson-developer-cloud · Sep 18, 2019 · f524f80 · f524f80
1 parent a0cf9b3
commit f524f80
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/visual-recognition/src/main/java/com/ibm/watson/visual_recognition/v3/VisualRecognition.java b/visual-recognition/src/main/java/com/ibm/watson/visual_recognition/v3/VisualRecognition.java
@@ -255,13 +255,16 @@ public ServiceCall<Classifier> createClassifier(CreateClassifierOptions createCl
     for (Map.Entry<String, InputStream> entry : createClassifierOptions.positiveExamples().entrySet()) {
       String partName = String.format("%s_positive_examples", entry.getKey());
       RequestBody part = RequestUtils.inputStreamBody(entry.getValue(), "application/octet-stream");
-      multipartBuilder.addFormDataPart(partName, entry.getKey(), part);
+      multipartBuilder.addFormDataPart(partName, entry.getKey() + ".zip", part);
     }
     if (createClassifierOptions.negativeExamples() != null) {
       RequestBody negativeExamplesBody = RequestUtils.inputStreamBody(createClassifierOptions.negativeExamples(),
           "application/octet-stream");
-      multipartBuilder.addFormDataPart("negative_examples", createClassifierOptions.negativeExamplesFilename(),
-          negativeExamplesBody);
+      String negativeExamplesFilename = createClassifierOptions.negativeExamplesFilename();
+      if (!negativeExamplesFilename.contains(".")) {
+        negativeExamplesFilename += ".zip";
+      }
+      multipartBuilder.addFormDataPart("negative_examples", negativeExamplesFilename, negativeExamplesBody);
     }
     builder.body(multipartBuilder.build());
     ResponseConverter<Classifier> responseConverter = ResponseConverterUtils.getValue(