juicy-chat-bot
Version:
A light-weight and totally "secure" library to easily deploy simple chat bots
39 lines • 1.18 kB
JavaScript
;
/*
* Copyright (c) 2020-2023 Bjoern Kimminich & the OWASP Juice Shop contributors.
* SPDX-License-Identifier: MIT
*/
var trainingSet = training.data;
const model = new Nlp({ languages: ['en'], nlu: { log: false }, autoSave: false, autoLoad: false, modelFileName: '' });
var users = {
idmap: {},
addUser: function (token, name) {
this.idmap[token] = name;
},
get: function (token) {
return this.idmap[token];
}
};
function train() {
trainingSet.data.forEach((query) => {
query.utterances.forEach((utterance) => {
model.addDocument(trainingSet.lang, utterance, query.intent);
});
query.answers.forEach((answer) => {
model.addAnswer(trainingSet.lang, query.intent, answer);
});
});
return model.train().then(() => { training.state = true; });
}
function processQuery(query, token) {
if (users.get(token)) {
return model.process(trainingSet.lang, query);
}
else {
return { action: 'unrecognized', body: 'user does not exist' };
}
}
function currentUser(token) {
return users.get(token);
}
//# sourceMappingURL=factory.js.map