Graph RAG with filtering
For large and complex graphs, LLMs can sometimes struggle to accurately infer the most relevant nodes and relationships to build the Cypher query. Quite often, you will only need the LLM to be aware of a subset of the graph, and excluding particular node types will not only make it easier for the LLM to accurately create the Cypher query, but it will improve the query latency.
The graph database you've been working with is available as graph
.
This exercise is part of the course
Retrieval Augmented Generation (RAG) with LangChain
Exercise instructions
- Create a graph QA chain that queries the
graph
database while ignoring nodes with the"Concept"
type; anllm
has been defined for you, and you should setverbose=True
. - Invoke the
graph_qa_chain
with the input provided.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Create the graph QA chain excluding Concept
graph_qa_chain = ____
# Invoke the chain with the input provided
result = ____({"query": "Who was Marie Curie married to?"})
print(f"Final answer: {result['result']}")