make training silent and disable autosave

Signed-off-by: Scar26 <mmatty26@gmail.com>
juice-shop · Jul 1, 2020 · 04d0d51 · 04d0d51
1 parent ed036da
commit 04d0d51
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 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
@@ -15,7 +15,8 @@ class Bot {
     this.factory = new VM({
       sandbox: {
         Nlp: NlpManager,
-        training: this.training
+        training: this.training,
+        fs: null,
       }
     })
     this.factory.run(ctx)
@@ -30,6 +31,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})`))
   }