Add user id support for model predict (#94)

Author: eddiezane

spec/models-spec.js

@@ -165,6 +165,26 @@ describe('Models', () => {
       .catch(errorHandler.bind(done));
   });
 
+  it('Call predict with a declared input id', done => {
+    app.models.predict(Clarifai.GENERAL_MODEL, [
+      {
+        id: 'test-id-1',
+        url: sampleImages[7]
+      },
+      {
+        id: 'test-id-2',
+        url: sampleImages[8]
+      }
+    ])
+      .then(response => {
+        expect(response.outputs).toBeDefined();
+        expect(response.outputs[0].input.id).toBe('test-id-1');
+        expect(response.outputs[1].input.id).toBe('test-id-2');
+        done();
+      })
+      .catch(errorHandler.bind(done));
+  });
+
   it('Call predict on video inputs', done => {
     app.models.predict(Clarifai.GENERAL_MODEL, sampleVideos[0], {video: true})
       .then(response => {

src/utils.js

@@ -84,9 +84,9 @@ module.exports = {
     return formatted;
   },
   formatMediaPredict: (data, type = 'image') => {
-    let media = data;
+    let media;
     if (checkType(/String/, data)) {
-      if (URL_REGEX.test(media) === true) {
+      if (URL_REGEX.test(data) === true) {
         media = {
           url: data
         };
@@ -95,12 +95,28 @@ module.exports = {
           base64: data
         };
       }
+    } else {
+      media = Object.assign({}, data);
+    }
+
+    // Users can specify their own id to distinguish batch results
+    let id;
+    if (media.id) {
+      id = media.id;
+      delete media.id;
     }
-    return {
+
+    let object = {
       data: {
         [type]: media
       }
     };
+
+    if (id) {
+      object.id = id;
+    }
+
+    return object;
   },
   formatImagesSearch: (image) => {
     let imageQuery;