adaptive_rag / requirements.txt
lanny xu
Use CPU-only PyTorch to reduce build size and time
83e265e
# 自适应RAG系统依赖文件
# 运行以下命令安装: pip install -r requirements.txt
# 核心框架
# 使用 CPU 版本的 PyTorch 以减小体积 (Hugging Face Spaces 免费版没有 GPU)
--extra-index-url https://download.pytorch.org/whl/cpu
torch>=2.0.0
langchain>=0.1.0
langgraph>=0.0.40
langchain-community>=0.0.20
langchain-core>=0.1.0
langchain-text-splitters>=0.0.1
# LLM集成
langchain-ollama>=0.1.0
# 向量数据库和嵌入
# chromadb>=0.4.0 # 已移除,统一使用 Milvus
pymilvus[milvus_lite]>=2.4.2 # Milvus 客户端及 Lite 本地模式支持
sentence-transformers>=2.2.0
torch>=2.0.0
transformers>=4.30.0
# 文档处理
tiktoken>=0.5.0
beautifulsoup4>=4.12.0
requests>=2.31.0
Pillow>=9.0.0 # 图像处理,支持多模态功能
rank-bm25>=0.2.2 # BM25检索,用于混合搜索
# 幻觉检测
sentence-transformers>=2.2.0 # NLI 模型支持
scikit-learn>=1.3.0 # 相似度计算
# 网络搜索
tavily-python>=0.3.0
# 数据处理
numpy>=1.24.0,<2.0 # 避免NumPy 2.x兼容性问题
pandas>=2.0.0
# 工具库
python-dotenv>=1.0.0
pydantic>=2.0.0
typing-extensions>=4.0.0
# 开发工具(可选)
jupyter>=1.0.0
ipykernel>=6.0.0
matplotlib>=3.7.0
seaborn>=0.12.0
# GraphRAG相关(可选)
networkx>=3.1 # 图结构处理
python-louvain>=0.16 # 社区检测
fastapi>=0.104.0
uvicorn>=0.23.0
python-multipart>=0.0.6
# pyngrok>=7.0.0 # 可选:用于Web服务穿透
# cloudflared # 可选:用于Web服务穿透