Repository files navigation

GraphRAG is a technique proposed by Microsoft Research to generate knowledge graphs from text using LLMs. This project is an implementation of GraphRAG in Python. It makes it simple to get up and running with GraphRAG and generate knowledge graphs from text. It also provides a semantic search interface to query the knowledge graph.

This project usesuv for Python dependency management.

To install uv, follow the instructionshere.

Install the dependencies with the following command:

uv run pip install

1. Add the files you want to add to your knowledge graph in thedata folder.

2. Set yourGOOGLE_API_KEY in the.env file. You can get it fromhere.

3. Run the following command to generate the knowledge graph:

First, start the Chroma server:

chroma run --path db

python3 graph.py

4. The knowledge graph will be saved in a chromadb database in thedb folder. To query it, run

python3 query_cli.py

5. To visualize the knowledge graph, run

streamlit run visualize.py