59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Shared bootstrap helpers for process startup."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
|
|
|
|
def ensure_models_downloaded(interactive: bool) -> bool:
|
|
"""Ensure declared deployment models exist locally, downloading if needed."""
|
|
try:
|
|
from app.utils.download_models import check_all_models, download_models
|
|
|
|
missing = check_all_models()
|
|
if not missing:
|
|
return True
|
|
|
|
print(f"\n⚠️ 检测到 {len(missing)} 个模型未下载")
|
|
for model_id, *_ in missing:
|
|
print(f" - {model_id}")
|
|
|
|
print("\n将自动下载缺失模型后继续启动。")
|
|
if download_models(auto_mode=True):
|
|
return True
|
|
|
|
print("\n模型自动下载失败。")
|
|
if interactive:
|
|
print("可手动运行以下命令排查：")
|
|
print(" uv run python -m app.utils.download_models")
|
|
print(" ./scripts/prepare-models.sh")
|
|
else:
|
|
print("非交互式终端下请确认网络可用，或预先准备模型缓存。")
|
|
return False
|
|
except Exception as exc:
|
|
print(f"⚠️ 模型检查失败: {exc}")
|
|
return False
|
|
|
|
|
|
def run_cli_preflight() -> bool:
|
|
"""Preflight checks for the CLI entrypoint."""
|
|
try:
|
|
from app.core.accelerator import get_accelerator_info, validate_accelerator_runtime
|
|
|
|
ok, message = validate_accelerator_runtime()
|
|
info = get_accelerator_info()
|
|
print(
|
|
"Accelerator | "
|
|
f"vendor={info.vendor} runtime={info.runtime} device={info.device} "
|
|
f"count={info.device_count}"
|
|
)
|
|
if not ok:
|
|
print(f"加速器运行栈检查失败: {message}")
|
|
return False
|
|
except Exception as exc:
|
|
print(f"加速器检查失败: {exc}")
|
|
return False
|
|
|
|
return ensure_models_downloaded(interactive=sys.stdin.isatty())
|