Elijah/convertor
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: 完善 runtime 離線支援

Browse source 
- 使用 babeldoc --warmup 下載完整資源（模型、字型、cmap、tiktoken）
- 修正驗證邏輯，指向正確的 BabelDOC cache 路徑
- 移除舊的 PDFMATHTRANSLATE_MODELS_PATH 環境變數
- 更新 pdfmathtranslate.ts 使用正確的模型路徑
- 移除手動指定 --onnx 參數（pdf2zh 內部自動處理）

確保所有組件 Runtime 不需要下載任何資源
This commit is contained in:
Your Name 2026-01-23 11:55:02 +08:00
parent cb8ee290bb
commit 9fc7217df1
2 changed files with 48 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

34
src/converters/pdfmathtranslate.ts
View file

@ -18,8 +18,10 @@ type TranslationService = (typeof TRANSLATION_SERVICES)[number];
* - original.pdf（原始 PDF）
* - translated-<lang>.pdf（翻譯後的 PDF）
*
* 必須在 Docker build 階段預先下載所需模型，
* 不允許在 runtime 隱式下載模型。
* 模型路徑說明：
* - pdf2zh 使用 babeldoc.assets 內部載入 ONNX 模型
* - 模型預先下載到 /root/.cache/babeldoc/models/ 目錄
* - Runtime 不會再下載任何模型（由 Docker build 預下載）
*/
// 支援的目標語言列表
@ -41,8 +43,10 @@ const SUPPORTED_LANGUAGES = [
"th", // Thai
] as const;
// 模型路徑（Docker 環境中）
const MODELS_PATH = process.env.PDFMATHTRANSLATE_MODELS_PATH || "/models/pdfmathtranslate";
// 模型路徑（BabelDOC cache 目錄，由 pdf2zh 內部使用）
// 注意：pdf2zh 會自動從 babeldoc.assets 載入模型，此路徑僅供參考
const BABELDOC_CACHE_PATH = process.env.BABELDOC_CACHE_PATH || "/root/.cache/babeldoc";
const MODELS_PATH = `${BABELDOC_CACHE_PATH}/models`;
// 生成 from/to 格式映射
function generateLanguageMappings(): {
@ -110,12 +114,21 @@ function normalizeLanguageCode(lang: string): string {
* @returns 模型是否存在
*/
function checkModelsExist(): boolean {
// 檢查 ONNX 模型目錄是否存在
// 檢查 BabelDOC ONNX 模型目錄是否存在
if (!existsSync(MODELS_PATH)) {
console.warn(`[PDFMathTranslate] Models directory not found: ${MODELS_PATH}`);
console.warn(`[PDFMathTranslate] BabelDOC models directory not found: ${MODELS_PATH}`);
console.warn(`[PDFMathTranslate] Models should be pre-downloaded during Docker build.`);
return false;
}
// 檢查特定的 ONNX 模型檔案
const onnxModelPath = join(MODELS_PATH, "doclayout_yolo_docstructbench_imgsz1024.onnx");
if (!existsSync(onnxModelPath)) {
console.warn(`[PDFMathTranslate] ONNX model not found: ${onnxModelPath}`);
console.warn(`[PDFMathTranslate] Model should be pre-downloaded during Docker build.`);
return false;
}
return true;
}
@ -195,13 +208,8 @@ function runPdf2zhWithService(
const args = [inputPath, "-lo", targetLang, "-o", outputDir, "-s", service];
// 如果設定了自訂模型路徑，使用 --onnx 參數
if (existsSync(MODELS_PATH)) {
const onnxModelPath = join(MODELS_PATH, "model.onnx");
if (existsSync(onnxModelPath)) {
args.push("--onnx", onnxModelPath);
}
}
// pdf2zh 使用 babeldoc.assets 內部載入模型，不需要手動指定 --onnx 參數
// 模型路徑：/root/.cache/babeldoc/models/doclayout_yolo_docstructbench_imgsz1024.onnx
console.log(`[PDFMathTranslate] Running: pdf2zh ${args.join(" ")} (service: ${service})`);