circlemind-ai?組織開發(fā)了一個(gè)名為fast-graphrag的開源項(xiàng)目。這個(gè)項(xiàng)目的目標(biāo)是提供一個(gè)高效、可解釋且精度高的快速圖檢索增強(qiáng)生成（Fast GraphRAG）框架。該框架專門為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)，他們開發(fā)了記憶和數(shù)據(jù)工具，使LLM應(yīng)用能夠利用高度專業(yè)化的檢索管道，而無需費(fèi)心設(shè)置和維護(hù)復(fù)雜的Agent工作流程。

該項(xiàng)目的主要特點(diǎn)和信息如下：

核心特性

• 知識(shí)的可解釋性和可調(diào)試性：利用圖形提供人類可瀏覽的知識(shí)視圖，支持查詢、可視化和更新。
• 高效、低成本、快速：針對(duì)大規(guī)模運(yùn)行而設(shè)計(jì)，無需昂貴的資源投入。
• 數(shù)據(jù)動(dòng)態(tài)性：自動(dòng)生成和優(yōu)化圖形，以最佳方式適應(yīng)特定領(lǐng)域和本體需求。
• 實(shí)時(shí)更新：支持?jǐn)?shù)據(jù)變化的即時(shí)更新。
• 智能探索能力：采用基于PageRank的圖形探索，提升準(zhǔn)確性和可靠性。
• 異步和類型化設(shè)計(jì)：完全異步，并提供完整的類型支持，確保工作流程的穩(wěn)健性和可預(yù)測(cè)性。

安裝指南和快速上手

• 安裝方式：可通過PyPi安裝（推薦）或從源代碼安裝。

PyPi安裝方法：

pip install fast-graphrag

• 源代碼安裝方法：

git clone https://github.com/circlemind-ai/fast-graphrag.git
cd fast-graphrag
poetry install

• 快速入門：
• 設(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)用不同的幽靈（過去、現(xiàn)在和未來）來引導(dǎo)斯克魯奇？",
    "為什么狄更斯選擇將故事分為'樂章'而非章節(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("斯克魯奇是誰？").response)

開源與托管服務(wù)

這個(gè)項(xiàng)目采用MIT許可證發(fā)布，允許自由使用和修改。對(duì)于那些希望快速可靠地開始使用Fast GraphRAG的用戶，項(xiàng)目還提供了托管服務(wù)選項(xiàng)。這項(xiàng)服務(wù)每月為用戶提供前100個(gè)免費(fèi)請(qǐng)求，之后則按實(shí)際使用量收費(fèi)。

項(xiàng)目地址：https://github.com/circlemind-ai/fast-graphrag

本文轉(zhuǎn)載自?? NLP前沿??，作者： ??NLP前沿??