mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-23 06:46:40 +08:00
Refa: only support MinerU-API now (#11977)
### What problem does this PR solve? Only support MinerU-API now, still need to complete frontend for pipeline to allow the configuration of MinerU options. ### Type of change - [x] Refactoring
This commit is contained in:
@ -201,64 +201,10 @@ function ensure_docling() {
|
||||
|| python3 -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://pypi.org/simple --no-cache-dir "docling${DOCLING_PIN}"
|
||||
}
|
||||
|
||||
function ensure_mineru() {
|
||||
[[ "${USE_MINERU}" == "true" ]] || { echo "[mineru] disabled by USE_MINERU"; return 0; }
|
||||
|
||||
export HUGGINGFACE_HUB_ENDPOINT="${HF_ENDPOINT:-https://hf-mirror.com}"
|
||||
|
||||
local default_prefix="/ragflow/uv_tools"
|
||||
local venv_dir="${default_prefix}/.venv"
|
||||
local exe="${MINERU_EXECUTABLE:-${venv_dir}/bin/mineru}"
|
||||
local mineru_backend="${MINERU_BACKEND:-pipeline}"
|
||||
local mineru_pkg="mineru[core]"
|
||||
|
||||
if [[ "${mineru_backend}" == vlm-* ]]; then
|
||||
mineru_pkg="mineru[core,vlm]"
|
||||
fi
|
||||
|
||||
if [[ -x "${exe}" ]]; then
|
||||
echo "[mineru] found: ${exe} (MINERU_BACKEND=${mineru_backend})"
|
||||
export MINERU_EXECUTABLE="${exe}"
|
||||
|
||||
if [[ "${mineru_backend}" == vlm-* ]]; then
|
||||
if ! "${venv_dir}/bin/python3" -c "import importlib.util, sys; sys.exit(0 if importlib.util.find_spec('vllm') else 1)" >/dev/null 2>&1; then
|
||||
echo "[mineru] vllm not found for MINERU_BACKEND=${mineru_backend}, installing ${mineru_pkg} ..."
|
||||
(
|
||||
set -e
|
||||
source "${venv_dir}/bin/activate"
|
||||
uv pip install -U "${mineru_pkg}" -i https://mirrors.aliyun.com/pypi/simple --extra-index-url https://pypi.org/simple
|
||||
deactivate
|
||||
) || return 1
|
||||
fi
|
||||
fi
|
||||
return 0
|
||||
fi
|
||||
|
||||
echo "[mineru] not found, bootstrapping with uv ... (MINERU_BACKEND=${mineru_backend}, pkg=${mineru_pkg})"
|
||||
|
||||
(
|
||||
set -e
|
||||
mkdir -p "${default_prefix}"
|
||||
cd "${default_prefix}"
|
||||
[[ -d "${venv_dir}" ]] || { echo "[mineru] creating venv at ${venv_dir} ..."; uv venv "${venv_dir}"; }
|
||||
|
||||
echo "[mineru] installing ${mineru_pkg} into ${venv_dir} ..."
|
||||
source "${venv_dir}/bin/activate"
|
||||
uv pip install -U "${mineru_pkg}" -i https://mirrors.aliyun.com/pypi/simple --extra-index-url https://pypi.org/simple
|
||||
deactivate
|
||||
)
|
||||
export MINERU_EXECUTABLE="${exe}"
|
||||
if ! "${MINERU_EXECUTABLE}" --help >/dev/null 2>&1; then
|
||||
echo "[mineru] installation failed: ${MINERU_EXECUTABLE} not working" >&2
|
||||
return 1
|
||||
fi
|
||||
echo "[mineru] installed: ${MINERU_EXECUTABLE}"
|
||||
}
|
||||
# -----------------------------------------------------------------------------
|
||||
# Start components based on flags
|
||||
# -----------------------------------------------------------------------------
|
||||
ensure_docling
|
||||
ensure_mineru
|
||||
|
||||
if [[ "${ENABLE_WEBSERVER}" -eq 1 ]]; then
|
||||
echo "Starting nginx..."
|
||||
|
||||
Reference in New Issue
Block a user