/
Chaizee
/
ZenithCode_Incident-LLM-analytics
Обзор
Документация
Войти
/
Chaizee
/
ZenithCode_Incident-LLM-analytics
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
src/runtime_deps.py
47 строк
1 KB
Chaizee
fix: fix speed
12 июн 2026, 16:28
12 июн 2026, 16:28
b2745cc
Код
Авторство
О чём код?
from __future__ import annotations from config import USE_GPU from src.device_utils import gpu_setup_hint RUNTIME_PACKAGES: tuple[tuple[str, str], ...] = ( ("polars", "polars"), ("pandas", "pandas"), ("sentence_transformers", "sentence-transformers"), ("sklearn", "scikit-learn"), ("joblib", "joblib"), ("xgboost", "xgboost"), ("streamlit", "streamlit"), ("loguru", "loguru"), ("openpyxl", "openpyxl"), ("llama_cpp", "llama-cpp-python"), ) def missing_runtime_packages() -> list[str]: missing: list[str] = [] for module_name, pip_name in RUNTIME_PACKAGES: try: __import__(module_name) except ImportError: missing.append(pip_name) return missing def runtime_deps_error_message(missing: list[str] | None = None) -> str: pkgs = missing if missing is not None else missing_runtime_packages() if not pkgs: return "" joined = " ".join(sorted(set(pkgs))) return ( "Не установлены обязательные пакеты для анализа.\n\n" f"В активированном venv выполните:\n\n" f" pip install -r requirements.txt\n\n" f"Или точечно:\n\n" f" pip install {joined}" ) def gpu_runtime_notice() -> str: if not USE_GPU: return "" return gpu_setup_hint()