【亲测免费】 Neo4j GenAI Python 开源项目使用教程

1. 项目介绍

Neo4j GenAI Python 是一个开源项目，旨在利用 Neo4j 图数据库和 Python 的强大功能，帮助开发人员构建基于图检索增强生成（GraphRAG）的应用程序。作为 Neo4j 的一方库，它提供了功能丰富、性能高效且长期支持的解决方案。

2. 项目快速启动

在开始之前，请确保您已经安装了 Neo4j 数据库，并且它正在运行中。以下是如何快速启动 Neo4j GenAI Python 的步骤：

# 安装 Neo4j GenAI Python 包
pip install neo4j-graphrag

# 导入必要的库
from neo4j import GraphDatabase
from neo4j_graphrag.embeddings import OpenAIEmbeddings
from neo4j_graphrag.experimental.pipeline.kg_builder import SimpleKGPipeline
from neo4j_graphrag.llm import OpenAILLM

# 设置 Neo4j 数据库连接
NEO4J_URI = "neo4j://localhost:7687"
NEO4J_USERNAME = "neo4j"
NEO4J_PASSWORD = "password"
driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USERNAME, NEO4J_PASSWORD))

# 创建 Embedder 对象
embedder = OpenAIEmbeddings(model="text-embedding-3-large")

# 实例化 LLM
llm = OpenAILLM(model_name="gpt-4o", model_params={"max_tokens": 2000, "response_format": {"type": "json_object"}, "temperature": 0})

# 实例化 SimpleKGPipeline
kg_builder = SimpleKGPipeline(llm=llm, driver=driver, embedder=embedder, entities=["Person", "House", "Planet"], relations=["PARENT_OF", "HEIR_OF", "RULES"], on_error="IGNORE", from_pdf=False)

# 运行管道
text = ("The son of Duke Leto Atreides and the Lady Jessica, Paul is the heir of House "
        "Atreides, an aristocratic family that rules the planet Caladan.")
asyncio.run(kg_builder.run_async(text=text))

# 关闭数据库连接
driver.close()

确保在运行上述代码前，您已经将环境变量 OPENAI_API_KEY 设置为您的 OpenAI API 密钥。

3. 应用案例和最佳实践

以下是一些使用 Neo4j GenAI Python 的应用案例和最佳实践：

• 构建知识图谱：利用 SimpleKGPipeline 类快速构建知识图谱，适合简单场景。
• 增强向量搜索：通过图遍历增强向量搜索，提高检索效率。
• 混合检索：结合传统检索和图检索，为 GraphRAG 应用程序提供更强大的检索能力。

4. 典型生态项目

在 Neo4j GenAI Python 的生态系统中，以下是一些典型的项目：

• Ollama：提供 LLMs（大型语言模型）服务。
• OpenAI：OpenAI 提供的 LLMs 服务。
• Google：Vertex AI 提供的 LLMs 服务。
• Cohere：Cohere 提供的 LLMs 服务。
• Weaviate、Pinecone、Qdrant：用于存储向量的数据库。

通过上述介绍和教程，您可以开始使用 Neo4j GenAI Python 进行图检索增强生成应用程序的开发。