wehub-resource-sync
83 lines
2.2 KiB
JavaScript
83 lines
2.2 KiB
JavaScript
import { spawn } from 'node:child_process'
|
|
|
|
import { LangHelper } from '@/helpers/lang-helper'
|
|
import { LogHelper } from '@/helpers/log-helper'
|
|
import { SystemHelper } from '@/helpers/system-helper'
|
|
import {
|
|
LANG as LEON_LANG,
|
|
NVIDIA_CUBLAS_PATH,
|
|
NVIDIA_CUDNN_PATH,
|
|
NVIDIA_CUSPARSE_PATH,
|
|
NVIDIA_CUSPARSE_FULL_PATH,
|
|
NVIDIA_NCCL_PATH,
|
|
NVIDIA_NVJITLINK_PATH,
|
|
NVIDIA_NVSHMEM_PATH,
|
|
NVIDIA_LIBS_PATH,
|
|
PYTHON_TCP_SERVER_ENTRY_PATH,
|
|
PYTHON_TCP_SERVER_RUNTIME_BIN_PATH,
|
|
PYTORCH_TORCH_PATH,
|
|
HAS_ASR,
|
|
HAS_TTS,
|
|
HAS_WAKE_WORD,
|
|
ASR_PROVIDER,
|
|
TTS_PROVIDER,
|
|
PYTHON_TCP_SERVER_HOST,
|
|
PYTHON_TCP_SERVER_PORT
|
|
} from '@/constants'
|
|
|
|
/**
|
|
* Run the Python TCP server directly from source with Leon's managed Python
|
|
* runtime. This mirrors the production/server boot path.
|
|
*/
|
|
;(async () => {
|
|
const langArg = process.argv[2] || LangHelper.getShortCode(LEON_LANG)
|
|
const args = [
|
|
PYTHON_TCP_SERVER_ENTRY_PATH,
|
|
langArg,
|
|
'--pytorch-path',
|
|
PYTORCH_TORCH_PATH,
|
|
'--nvidia-path',
|
|
NVIDIA_LIBS_PATH
|
|
]
|
|
const env = { ...process.env }
|
|
env['LEON_ASR'] = HAS_ASR ? 'true' : 'false'
|
|
env['LEON_ASR_PROVIDER'] = ASR_PROVIDER || ''
|
|
env['LEON_TTS'] = HAS_TTS ? 'true' : 'false'
|
|
env['LEON_TTS_PROVIDER'] = TTS_PROVIDER || ''
|
|
env['LEON_WAKE_WORD'] = HAS_WAKE_WORD ? 'true' : 'false'
|
|
env['LEON_PY_TCP_SERVER_HOST'] = PYTHON_TCP_SERVER_HOST
|
|
env['LEON_PY_TCP_SERVER_PORT'] = String(PYTHON_TCP_SERVER_PORT)
|
|
|
|
if (SystemHelper.isLinux()) {
|
|
const torchLibPath = `${PYTORCH_TORCH_PATH}/lib`
|
|
const nvidiaLibPaths = [
|
|
`${NVIDIA_CUBLAS_PATH}/lib`,
|
|
`${NVIDIA_CUDNN_PATH}/lib`,
|
|
`${NVIDIA_CUSPARSE_PATH}/lib`,
|
|
`${NVIDIA_CUSPARSE_FULL_PATH}/lib`,
|
|
`${NVIDIA_NCCL_PATH}/lib`,
|
|
`${NVIDIA_NVSHMEM_PATH}/lib`,
|
|
`${NVIDIA_NVJITLINK_PATH}/lib`
|
|
]
|
|
|
|
env['LD_LIBRARY_PATH'] = [torchLibPath, ...nvidiaLibPaths, env['LD_LIBRARY_PATH']]
|
|
.filter(Boolean)
|
|
.join(':')
|
|
}
|
|
|
|
const child = spawn(PYTHON_TCP_SERVER_RUNTIME_BIN_PATH, args, {
|
|
stdio: 'inherit',
|
|
env,
|
|
windowsHide: true
|
|
})
|
|
|
|
child.on('exit', (code) => {
|
|
process.exit(code ?? 0)
|
|
})
|
|
|
|
child.on('error', (error) => {
|
|
LogHelper.error(`Failed to start the Python TCP server: ${error}`)
|
|
process.exit(1)
|
|
})
|
|
})()
|