Skip to main content

Multi-agentic knowledge graph construction and reasoning system

Project description

GraphAide

GraphAide Logo

GraphAide is an advanced multi-agentic query and reasoning system that constructs knowledge graphs (KG) from diverse sources and enables querying and reasoning over the resulting graphs. It harnesses both knowledge graphs and LLMs to rapidly develop domain-specific digital assistants.

Technical Summary

GraphAide v2 is a multi-agent system that uses LangGraph for workflow orchestration, LLMs for entity/relationship extraction, Neo4j for graph storage, and ChromaDB for vector embeddings.

The codebase follows a factory pattern: ModelFactory, GraphDBFactory, VectorStoreFactory, and AgentFactory create and inject dependencies into specialized agents. Agents communicate through shared state objects (KGGenerationState) that track nodes, edges, and messages as data flows through the pipeline.

Project Structure:

  • src/graphgen/v2/ - Core modules (factories, state, workflows)
  • src/graphgen/v2/agents/ - Agent implementations (extractor, loaders, visualizers)
  • templates/ - Prompt templates for each agent
  • tests/notebooks/ - Demo notebooks and test data

Tested Models:

Provider LLM Embedding
LMStudio (local) google/gemma-3-27b:2 text-embedding-nomic-embed-text-v1.5
AWS Bedrock anthropic.claude-3-5-sonnet-20241022-v2:0 amazon.titan-embed-text-v1
OpenAI gpt-4o text-embedding-3-small

Quick Start

📚 Documentation:

  • QUICKSTART.md - Installation, setup, and first examples
  • ARCHITECTURE.md - Deep technical overview (factories, agents, state, workflows)
  • REFERENCE.md - Complete CLI/API reference and feature matrix
  • CLAUDE.md - Guide for Claude AI Code (developers only)

Installation

# From project root
pip install .

# With Docker
docker compose up -d

Deployment Scripts

Linux/Mac/WSL

./graphaide_docker_build.sh 2.0.1              # Build production + dev images
./start-graphaide-prod.sh                      # Start production services
./start-graphaide-dev.sh                       # Start development services

Windows (PowerShell)

.\scripts\windows\graphaide_docker_build.ps1 2.0.1        # Build images
.\scripts\windows\start-graphaide-prod.ps1                # Start production
.\scripts\windows\start-graphaide-dev.ps1                 # Start development

Note: For Windows users, we recommend using WSL (Windows Subsystem for Linux) with the bash scripts above. Native PowerShell scripts are available in scripts/windows/ for reference.

Usage

Ways to Run GraphAide:

Mode Description
CLI (Host) graphaide extract, graphaide query, etc.
CLI (Docker) Run inside container via docker exec
Python API Import WorkflowManager for programmatic access
REST API FastAPI endpoints at http://localhost:8000

Getting Started:

  • Start with QUICKSTART.md for 5-minute setup
  • Run graphaide extract document.pdf to test locally
  • See tests/notebooks/demoGraphAide.ipynb for advanced examples

Support

Please reach out to Sumit.Purohit@pnnl.gov for any questions.

Authors and acknowledgment

Please reach out to Sumit.Purohit@pnnl.gov for any questions.

Citation

If you use GraphAide in your research, please cite:

@inproceedings{purohit2024graphaide,
  title={GraphAide: Advanced Graph-Assisted Query and Reasoning System},
  author={Purohit, Sumit and Chin, George and Mackey, Patrick S and Cottam, Joseph A},
  booktitle={2024 IEEE International Conference on Big Data (BigData)},
  pages={3485--3493},
  year={2024},
  organization={IEEE}
}

The research described in this paper is partially supported by the Resilience Through Data Driven, Intelligently Designed Control (RD2C) Initiative at Pacific Northwest National Laboratory (PNNL) and the United States federal government. Pacific Northwest National Laboratory is a multiprogram national laboratory operated for the US Department of Energy (DOE) by Battelle Memorial Institute under Contract No. DEAC05-76RL01830. PNNL Information Release PNNL-SA205147.

License

Please refer LICENSE and DISCLAIMER files for details.

Project status

Active development

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

graphaide-0.4.4.tar.gz (1.0 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

graphaide-0.4.4-py3-none-any.whl (121.4 kB view details)

Uploaded Python 3

File details

Details for the file graphaide-0.4.4.tar.gz.

File metadata

  • Download URL: graphaide-0.4.4.tar.gz
  • Upload date:
  • Size: 1.0 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.12.12

File hashes

Hashes for graphaide-0.4.4.tar.gz
Algorithm Hash digest
SHA256 9e5713e2c8d6ae10db376cd7554856bae7a7b2a906b17bce525f69804cebab86
MD5 a307eb2013012e680991d4b94eb3f552
BLAKE2b-256 6ab95ed60a266d65af13a31c66d73e1d63684a0b271b71d47a3dad75463d2759

See more details on using hashes here.

File details

Details for the file graphaide-0.4.4-py3-none-any.whl.

File metadata

  • Download URL: graphaide-0.4.4-py3-none-any.whl
  • Upload date:
  • Size: 121.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/7.0.0 CPython/3.12.12

File hashes

Hashes for graphaide-0.4.4-py3-none-any.whl
Algorithm Hash digest
SHA256 6bca494e6ce918935654a890f7f9fe5f9168d0e4757986a9a30cc611e23a9e48
MD5 9257cc2748514e302aee7cc54fbd250d
BLAKE2b-256 54a28cc09b251080351e0220dabb6683a421f056cda2963f4c96c2614e928c62

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page