wehub-resource-sync
83 lines
1.9 KiB
JavaScript
83 lines
1.9 KiB
JavaScript
import fs from 'node:fs'
|
|
import path from 'node:path'
|
|
|
|
import {
|
|
LLM_MANIFEST_PATH,
|
|
PYTORCH_MANIFEST_PATH,
|
|
PYTHON_TCP_SERVER_ASR_MODEL_DIR_PATH,
|
|
PYTHON_TCP_SERVER_TTS_BERT_BASE_DIR_PATH,
|
|
PYTHON_TCP_SERVER_TTS_MODEL_PATH
|
|
} from '@/constants'
|
|
import {
|
|
ASR_MODEL_FILES,
|
|
TTS_BERT_BASE_MODEL_FILES
|
|
} from '@/core/voice/voice-resource-state'
|
|
|
|
function readManifest(manifestPath) {
|
|
if (!fs.existsSync(manifestPath)) {
|
|
return null
|
|
}
|
|
|
|
try {
|
|
return JSON.parse(fs.readFileSync(manifestPath, 'utf8'))
|
|
} catch {
|
|
return null
|
|
}
|
|
}
|
|
|
|
function hasManifest(manifestPath) {
|
|
return readManifest(manifestPath) !== null
|
|
}
|
|
|
|
function hasAllFiles(directoryPath, fileNames) {
|
|
return fileNames.every((fileName) =>
|
|
fs.existsSync(path.join(directoryPath, fileName))
|
|
)
|
|
}
|
|
|
|
/**
|
|
* Inspect whether Leon's default local LLM has already been installed.
|
|
*/
|
|
export function inspectLocalAISetupState() {
|
|
const manifest = readManifest(LLM_MANIFEST_PATH)
|
|
const defaultInstalledLLMPath = manifest?.defaultInstalledLLMPath
|
|
|
|
if (
|
|
typeof defaultInstalledLLMPath !== 'string' ||
|
|
defaultInstalledLLMPath.trim() === ''
|
|
) {
|
|
return {
|
|
isInstalled: false,
|
|
label: ''
|
|
}
|
|
}
|
|
|
|
const modelPath = path.resolve(defaultInstalledLLMPath)
|
|
|
|
return {
|
|
isInstalled: fs.existsSync(modelPath),
|
|
label:
|
|
manifest?.name && manifest?.version
|
|
? `${manifest.name} (${manifest.version})`
|
|
: path.basename(modelPath)
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Inspect whether voice mode has already been installed at least once.
|
|
* Version-specific updates are handled by the voice setup pass.
|
|
*/
|
|
export function inspectVoiceSetupState() {
|
|
const hasVoiceModels =
|
|
hasAllFiles(
|
|
PYTHON_TCP_SERVER_TTS_BERT_BASE_DIR_PATH,
|
|
TTS_BERT_BASE_MODEL_FILES
|
|
) &&
|
|
fs.existsSync(PYTHON_TCP_SERVER_TTS_MODEL_PATH) &&
|
|
hasAllFiles(PYTHON_TCP_SERVER_ASR_MODEL_DIR_PATH, ASR_MODEL_FILES)
|
|
|
|
return {
|
|
isInstalled: hasManifest(PYTORCH_MANIFEST_PATH) && hasVoiceModels
|
|
}
|
|
}
|