RAPTOR树与BEG集成使用

与BEG集成使用

# 集成BGE和RAPTOR
bge = BGEEmbedding()
summarizer = LLMSummarizer()
raptor_builder = RAPTORTreeBuilder(
    embedding_model=bge,
    summarizer=summarizer
)

# 创建RAPTOR向量存储
vector_store = RAPTORVectorStore(raptor_builder)

# 添加文档
chunks = [
    {"text": "装备维修手册第1章...", "metadata": {"source": "manual"}},
    {"text": "装备维修手册第2章...", "metadata": {"source": "manual"}}
]
vector_store.add_documents(chunks)

# 搜索
results = vector_store.search("如何更换装备零件", top_k=5)