easy-rag/setup.sh

67 lines
1.8 KiB
Bash
Executable File

#!/bin/bash
# Easy RAG Service 环境设置脚本
echo "=== Easy RAG Service 环境设置 ==="
# 检查 Python 版本
echo "检查 Python 版本..."
python_version=$(python3 --version 2>&1 | cut -d" " -f2 | cut -d"." -f1,2)
required_version="3.8"
if [ "$(printf '%s\n' "$required_version" "$python_version" | sort -V | head -n1)" = "$required_version" ]; then
echo "✓ Python 版本符合要求: $python_version"
else
echo "✗ Python 版本过低: $python_version (需要 >= $required_version)"
echo "请使用 pyenv 安装合适的 Python 版本:"
echo " pyenv install 3.13.0 # 推荐最新稳定版"
echo " pyenv install 3.12.0 # 或其他 3.8+ 版本"
echo " pyenv local 3.13.0 # 使用你安装的版本"
exit 1
fi
# 创建虚拟环境
if [ ! -d "venv" ]; then
echo "创建虚拟环境..."
python3 -m venv venv
echo "✓ 虚拟环境创建完成"
else
echo "✓ 虚拟环境已存在"
fi
# 激活虚拟环境
echo "激活虚拟环境..."
source venv/bin/activate
# 升级 pip
echo "升级 pip..."
pip install --upgrade pip
# 安装依赖
echo "安装项目依赖..."
pip install -r requirements.txt
# 创建环境变量文件
if [ ! -f ".env" ]; then
echo "创建环境变量文件..."
cp .env.example .env
echo "✓ 已创建 .env 文件,请编辑设置你的 API 密钥"
else
echo "✓ .env 文件已存在"
fi
# 创建必要目录
echo "创建必要目录..."
mkdir -p uploads
mkdir -p chroma_db
echo ""
echo "=== 设置完成 ==="
echo "请完成以下步骤:"
echo "1. 编辑 .env 文件,设置你的 OpenAI API 密钥"
echo "2. 激活虚拟环境: source venv/bin/activate"
echo "3. 启动服务: ./start.sh 或 python main.py"
echo ""
echo "快速启动:"
echo " source venv/bin/activate && python main.py"