提升Graph-RAG查询生成的提示策略指南
在这篇文章中,我们将探讨如何优化对图形数据库查询的生成,重点关注在提示中获取相关数据库信息的方法。
引言
在处理图形数据库时,生成有效的Cypher查询至关重要。本文旨在提供实用策略,帮助开发者精确生成这些查询,特别是在使用LangChain和Neo4j时。
主要内容
设置环境
首先安装所需的包并设置环境变量:
%pip install --upgrade --quiet langchain langchain-community langchain-openai neo4j
由于某些地区的网络限制,开发者可能需要考虑使用API代理服务来提高访问稳定性。
import getpass
import os
os.environ["OPENAI_API_KEY"] = getpass.getpass()
os.environ["NEO4J_URI"] = "bolt://localhost:7687"
os.environ["NEO4J_USERNAME"] = "neo4j"
os.environ["NEO4J_PASSWORD"] = "password"
导入数据
接下来,我们将连接到Neo4j数据库并导入关于电影和演员的示例数据:
from langchain_community.graphs import Neo4jGraph
graph = Neo4jGraph()
movies_query = """
LOAD CSV WITH HEADERS FROM
'https://raw.githubusercontent.com/tomasonjo/blog-datasets/main/movies/movies_small.csv'
AS row
MERGE (m:Movie {id:row.movieId})
SET m.released = date(row.released),
m.title = row.title,
m.imdbRating = toFloat(row.imdbRating)
FOREACH (director in split(row.director, '|') |
MERGE (p:Person {name:trim(director)})
MERGE (p)-[:DIRECTED]->(m))
FOREACH (actor in split(row.actors, '|') |
MERGE (p:Person {name:trim(actor)})
MERGE (p)-[:ACTED_IN]->(m))
FOREACH (genre in split(row.genres, '|') |
MERGE (g:Genre {name:trim(genre)})
MERGE (m)-[:IN_GENRE]->(g))
"""
graph.query(movies_query)
过滤图模式
如果你需要生成特定子集的Cypher语句,可以使用如下方法排除不需要的节点:
from langchain.chains import GraphCypherQAChain
from langchain_openai import ChatOpenAI
llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
chain = GraphCypherQAChain.from_llm(
graph=graph, llm=llm, exclude_types=["Genre"], verbose=True
)
print(chain.graph_schema)
Few-shot示例
通过增加自然语言问题及其对应的Cypher查询示例,可以提升模型性能。以下是一些示例:
examples = [
{"question": "How many artists are there?", "query": "MATCH (a:Person)-[:ACTED_IN]->(:Movie) RETURN count(DISTINCT a)"},
{"question": "Which actors played in the movie Casino?", "query": "MATCH (m:Movie {title: 'Casino'})<-[:ACTED_IN]-(a) RETURN a.name"},
# 更多示例...
]
from langchain_core.prompts import FewShotPromptTemplate, PromptTemplate
example_prompt = PromptTemplate.from_template(
"User input: {question}\nCypher query: {query}"
)
prompt = FewShotPromptTemplate(
examples=examples[:5],
example_prompt=example_prompt,
prefix="You are a Neo4j expert. Given an input question, create a syntactically correct Cypher query to run.\n\nHere is the schema information\n{schema}.\n\nBelow are a number of examples of questions and their corresponding Cypher queries.",
suffix="User input: {question}\nCypher query: ",
input_variables=["question", "schema"],
)
动态Few-shot示例
使用SemanticSimilarityExampleSelector
来动态选择最相关的Few-shot示例:
from langchain_community.vectorstores import Neo4jVector
from langchain_core.example_selectors import SemanticSimilarityExampleSelector
from langchain_openai import OpenAIEmbeddings
example_selector = SemanticSimilarityExampleSelector.from_examples(
examples,
OpenAIEmbeddings(),
Neo4jVector,
k=5,
input_keys=["question"],
)
prompt = FewShotPromptTemplate(
example_selector=example_selector,
example_prompt=example_prompt,
prefix="You are a Neo4j expert. Given an input question, create a syntactically correct Cypher query to run.\n\nHere is the schema information\n{schema}.\n\nBelow are a number of examples of questions and their corresponding Cypher queries.",
suffix="User input: {question}\nCypher query: ",
input_variables=["question", "schema"],
)
代码示例
以下是一个完整的代码示例,帮助你生成并执行查询:
llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
chain = GraphCypherQAChain.from_llm(
graph=graph, llm=llm, cypher_prompt=prompt, verbose=True
)
response = chain.invoke("How many actors are in the graph?")
print(response)
常见问题和解决方案
- 网络限制问题:在无法访问Neo4j或API服务时,请使用API代理服务。
- 查询性能:确保查询优化,避免冗余节点和关系。
总结和进一步学习资源
通过优化模型提示并使用适当的示例选择策略,开发者可以有效提高图数据库查询的生成效率。
参考资料
- LangChain官方文档
- Neo4j图形数据库文档
如果这篇文章对你有帮助,欢迎点赞并关注我的博客。您的支持是我持续创作的动力!
—END—