leon-ai/leon
1
1
Fork 0
mirror of https://github.com/leon-ai/leon.git synced 2024-12-01 03:15:58 +03:00
Code Issues Projects Releases Wiki Activity

refactor(server): load models and init core instances only once

Browse Source
This commit is contained in:
louistiti 2023-02-20 21:56:22 +08:00
parent e182da0048
commit 4d7c49e1ee
1 changed files with 34 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
server/src/core/socket-server.ts
View File

@ -35,7 +35,7 @@ export default class SocketServer {
}
}
public init(): void {
public async init(): void {
const { httpServer, host } = HTTP_SERVER
const io = IS_DEVELOPMENT_ENV
? new SocketIOServer(httpServer, {
@ -43,6 +43,37 @@ export default class SocketServer {
})
: new SocketIOServer(httpServer)
let sttState = 'disabled'
let ttsState = 'disabled'
// TODO
// provider.brain.socket = socket
if (HAS_STT) {
sttState = 'enabled'
// TODO
// provider.brain.stt = new Stt(socket, STT_PROVIDER)
// provider.brain.stt.init(() => null)
await STT.init()
}
if (HAS_TTS) {
ttsState = 'enabled'
// TODO
// provider.brain.tts = new Tts(socket, TTS_PROVIDER)
// provider.brain.tts.init('en', () => null)
await TTS.init(
LangHelper.getShortCode(LANG)
)
}
LogHelper.title('Initialization')
LogHelper.success(`STT ${sttState}`)
LogHelper.success(`TTS ${ttsState}`)
await NLU.loadNLPModels()
io.on('connection', (socket) => {
LogHelper.title('Client')
LogHelper.success('Connected')
@ -50,9 +81,9 @@ export default class SocketServer {
this.socket = socket
// Init
this.socket.on('init', async (data) => {
this.socket.on('init', (data) => {
LogHelper.info(`Type: ${data}`)
LogHelper.info(`Socket id: ${this.socket.id}`)
LogHelper.info(`Socket ID: ${this.socket.id}`)
// TODO
// const provider = await addProvider(socket.id)
@ -75,37 +106,6 @@ export default class SocketServer {
this.socket.broadcast.emit('enable-record')
})
} else {
let sttState = 'disabled'
let ttsState = 'disabled'
// TODO
// provider.brain.socket = socket
if (HAS_STT) {
sttState = 'enabled'
// TODO
// provider.brain.stt = new Stt(socket, STT_PROVIDER)
// provider.brain.stt.init(() => null)
await STT.init()
}
if (HAS_TTS) {
ttsState = 'enabled'
// TODO
// provider.brain.tts = new Tts(socket, TTS_PROVIDER)
// provider.brain.tts.init('en', () => null)
await TTS.init(
LangHelper.getShortCode(LANG)
)
}
LogHelper.title('Initialization')
LogHelper.success(`STT ${sttState}`)
LogHelper.success(`TTS ${ttsState}`)
await NLU.loadNLPModels()
// Listen for new utterance
this.socket.on('utterance', async (data) => {
LogHelper.title('Socket')