50 lines
1.8 KiB
Bash
Executable File
50 lines
1.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 本地 PDF→Markdown 微服务（与 langchain file_converter.pdf_to_html 约定一致，默认 :6006）
|
||
# 依赖 conda 环境 langchain-chat（含 fastapi、uvicorn、PyMuPDF）
|
||
[ -n "${BASH_VERSION:-}" ] || exec /usr/bin/env bash "$0" ${1+"$@"}
|
||
set -u
|
||
source "$(cd "$(dirname "$0")" && pwd)/common-restart.sh"
|
||
LOG_FILE="$LOG_DIR/pdf-convert-service.log"
|
||
LC_ROOT="$GANGYAN_ROOT/langchain-chat"
|
||
CONDA_SH="/opt/software/miniconda3/etc/profile.d/conda.sh"
|
||
|
||
log_line() {
|
||
local s="[$(date '+%Y-%m-%d %H:%M:%S')] $*"
|
||
echo "$s" >> "$LOG_FILE"
|
||
echo "$s"
|
||
}
|
||
|
||
log_line "======== 停止 pdf-convert-service (6006) ========"
|
||
pkill -f "uvicorn.*pdf_convert_service\.app:app" 2>/dev/null || true
|
||
sleep 1
|
||
if command -v lsof >/dev/null 2>&1; then
|
||
for pid in $(lsof -ti :6006 -sTCP:LISTEN 2>/dev/null || true); do
|
||
kill -TERM "$pid" 2>/dev/null && log_line "TERM pid=$pid :6006"
|
||
done
|
||
sleep 1
|
||
for pid in $(lsof -ti :6006 -sTCP:LISTEN 2>/dev/null || true); do
|
||
kill -KILL "$pid" 2>/dev/null && log_line "KILL pid=$pid :6006"
|
||
done
|
||
fi
|
||
|
||
log_line "======== 启动 pdf-convert-service ========"
|
||
if [ ! -f "$CONDA_SH" ]; then
|
||
log_line "错误: 未找到 $CONDA_SH"
|
||
exit 1
|
||
fi
|
||
# shellcheck source=/dev/null
|
||
source "$CONDA_SH"
|
||
conda activate langchain-chat
|
||
cd "$LC_ROOT"
|
||
export PYTHONPATH="$LC_ROOT"
|
||
export PDF_CONVERT_KB_ROOT="${PDF_CONVERT_KB_ROOT:-$LC_ROOT/knowledge_base}"
|
||
export PDF_CONVERT_HOST="${PDF_CONVERT_HOST:-127.0.0.1}"
|
||
export PDF_CONVERT_PORT="${PDF_CONVERT_PORT:-6006}"
|
||
|
||
nohup python -m uvicorn pdf_convert_service.app:app \
|
||
--host "$PDF_CONVERT_HOST" --port "$PDF_CONVERT_PORT" \
|
||
>> "$LOG_FILE" 2>&1 &
|
||
log_line "已后台启动 PID=$! 监听 ${PDF_CONVERT_HOST}:${PDF_CONVERT_PORT}"
|
||
log_line "日志: $LOG_FILE KB_ROOT=$PDF_CONVERT_KB_ROOT"
|
||
log_line "健康检查: curl -sS http://${PDF_CONVERT_HOST}:${PDF_CONVERT_PORT}/health"
|