🖇️ Repository • 📙 Documentation
Graphrag-UI is a cutting-edge tool that integrates AI capabilities into your rag analysis process. It leverages both Graph RAG and traditional RAG techniques to offer enhanced retrieval and generation capabilities. Graphrag-UI provides global and local search modes, file management, and indexing capabilities, all through a Streamlit-based user interface.
| 🔍 Advanced retrieval | Using Graph RAG and traditional RAG |
| 🌐 Global and local search modes | For comprehensive data exploration |
| 🤖 Integration with OpenAI | Leverages OpenAI's language models |
| 📂 File management | Efficient file management and indexing capabilities |
| 🖥️ Streamlit-based UI | User-friendly interface for seamless interaction |
| ⚙️ Customizable search parameters | Tailor search parameters to fit specific needs |
Graphrag-UI incorporates Graph RAG, a superior technique for retrieval that offers several benefits over traditional RAG:
- Enhanced context understanding: Graph RAG leverages knowledge graphs to capture complex relationships between entities, providing more accurate and contextually rich answers.
- Improved handling of structured and unstructured data: It excels at representing and retrieving heterogeneous and interconnected information.
- Better performance in domain-specific queries: Graph RAG addresses limitations of generic embedding models in company-specific knowledge retrieval.
-
Clone the repository:
git clone https://github.com/shahshrey/Graphrag-UI.git cd Graphrag-UI -
Create a virtual environment and activate it:
python -m venv venv source venv/bin/activate # On Windows, use venv\Scripts\activate
-
Install the required dependencies:
pip install -r requirements.txt
-
Set up your environment variables:
- Copy the
.env.examplefile to.env - Fill in your API keys and other configuration values
- Copy the
-
Run the Streamlit app:
streamlit run main.py
-
Open your web browser and navigate to the URL provided by Streamlit (usually http://localhost:8501)
-
Use the sidebar to configure Graphrag-UI and upload your documents
-
Ask questions and interact with Graphrag-UI through the main interface, leveraging both Graph RAG and traditional RAG capabilities for comprehensive information retrieval
src/: Contains the main application codetests/: Contains test files (to be implemented)docs/: Contains additional documentationmain.py: The entry point of the applicationrequirements.txt: Lists all Python dependencies.env.example: Template for environment variables
Contributions are welcome! Please feel free to submit a Pull Request.
This project is licensed under the MIT License.