Skip to main content
在 GitHub 上打开

Graph RAG

本指南介绍了 Graph RAG。有关所有 支持的功能和配置,请参阅 Graph RAG 项目页面

概述

GraphRetrieverlangchain-graph-retrieverpackage 提供了一个 LangChain 检索器,它结合了非结构化的相似性搜索 在具有元数据属性的结构化遍历的向量上。这将启用基于图形的 检索现有向量存储。

集成详细信息

RetrieverPyPI 软件包最近的项目页面
GraphRetrievergithub.com/datastax/graph-raglangchain-graph-retrieverPyPI - VersionGraph RAG

好处

设置

安装

这只Retriever生活在langchain-graph-retriever包。

pip install -qU langchain-graph-retriever

实例

以下示例将演示如何对某些样本执行图形遍历 关于动物的文件。

先决条件

Toggle for Details
  1. 确保您已安装 Python 3.10+

  2. 安装以下提供示例数据的包。

    pip install -qU graph_rag_example_helpers
  3. 下载测试文件:

    from graph_rag_example_helpers.datasets.animals import fetch_documents
    animals = fetch_documents()
  4. 选择嵌入模型
  5. OpenAI
  6. 天蓝色
  7. Google
  8. AWS
  9. Hugging Face
  10. 奥拉马
  11. Cohere
  12. MistralAI
  13. 诺米克
  14. 英伟达
  15. 航海 AI
  16. IBM 沃森
pip install -qU langchain-openai
import getpass
import os

if not os.environ.get("OPENAI_API_KEY"):
os.environ["OPENAI_API_KEY"] = getpass.getpass("Enter API key for OpenAI: ")

from langchain_openai import OpenAIEmbeddings

embeddings = OpenAIEmbeddings(model="text-embedding-3-large")