Merge pull request #6 from bkimminich/nlp-support

Make training silent and disable model autosave
juice-shop · Jul 4, 2020 · c1b5dd6 · c1b5dd6
2 parents 2d7e58f + 5676ec4
commit c1b5dd6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/factory.js b/factory.js
@@ -1,8 +1,7 @@
 /* eslint-disable no-unused-vars */
 /* global Nlp, training */
 var trainingSet = training.data
-const model = new Nlp({ languages: ['en'] })
-
+const model = new Nlp({ languages: ['en'], nlu: { log: false }, autoSave: false, autoLoad: false, modelFileName: '' })
 var users = {
   idmap: {},
 

diff --git a/index.js b/index.js
@@ -30,6 +30,10 @@ class Bot {
     this.factory.run(`users.addUser("${token}", "${name}")`)
   }
 
+  getUser (token) {
+    return this.factory.run(`users.get("${token}")`)
+  }
+
   render (statement, token) {
     return statement.replace(/<bot-name>/g, this.name).replace(/<customer-name>/g, this.factory.run(`currentUser(${token})`))
   }