INSIGHT-1098: attaching corpus correctly

nuxeo · May 27, 2021 · b35b9c5 · b35b9c5
1 parent 7907244
commit b35b9c5
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/ExportCaller.java b/insight-rest-client/src/main/java/org/nuxeo/ai/sdk/rest/api/ExportCaller.java
@@ -50,8 +50,6 @@
  */
 public class ExportCaller implements Resource {
 
-    public static final String CORPUS_PARAM = "corpus";
-
     private final Logger log = LogManager.getLogger(ExportCaller.class);
 
     private final InsightClient client;
@@ -82,7 +80,7 @@ public <T> T call(Map<String, Serializable> parameters, Serializable payload)
         case BIND:
             return (T) handleBind(parameters);
         case ATTACH:
-            return (T) handleAttach(parameters, (AICorpus) payload);
+            return (T) handleAttach((String) parameters.get(CORPORA_ID_PARAM), (AICorpus) payload);
         case DONE:
             return (T) handleDone(parameters);
         default:
@@ -134,15 +132,14 @@ private Boolean handleBind(Map<String, Serializable> parameters) {
         });
     }
 
-    private String handleAttach(Map<String, Serializable> parameters, AICorpus payload) throws IOException {
+    private String handleAttach(String corporaId, AICorpus payload) throws IOException {
         String jsonString;
         try (StringWriter writer = new StringWriter()) {
-            MAPPER.writeValue(writer, parameters.get(CORPUS_PARAM));
+            MAPPER.writeValue(writer, payload);
             jsonString = writer.toString();
         }
 
         log.info("Creating dataset document");
-        String corporaId = (String) parameters.get(CORPORA_ID_PARAM);
         Objects.requireNonNull(corporaId, "Attach API requires corpora ID");
         JsonNode node = client.post(API.Export.ATTACH.toPath(client.getProjectId(), corporaId), jsonString, (resp) -> {
             if (!resp.isSuccessful()) {