Add classifications to NLU node

Author: chughts

README.md

@@ -16,7 +16,8 @@ Node-RED Watson Nodes for IBM Cloud
 - Drop Visual Recognition nodes
 - Remove Username / Password fields from all nodes
 - Bump to latest version of services
-- New Discovery V2 node 
+- New Discovery V2 node
+- Add Classifications to Natural Language Understanding node
 
 
 ### Watson Nodes for Node-RED

services/natural_language_understanding/v1.html

@@ -51,6 +51,16 @@
                      id="node-input-limitcategories" />
             </div>
 
+            <div>
+              <input style="width: 30px; margin-left: 20px; margin-top: 0;"
+                     type="checkbox" id="node-input-classifications" />
+              <label style="width: auto;" for="node-input-classifications">Classifications</label>
+            </div>
+            <div>
+              <label style="width: auto;" for="node-input-classifications-model">Classification model</label>
+              <input type="text" id="node-input-classifications-model" />
+            </div>
+
             <div>
               <input style="width: 30px; margin-left: 20px; margin-top: 0;"
                      type="checkbox" id="node-input-concepts" />
@@ -200,6 +210,7 @@
     <p>The following features are available for analysis:</p>
     <ul>
         <li><b>Categories</b>, categorize content into a taxonomy.</li>
+        <li><b>Classifications</b>, text classifications for the content.</li>
         <li><b>Concepts</b>, recognize high-level concepts.</li>
         <li><b>Document Emotion</b>, generate emotion scores.</li>
         <li><b>Emotion Targets</b>, optionally target the emotions scores to a
@@ -241,6 +252,8 @@
       <code>msg.nlu_options.entity_model</code>.
       To override the model used for relations set
       <code>msg.nlu_options.relations_model</code>.
+      To override the model used for classifications set
+      <code>msg.nlu_options.classifications_model</code>.
     </p>
     <br>
     <p>Results from the API service will made available
@@ -257,6 +270,7 @@
   var nluV1 = new NaturalLanguageUnderstandingV1();
 
   nluV1.hideAll = function() {
+    $('#node-input-classifications-model').parent().hide();
     $('#node-input-maxconcepts').parent().hide();
     $('#node-input-limitcategories').parent().hide();
     $('#node-input-doc-emotion-target').parent().hide();
@@ -295,6 +309,9 @@
   }
 
   nluV1.UIListeners = function () {
+    nluV1.CreateIListener($('#node-input-classifications'),
+                               $('#node-input-classifications-model'));
+
     nluV1.CreateIListener($('#node-input-concepts'),
                                $('#node-input-maxconcepts'));
 
@@ -357,6 +374,8 @@
               'name': {value: ''},
               'categories': {value: false},
               'limitcategories': {value: '3'},
+              'classifications': {value: false},
+              'classifications-model': {value: '8'},
               'concepts': {value: false},
               'maxconcepts': {value: '8'},
               'doc-emotion': {value: false},

services/natural_language_understanding/v1.js

@@ -21,6 +21,7 @@ module.exports = function (RED) {
 
   const NLU_FEATURES = {
     'categories': 'categories',
+    'classifications': 'classifications',
     'concepts': 'concepts',
     'doc-emotion': 'emotion',
     'doc-sentiment': 'sentiment',
@@ -112,6 +113,18 @@ module.exports = function (RED) {
     }
   }
 
+
+
+  function processClassificationsOptions(msg, config, features) {
+    if (features.classifications) {
+      if (msg.nlu_options && msg.nlu_options.classifications_model) {
+        features.classifications.model = msg.nlu_options.classifications_model;
+      } else if (config['classifications-model']) {
+        features.classifications.model = config['classifications-model'] ;
+      }
+    }
+  }
+
   function processCategoriesOptions(config, features) {
     if (features.categories) {
       features.categories.limit =
@@ -195,14 +208,15 @@ module.exports = function (RED) {
   function checkFeatureOptions(msg, config, options) {
     if (options && options.features) {
       processConceptsOptions(config, options.features);
+      processClassificationsOptions(msg, config, options.features);
       processCategoriesOptions(config, options.features);
       processEmotionOptions(config, options.features);
       processSentimentOptions(config, options.features);
-      processEntitiesOptions(msg,config, options.features);
-      processRelationsOptions(msg,config, options.features);
+      processEntitiesOptions(msg, config, options.features);
+      processRelationsOptions(msg, config, options.features);
       processKeywordsOptions(config, options.features);
       processSemanticRolesOptions(config, options.features);
-      processSyntaxOptions(msg,config, options.features);
+      processSyntaxOptions(msg, config, options.features);
     }
     return Promise.resolve();
   }