GraphRAG進(jìn)化，效率翻倍！
circlemind-ai?組織開(kāi)發(fā)了一個(gè)名為fast-graphrag的開(kāi)源項(xiàng)目。這個(gè)項(xiàng)目的目標(biāo)是提供一個(gè)高效、可解釋且精度高的快速圖檢索增強(qiáng)生成（Fast GraphRAG）框架。該框架專(zhuān)門(mén)為Agent驅(qū)動(dòng)的檢索工作流程設(shè)計(jì)，能夠輕松融入檢索管道中，提供先進(jìn)的RAG功能，同時(shí)避免了構(gòu)建和設(shè)計(jì)Agent工作流程的繁瑣復(fù)雜性。
項(xiàng)目的宗旨是增加全球成功的通用人工智能（GenAI）應(yīng)用數(shù)量。為實(shí)現(xiàn)這一目標(biāo)，他們開(kāi)發(fā)了記憶和數(shù)據(jù)工具，使LLM應(yīng)用能夠利用高度專(zhuān)業(yè)化的檢索管道，而無(wú)需費(fèi)心設(shè)置和維護(hù)復(fù)雜的Agent工作流程。
該項(xiàng)目的主要特點(diǎn)和信息如下：
核心特性
- 知識(shí)的可解釋性和可調(diào)試性：利用圖形提供人類(lèi)可瀏覽的知識(shí)視圖，支持查詢(xún)、可視化和更新。
- 高效、低成本、快速：針對(duì)大規(guī)模運(yùn)行而設(shè)計(jì)，無(wú)需昂貴的資源投入。
- 數(shù)據(jù)動(dòng)態(tài)性：自動(dòng)生成和優(yōu)化圖形，以最佳方式適應(yīng)特定領(lǐng)域和本體需求。
- 實(shí)時(shí)更新：支持?jǐn)?shù)據(jù)變化的即時(shí)更新。
- 智能探索能力：采用基于PageRank的圖形探索，提升準(zhǔn)確性和可靠性。
- 異步和類(lèi)型化設(shè)計(jì)：完全異步，并提供完整的類(lèi)型支持，確保工作流程的穩(wěn)健性和可預(yù)測(cè)性。
安裝指南和快速上手
- 安裝方式：可通過(guò)PyPi安裝（推薦）或從源代碼安裝。
PyPi安裝方法：
pip install fast-graphrag- 源代碼安裝方法：
git clone https://github.com/circlemind-ai/fast-graphrag.git
cd fast-graphrag
poetry install- 快速入門(mén)：
- 設(shè)置OpenAI API密鑰：
export OPENAI_API_KEY="sk-..."- 獲取《圣誕頌歌》文本：
curl https://raw.githubusercontent.com/circlemind-ai/fast-graphrag/refs/heads/main/mock_data.txt > ./book.txt- Python代碼示例：
from fast_graphrag import GraphRAG
DOMAIN = "分析這個(gè)故事并識(shí)別其中的角色。重點(diǎn)關(guān)注他們之間的互動(dòng)、探索的地點(diǎn)以及彼此的關(guān)系。"
EXAMPLE_QUERIES = [
"圣誕前夜在《圣誕頌歌》中有何重要意義？",
"維多利亞時(shí)代的倫敦背景如何影響故事主題？",
"描述導(dǎo)致斯克魯奇性格轉(zhuǎn)變的一系列事件。",
"狄更斯如何運(yùn)用不同的幽靈（過(guò)去、現(xiàn)在和未來(lái)）來(lái)引導(dǎo)斯克魯奇？",
"為什么狄更斯選擇將故事分為'樂(lè)章'而非章節(jié)？"
]
ENTITY_TYPES = ["人物", "動(dòng)物", "地點(diǎn)", "物品", "活動(dòng)", "事件"]
grag = GraphRAG(
working_dir="./book_example",
domain=DOMAIN,
example_queries="\n".join(EXAMPLE_QUERIES),
entity_types=ENTITY_TYPES
)
with open("./book.txt") as f:
grag.insert(f.read())
print(grag.query("斯克魯奇是誰(shuí)？").response)開(kāi)源與托管服務(wù)
這個(gè)項(xiàng)目采用MIT許可證發(fā)布，允許自由使用和修改。對(duì)于那些希望快速可靠地開(kāi)始使用Fast GraphRAG的用戶(hù)，項(xiàng)目還提供了托管服務(wù)選項(xiàng)。這項(xiàng)服務(wù)每月為用戶(hù)提供前100個(gè)免費(fèi)請(qǐng)求，之后則按實(shí)際使用量收費(fèi)。
項(xiàng)目地址：https://github.com/circlemind-ai/fast-graphrag
本文轉(zhuǎn)載自?? NLP前沿??，作者： ??NLP前沿??
