leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-12-18 14:21:32 +03:00
Code Issues Projects Releases Wiki Activity

fix(server): NER due to modules to skills refactoring

Browse Source
This commit is contained in:
louistiti 2022-02-14 16:48:03 +08:00
parent 65398d8231
commit e4d3904cee
No known key found for this signature in database
GPG Key ID: 7ECA3DD523793FE6
2 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/src/core/ner.js
View File

@ -35,10 +35,10 @@ class Ner {
const { classification } = obj
// Remove end-punctuation and add an end-whitespace
const utterance = `${string.removeEndPunctuation(obj.utterance)} `
const utteranceSamplesObj = JSON.parse(fs.readFileSync(utteranceSamplesFilePath, 'utf8'))
const { module, action } = classification
const { actions } = JSON.parse(fs.readFileSync(utteranceSamplesFilePath, 'utf8'))
const { action } = classification
const promises = []
const actionEntities = utteranceSamplesObj[module][action].entities || []
const actionEntities = actions[action].entities || []
/**
* Browse action entities

14
server/src/core/nlu.js
View File

@ -91,13 +91,13 @@ class Nlu {
const {
locale, domain, intent, score
} = result
const [moduleName, actionName] = intent.split('.')
const [skillName, actionName] = intent.split('.')
let obj = {
utterance,
entities: [],
classification: {
package: domain,
module: moduleName,
domain,
skill: skillName,
action: actionName,
confidence: score
}
@ -153,10 +153,11 @@ class Nlu {
try {
obj.entities = await this.ner.extractEntities(
this.brain.lang,
join(__dirname, '../../../packages', obj.classification.package, `data/expressions/${this.brain.lang}.json`),
join(process.cwd(), 'skills', obj.classification.domain, obj.classification.skill, `nlu/${this.brain.lang}.json`),
obj
)
} catch (e) /* istanbul ignore next */ {
console.error(e)
log[e.type](e.obj.message)
if (!opts.mute) {
@ -176,6 +177,7 @@ class Nlu {
nluProcessingTime: processingTime - data?.executionTime // In ms, NLU processing time only
})
} catch (e) /* istanbul ignore next */ {
console.error(e)
log[e.type](e.obj.message)
if (!opts.mute) {
@ -207,8 +209,8 @@ class Nlu {
if (JSON.stringify(tmpWords) === JSON.stringify(fallbacks[i].words)) {
obj.entities = []
obj.classification.package = fallbacks[i].package
obj.classification.module = fallbacks[i].module
obj.classification.domain = fallbacks[i].domain
obj.classification.skill = fallbacks[i].skill
obj.classification.action = fallbacks[i].action
obj.classification.confidence = 1