pip安装
额外依赖（如 PDF 加载、向量库）
pip install pymupdf chromadb
uv安装（推荐）
读取文档库
办公文档全套
uv add python-docx python-pptx openpyxl xlrd
解释：
- python-docx → 解析 Word
- python-pptx → 解析 PPT
- openpyxl、xlrd → 解析 Excel
PDF解析
uv add pymupdf pdfplumber
网页/HTML
uv add beautifulsoup4 lxml html5lib
markdown文档
自带支持，无需额外装
配置文件
uv add python-dotenv
向量数据库
chromadb
基础 RAG 示例
from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
from llama_index.llms.openai import OpenAI
# 1. 加载数据（./data 目录下的 PDF/Markdown）
documents = SimpleDirectoryReader("./data").load_data()
# 2. 构建向量索引（默认用 OpenAI Embedding+Chroma）
index = VectorStoreIndex.from_documents(documents)
# 3. 创建查询引擎
query_engine = index.as_query_engine(llm=OpenAI(model="gpt-3.5-turbo"))
# 4. 提问并获取答案
response = query_engine.query("文档的核心结论是什么？")
print(response)