GraphAide

Build knowledge graphs from any document with LLMs and Neo4j. GraphAide is a multi-agentic system that extracts entities and relationships from diverse sources, constructs knowledge graphs, and enables natural language reasoning over them—all with minimal setup.

30-Second Example

Simplest (Recommended)

from graphaide import GraphAide, ModelConfig, get_sample_data_path

# Setup (once) - uses default Neo4j at localhost:7687
graphaide = GraphAide(
    model_config=ModelConfig(
        provider="openai", 
        model_name="gpt-4o",
        api_key="sk-..."
    )
)

# Extract from sample document
sample_pdf = get_sample_data_path("PNNL_About.pdf")
sample_ontology = get_sample_data_path("ontology_cyber.nt")
result = graphaide.extract(file_path=sample_pdf, ontology_path=sample_ontology)
print(f"✓ Extracted {len(result.nodes)} entities, {len(result.edges)} relationships")

With Custom Neo4j

from graphaide import GraphAide, ModelConfig, GraphDBConfig, get_sample_data_path

graphaide = GraphAide(
    model_config=ModelConfig(provider="openai", model_name="gpt-4o", api_key="sk-..."),
    graphdb_config=GraphDBConfig(uri="neo4j://my-server:7687", username="neo4j", password="password"),
)

sample_pdf = get_sample_data_path("PNNL_About.pdf")
result = graphaide.extract(file_path=sample_pdf)
graphaide.ingest(file_path=sample_pdf)
answer = graphaide.query("What organizations are mentioned?")

What It Does

• Extract entities & relationships from PDFs, text, images, or web data
• Ground entities to Wikidata for semantic enrichment
• Load to Neo4j for queryable knowledge graphs
• Reason over graphs with natural language questions
• Track chunk provenance for result attribution

Supported Providers

Documentation

📚 Full documentation: https://pnnl.github.io/GraphAide

Quick links:

• Installation — Get started in minutes
• Configuration — Set up LLMs, databases, embeddings
• Extraction Guide — Learn entity/relationship extraction
• API Reference — Complete API documentation

Install

pip install graphaide

See QUICKSTART.md for detailed setup, Docker, and environment configuration.

Use Cases

• Knowledge Graph Generation — Extract structured data from documents into queryable graphs
• Semantic Search — Find related entities and concepts across datasets
• Domain-Specific Reasoning — Answer questions requiring cross-document reasoning
• Data Integration — Merge disparate sources with automated entity grounding
• Compliance & Audit — Track relationships and dependencies with provenance

Architecture

GraphAide uses a multi-agent architecture orchestrated by LangGraph:

Document → [FileLoader] → [Extractor] → [Nodes4EdgesLoader] 
         → [NodesLoader] → [EdgesLoader] → Neo4j
         ↓
      [VectorDB] ← [Embeddings] ← RAG Context

Factory pattern manages LLMs, vector stores, and graph databases:

• ModelFactory — Loads LLMs from OpenAI, Anthropic, Google, AWS Bedrock, or local LMStudio
• GraphDBFactory — Manages Neo4j/Neptune connections
• VectorStoreFactory — Manages ChromaDB or Qdrant vector indices
• AgentFactory — Builds extraction, loading, and query agents

Agents communicate via KGGenerationState (append-only shared state with reducers):

• raw_text, file_path — Input document
• nodes, edges — Extracted entities and relationships
• rag_context — Ontology constraints and Wikidata metadata
• messages — Status and debug info

Tested Configurations

Run Modes

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 & Legal

GraphAide is licensed under the BSD 2-Clause License (Simplified BSD).

Important: Please review the following legal documents:

• LICENSE.txt — Full license terms
• DISCLAIMER — Limitations and use restrictions
• ACKNOWLEDGMENT — Credits, funding, citations
• NOTICE — Software identification and legal notices

See the Legal section in documentation for more details.

Project status

Status: Active development
Version: 0.5.29
Last Updated: October 2026