Apply model version in getOutputInfo

Author: rok-povsic

src/Model.js

@@ -19,6 +19,7 @@ let {
   PREDICT_PATH,
   VERSION_PREDICT_PATH,
   MODEL_INPUTS_PATH,
+  MODEL_VERSION_OUTPUT_PATH,
   MODEL_OUTPUT_PATH,
   MODEL_VERSION_INPUTS_PATH,
   MODEL_VERSION_METRICS_PATH
@@ -277,7 +278,9 @@ class Model {
    * @return {Promise(Model, error)} A Promise that is fulfilled with a Model instance or rejected with an error
    */
   getOutputInfo() {
-    let url = `${this._config.basePath}${replaceVars(MODEL_OUTPUT_PATH, [this.id])}`;
+    let url = `${this._config.basePath}${this.versionId ?
+      replaceVars(MODEL_VERSION_OUTPUT_PATH, [this.id, this.versionId]) :
+      replaceVars(MODEL_OUTPUT_PATH, [this.id])}`;
     return wrapToken(this._config, (headers) => {
       return new Promise((resolve, reject) => {
         axios.get(url, {headers}).then((response) => {

src/constants.js

@@ -14,6 +14,7 @@ module.exports = {
     MODEL_VERSION_PATH: '/models/$0/versions/$1',
     MODEL_PATCH_PATH: '/models/$0/output_info/data/concepts',
     MODEL_OUTPUT_PATH: '/models/$0/output_info',
+    MODEL_VERSION_OUTPUT_PATH: '/models/$0/versions/$1/output_info',
     MODEL_SEARCH_PATH: '/models/searches',
     MODEL_FEEDBACK_PATH: '/models/$0/feedback',
     MODEL_VERSION_FEEDBACK_PATH: '/models/$0/versions/$1/feedback',

tests/integration/models-int-tests.js

@@ -607,4 +607,23 @@ describe('Integration Tests - Models', () => {
       })
       .catch(errorHandler.bind(done));
   });
+
+  it('Gets output info', done => {
+    app.models.getOutputInfo(Clarifai.GENERAL_MODEL)
+      .then(model => {
+        expect(model.id).toEqual(Clarifai.GENERAL_MODEL);
+        done();
+      })
+      .catch(errorHandler.bind(done));
+  });
+
+  it('Gets output info when specifying model version', done => {
+    let olderGeneralModelVersionID = 'aa9ca48295b37401f8af92ad1af0d91d';
+    app.models.getOutputInfo({id: Clarifai.GENERAL_MODEL, version: olderGeneralModelVersionID})
+      .then(model => {
+        expect(model.modelVersion.id).toEqual(olderGeneralModelVersionID);
+        done();
+      })
+      .catch(errorHandler.bind(done));
+  });
 });