leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2025-01-02 05:31:38 +03:00
Code Issues Projects Releases Wiki Activity

refactor(server): slot filling

Browse Source
This commit is contained in:
louistiti 2023-03-27 22:23:10 +08:00
parent 26ff4520c8
commit 8282c37a5a
2 changed files with 19 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/src/core/nlp/conversation.ts
View File

@ -29,7 +29,7 @@ interface ConversationContext {
type ConversationPreviousContext = Record<string, ConversationContext> | null
const MAX_CONTEXT_HISTORY = 5
const DEFAULT_ACTIVE_CONTEXT = {
export const DEFAULT_ACTIVE_CONTEXT = {
name: null,
domain: '',
intent: '',

33
server/src/core/nlp/nlu/slot-filling.ts
View File

@ -1,11 +1,11 @@
import fs from 'node:fs'
import { join } from 'node:path'
import type { NLPUtterance } from '@/core/nlp/types'
import type { BrainProcessResult } from '@/core/brain/types'
import { BRAIN, MODEL_LOADER, NER, NLU, SOCKET_SERVER } from '@/core'
import { DEFAULT_NLU_RESULT } from '@/core/nlp/nlu/nlu'
import { SkillDomainHelper } from "@/helpers/skill-domain-helper";
import { SkillDomainHelper } from '@/helpers/skill-domain-helper'
import { DEFAULT_ACTIVE_CONTEXT } from '@/core/nlp/conversation'
export class SlotFilling {
/**
@ -25,16 +25,17 @@ export class SlotFilling {
if (processedData && Object.keys(processedData).length > 0) {
// Set new context with the next action if there is one
if (processedData.action.next_action) {
if (processedData.action?.next_action) {
NLU.conversation.activeContext = {
...DEFAULT_ACTIVE_CONTEXT,
lang: BRAIN.lang,
slots: processedData.slots,
isInActionLoop: !!processedData.nextAction.loop,
originalUtterance: processedData.utterance,
skillConfigPath: processedData.skillConfigPath,
slots: processedData.slots || {},
isInActionLoop: !!processedData.nextAction?.loop,
originalUtterance: processedData.utterance ?? null,
skillConfigPath: processedData.skillConfigPath || '',
actionName: processedData.action.next_action,
domain: processedData.classification.domain,
intent: `${processedData.classification.skill}.${processedData.action.next_action}`,
domain: processedData.classification?.domain || '',
intent: `${processedData.classification?.skill}.${processedData.action.next_action}`,
entities: []
}
}
@ -53,7 +54,7 @@ export class SlotFilling {
}
const { domain, intent } = NLU.conversation.activeContext
const [skillName, actionName] = intent.split('.')
const [skillName, actionName] = intent.split('.') as [string, string]
const skillConfigPath = join(
process.cwd(),
'skills',
@ -69,7 +70,8 @@ export class SlotFilling {
classification: {
domain,
skill: skillName,
action: actionName
action: actionName,
confidence: 1
}
}
@ -83,7 +85,7 @@ export class SlotFilling {
let notFilledSlot = NLU.conversation.getNotFilledSlot()
if (notFilledSlot && entities.length > 0) {
const hasMatch = entities.some(
({ entity }) => entity === notFilledSlot.expectedEntity
({ entity }) => entity === notFilledSlot?.expectedEntity
)
if (hasMatch) {
@ -108,7 +110,7 @@ export class SlotFilling {
slots: NLU.conversation.activeContext.nextAction
? NLU.conversation.activeContext.slots
: {},
utterance: NLU.conversation.activeContext.originalUtterance,
utterance: NLU.conversation.activeContext.originalUtterance ?? '',
skillConfigPath,
classification: {
domain,
@ -139,6 +141,7 @@ export class SlotFilling {
if (hasMandatorySlots) {
NLU.conversation.activeContext = {
...DEFAULT_ACTIVE_CONTEXT,
lang: BRAIN.lang,
slots,
isInActionLoop: false,
@ -156,9 +159,9 @@ export class SlotFilling {
const { actions } = SkillDomainHelper.getSkillConfig(NLU.nluResult.skillConfigPath, BRAIN.lang)
const [currentSlot] = actions[
NLU.nluResult.classification.action
].slots.filter(({ name }) => name === notFilledSlot.name)
]?.slots?.filter(({ name }) => name === notFilledSlot.name) ?? []
SOCKET_SERVER.socket.emit('suggest', currentSlot.suggestions)
SOCKET_SERVER.socket.emit('suggest', currentSlot?.suggestions)
BRAIN.talk(notFilledSlot.pickedQuestion)
SOCKET_SERVER.socket.emit('is-typing', false)