fix: Error retrieving entity from whitelist

axa-group · Jul 15, 2019 · 2654b65 · 2654b65
1 parent 83bd4a9
commit 2654b65
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/ner/ner-manager.js b/lib/ner/ner-manager.js
@@ -517,7 +517,11 @@ class NerManager {
     if (witelistMap) {
       edges = edges.map(item => {
         const edge = item;
-        edge.entity = witelistMap[item.entity].shift() || item.entity;
+        if (witelistMap[item.entity]) {
+          edge.entity = witelistMap[item.entity].shift() || item.entity;
+        } else {
+          edge.entity = item.entity;
+        }
         return edge;
       });
     }