graphaide 0.4.1__py3-none-any.whl

graphaide-0.4.1.dist-info/METADATA

@@ -0,0 +1,191 @@
+Metadata-Version: 2.4
+Name: graphaide
+Version: 0.4.1
+Summary: Multi-agentic knowledge graph construction and reasoning system
+Home-page: https://github.com/pnnl-int/GraphAide
+Author: Sumit Purohit
+Author-email: sumit.purohit@pnnl.gov
+License: MIT
+Project-URL: Source, https://github.com/pnnl-int/GraphAide
+Project-URL: Documentation, https://github.com/pnnl-int/GraphAide#quickstart
+Project-URL: Tracker, https://github.com/pnnl-int/GraphAide/issues
+Project-URL: License, https://github.com/pnnl-int/GraphAide/blob/main/LICENSE.txt
+Platform: any
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE.txt
+Requires-Dist: importlib-metadata==8.7.1
+Requires-Dist: openai==2.15.0
+Requires-Dist: langchain<1.2.0,>=1.0.7
+Requires-Dist: sentence_transformers==5.2.1
+Requires-Dist: langchain_community
+Requires-Dist: langchain_openai
+Requires-Dist: langchain-chroma
+Requires-Dist: langgraph==1.0.8
+Requires-Dist: langgraph-prebuilt==1.0.7
+Requires-Dist: langchain-aws
+Requires-Dist: botocore[crt]>=1.42.0
+Requires-Dist: boto3>=1.42.0
+Requires-Dist: langchain-anthropic
+Requires-Dist: langchain_google_genai
+Requires-Dist: langchain-mcp-adapters
+Requires-Dist: langchain-neo4j==0.8.0
+Requires-Dist: fastmcp==2.14.5
+Requires-Dist: neo4j==6.1.0
+Requires-Dist: ragas==0.4.3
+Requires-Dist: pathlib==1.0.1
+Requires-Dist: pypdf==6.6.2
+Requires-Dist: pydantic==2.12.5
+Requires-Dist: rdflib>=6.0.0
+Requires-Dist: numba==0.61.2
+Requires-Dist: umap-learn==0.5.11
+Requires-Dist: plotly==6.5.2
+Requires-Dist: kaleido==0.2.1
+Requires-Dist: pandas==3.0.0
+Requires-Dist: numpy==2.2.6
+Requires-Dist: matplotlib==3.10.8
+Requires-Dist: dash==3.4.0
+Requires-Dist: py2neo==2021.2.4
+Requires-Dist: autogen-agentchat==0.7.5
+Requires-Dist: autogen-ext[openai]==0.7.5
+Requires-Dist: python-dotenv==1.2.1
+Requires-Dist: ipykernel==7.1.0
+Requires-Dist: jupyterlab==4.5.3
+Requires-Dist: webencodings==0.5.1
+Requires-Dist: langchain-text-splitters==1.1.0
+Requires-Dist: typer[all]>=0.9.0
+Requires-Dist: fastapi>=0.110.0
+Requires-Dist: uvicorn[standard]>=0.29.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: qdrant-client<=1.18.0,>=1.18.0
+Requires-Dist: langchain-qdrant>=0.1.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pdf2image>=1.16.0
+Provides-Extra: testing
+Requires-Dist: setuptools; extra == "testing"
+Requires-Dist: pytest; extra == "testing"
+Requires-Dist: pytest-cov; extra == "testing"
+Dynamic: license-file
+
+# GraphAide
+
+<p align="center">
+  <img src="tests/notebooks/GraphAideLogo_Label_Final.png" alt="GraphAide Logo" width="400">
+</p>
+
+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](QUICKSTART.md)** - Installation, setup, and first examples
+- **[ARCHITECTURE.md](ARCHITECTURE.md)** - Deep technical overview (factories, agents, state, workflows)
+- **[REFERENCE.md](REFERENCE.md)** - Complete CLI/API reference and feature matrix
+- **[CLAUDE.md](CLAUDE.md)** - Guide for Claude AI Code (developers only)
+
+## Installation
+
+```bash
+# From project root
+pip install .
+
+# With Docker
+docker compose up -d
+```
+
+## Deployment Scripts
+
+### Linux/Mac/WSL
+```bash
+./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)
+```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](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:
+
+```bibtex
+@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

graphaide-0.4.1.dist-info/RECORD

@@ -0,0 +1,57 @@
+graphaide-0.4.1.dist-info/licenses/LICENSE.txt,sha256=XizNJii1UyZPqD_WoPhuwJLmiYj4VN21wQZwtic9xFQ,1309
+graphgen/__init__.py,sha256=1LPGNCSfVziFCQWdmP-b-RC6jAAsMnq01DD_6TTX_sE,580
+graphgen/api.py,sha256=5c8LLKVD0VwTzje-OgcvwPYl1Ibx5vggcRoBHAe8eXo,15973
+graphgen/cli.py,sha256=0z5foIoFVgTtBBhYZHFSgH4A31w0Xzzp3H-4Seyc7uE,28390
+graphgen/core.py,sha256=yheFcQczpnze6f1IMCuN-gD6Uq63zqEfb0b77UFDNMs,27666
+graphgen/v2/GraphDBManager.py,sha256=quHwfUr5m2q-zbtWTk7E9Q_hfS3PSDeZwKWrcChXnIM,5305
+graphgen/v2/ModelManager.py,sha256=1FwnYaXj6ot8gI7yZuCYQBbX9Ipw6xO0jj0Kz59u9x4,10606
+graphgen/v2/VectorStoreManager.py,sha256=lAFAKlH203jpO4gcnPfm_ysTGtpF_7Ktql59H67ZZQ8,6906
+graphgen/v2/__init__.py,sha256=2m46cz8aZsaEy7kxDiY3DUr47rwEGeSyoDV9SHyIEPk,77
+graphgen/v2/functionnode.py,sha256=vuSpeZSohru_WBp0MEwT_Hc7oAuofaAuUynBUa7LIIM,9805
+graphgen/v2/globalconfig.py,sha256=4_6lWGCHv-CJdiyivB5DVp1kRPwP_eqFlP269OuUBvg,1856
+graphgen/v2/mcp_server.py,sha256=tHt5vnqvEjoXhLMzycn4Ft8wWGimq31TD8JpvhqMLHM,1205
+graphgen/v2/query.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+graphgen/v2/state.py,sha256=8Wqj2BDGhod-37YdI2L8u3Y98LQKAW6d3MQL-tGNF-c,18997
+graphgen/v2/tools.py,sha256=_oSC5XMelmBO08hx0fWuvjN2IBR3euXnhSxMonKDYDk,11403
+graphgen/v2/utils.py,sha256=5PXBfmoCf215sMXZ2PvrL070DZvyjEUoGcrAinzqhlU,5750
+graphgen/v2/agents/AgentManager.py,sha256=-L_UxrmUZg810aeyYkQW6xQJa5f-5WOEB7Qjvz5DfPw,7838
+graphgen/v2/agents/__init__.py,sha256=uW982BAP7E5gBikOBIvZ3B0imrOqJfAGDKxxRtPgn3A,4398
+graphgen/v2/agents/base.py,sha256=dMcp5A_mDxqUA1KC5WjkeeZxq8C6KcItQ9a8d8rnb2s,5153
+graphgen/v2/agents/edgesloader.py,sha256=HL_PBnbl09AJ8saDQviJ0YFKc1BCr-ZNK2dfOIa4mAc,5401
+graphgen/v2/agents/edgetable2textgenerator.py,sha256=AGTGerYtRTXojPNNe5leVy5lOn8D0pzmjpDyfTuszW4,2398
+graphgen/v2/agents/embeddingprojector.py,sha256=byRnPHLXRh1-YxBcUKisBoG9OtP5aaqRdgEqcBSMvf8,5332
+graphgen/v2/agents/embeddingvisualizer.py,sha256=QDdMP3_jcj9nCJLYU5vDoEsAf-3Qg0NrU5aHqBKfuU8,8975
+graphgen/v2/agents/extractmerger.py,sha256=JJvBJ2DjBl7GHCQLWZUMy9lvuS1WL9EXA9WGnXgz9Y0,7478
+graphgen/v2/agents/extractor.py,sha256=9Sxc-o1VlibkKhuavsr26XXlJMOj_QR0EI0M4gOolFw,21854
+graphgen/v2/agents/intentpredictor.py,sha256=uYx57hzCt1MYYdqovanvGoRp0GKj7zF_Aq-zPUPEAO0,1898
+graphgen/v2/agents/neo4jloader.py,sha256=rO5EqDUP8HXDFp5fD4ZrkxJVpoGx0g9vSJT0sczQ7e4,4398
+graphgen/v2/agents/neo4jpostprocessor.py,sha256=GS-2CjdB23TrgV1acwvpAAQOOM-UIHXHFO-xUpkjQrg,7614
+graphgen/v2/agents/neo4jqueryrunner.py,sha256=Z8YIBUMA4-cgWUTk0j3GUycU7WdUYFGiowmwYz0wOEA,3134
+graphgen/v2/agents/nodes4edgesloader.py,sha256=rW3znLSkDep8NZhoFG-O641JLblefDjNSsdnQ9sbsw0,6054
+graphgen/v2/agents/nodesloader.py,sha256=H9yg-DKeu94peEOJUsiDPIbw9RRCHacrqjfrOQJXlak,5076
+graphgen/v2/agents/ontologygenerator.py,sha256=-zH0fednMI3hmbKdUnNp6IGUW1iZ9UdBXXCc-2c4pvY,9781
+graphgen/v2/agents/ontologyloader.py,sha256=AoatH-_Nw30kcttZ7840uJL81Um7WrwmJzKr-L0cvLc,6487
+graphgen/v2/agents/questiongraphgenerator.py,sha256=UwbVzvfnVApLnCX1XV1yipg1Wnvc1EuZ4T3j3eUuzps,1887
+graphgen/v2/agents/simpleagent.py,sha256=07k5VK8CetdEtsdMSOVMgTsew5QX_5MkE52cdQWxOMc,1896
+graphgen/v2/agents/simplerag.py,sha256=KHWp3P0TWMncM8fyMigcRimQGa4XwDAPC3bloXvfPoA,3904
+graphgen/v2/agents/textaugmentor.py,sha256=NoenlYCgwNXQBDO4Q_GxNS7oMmNJl-TvXPBts30hlH8,2215
+graphgen/v2/agents/vectordbloader.py,sha256=B9STfAMljjrl-pEa_mfrWs36PC6iMPOWf_VcfB5tr0Q,62970
+graphgen/v2/docs/GA_VDB/chroma.sqlite3,sha256=WxVim5siFJrVHiyL-xz8tLaya2hCeg0KSRj6ANljZJY,196608
+graphgen/v2/docs/GA_VDB/6fc79c82-1c20-4a02-bb10-71dcaf3a17d7/data_level0.bin,sha256=ErXt3nOYJ48PFYpTXpJnKd6H7G4ye6Yk_H8T2sPRVhA,423600
+graphgen/v2/docs/GA_VDB/6fc79c82-1c20-4a02-bb10-71dcaf3a17d7/header.bin,sha256=vxLUSGUYx63fSIy0hUUmkCx46RlRmQ4eL04FXOyBTl0,100
+graphgen/v2/docs/GA_VDB/6fc79c82-1c20-4a02-bb10-71dcaf3a17d7/length.bin,sha256=2tsvOqjW_e4v5cUPpXVS7QgK-fXpzbl8gUVI1zAmt8I,400
+graphgen/v2/docs/GA_VDB/6fc79c82-1c20-4a02-bb10-71dcaf3a17d7/link_lists.bin,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+graphgen/v2/utils/__init__.py,sha256=fnHehIMDRQhKB3lbGwH13HruuDMTMUJuxoBiBOKUX-M,7835
+graphgen/v2/utils/image_utils.py,sha256=gvS9cwmLNE2wh2RU6iaDH-fv3XT26gHGrUTl4lGVF0k,5421
+graphgen/v2/workflows/KGIngest.py,sha256=dgOyCzQyJaOhoEoOTyNi4sKn6hV2ciYK7kA1mRsqL0I,3654
+graphgen/v2/workflows/__init__.py,sha256=rhoo2VttkXPANeFv_vdMLG_Zk5cJvCy74iETz3iBkak,1987
+graphgen/v2/workflows/base.py,sha256=3-Bst-0XfsLe02DvxqmEn7PDiCc3Iebm_i8R1DTQ_aU,8680
+graphgen/v2/workflows/manager.py,sha256=XnhFQ7dFt2vMJ794aMaGapFmPfdGNfNbHTeyEUvQqmQ,51308
+graphgen/v2/workflows/retriever_factory.py,sha256=h_DLVQtvSfToIe1oo158bbdGkWQsPS6yh299anrVBUw,9516
+graphgen/v2/workflows/schemas.py,sha256=ODLegGebQS06l63yvcnoDQDX-8GfvDGd-KfVsFiV18o,16760
+graphgen/v2/workflows/utils.py,sha256=8Vx7nlnQDeHjV5MKbdHYV_zHUh7CF2y1tUpRnL_mWe8,8879
+graphaide-0.4.1.dist-info/METADATA,sha256=AEd8M3TJbCRHUOUIy9vz4H36qgiUDWXK7ilqbxSVCRI,7473
+graphaide-0.4.1.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+graphaide-0.4.1.dist-info/entry_points.txt,sha256=nk2ho92XQRjingyCWGkzNW1r9Ka_EzznmvHbmyQF1fI,48
+graphaide-0.4.1.dist-info/top_level.txt,sha256=z2O0X06I0LLZ5qPMl9yvMh3KoPdmwIgqYvXoJ8aZ_1U,9
+graphaide-0.4.1.dist-info/RECORD,,

graphaide-0.4.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

graphaide-0.4.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+graphaide = graphgen.cli:main

graphaide-0.4.1.dist-info/licenses/LICENSE.txt

@@ -0,0 +1,23 @@
+LICENSE
+Copyright Battelle Memorial Institute 2026
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+this list of conditions and the following disclaimer in the documentation
+and/or other materials provided with the distribution.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

graphaide-0.4.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+graphgen

graphgen/__init__.py

@@ -0,0 +1,16 @@
+import sys
+
+if sys.version_info[:2] >= (3, 8):
+    # TODO: Import directly (no need for conditional) when `python_requires = >= 3.8`
+    from importlib.metadata import PackageNotFoundError, version  # pragma: no cover
+else:
+    from importlib_metadata import PackageNotFoundError, version  # pragma: no cover
+
+try:
+    # Change here if project is renamed and does not equal the package name
+    dist_name = "GraphAide"
+    __version__ = version(dist_name)
+except PackageNotFoundError:  # pragma: no cover
+    __version__ = "unknown"
+finally:
+    del version, PackageNotFoundError