LLMs for Graph Repair

Components

connect.py: Defines Graph class which manages connection to Neo4j Graph Database
dataset.py: Defines GraphDataset class, building on Graph, to provide functions for loading data

add_inconsistency_synthea.py: Add controlled inconsistencies to Synthea Dataset
dataset_synthea.py: Queries for Synthea Dataset
load_synthea.py: Load a dataset

graph.py: Extends networkx.DiGraph to define PropertyGraph class,
inconsistency.py: Find inconsistencies and store them in a pickle with PropertyGraph format
encoding.py: Provides functions for computing text representations of a PG
llm.py: Provides functions for connecting to LLMs and asking questions and getting answers
machine_repair.py: Ask LLM to repair the graph
response_statistics.py: Prepare response statistics

Load dataset using python3 load_synthea.py
Find inconsistencies using python3 inconsistency.py
Control repair parameters in machine_repair.py
Query LLMs for graph repair using python3 machine_repair.py
Prepare response statistics (generating tables, plots) using python3 response_statistics.py

This project is licensed under the terms of the GNU General Public License v3.0. See the LICENSE file for details.

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
encode_mode_cypher		encode_mode_cypher
encode_mode_graph		encode_mode_graph
encode_mode_llm_deepseek-r1		encode_mode_llm_deepseek-r1
encode_mode_llm_gemma2		encode_mode_llm_gemma2
encode_mode_llm_llama3.2		encode_mode_llm_llama3.2
encode_mode_llm_mistral		encode_mode_llm_mistral
encode_mode_llm_phi4		encode_mode_llm_phi4
encode_mode_llm_qwen2.5		encode_mode_llm_qwen2.5
encode_mode_template		encode_mode_template
example_mode_none		example_mode_none
example_mode_one_large		example_mode_one_large
example_mode_one_small		example_mode_one_small
example_mode_two_mix		example_mode_two_mix
example_mode_two_small		example_mode_two_small
inconsistency		inconsistency
LICENSE		LICENSE
README.md		README.md
add_inconsistency_synthea.py		add_inconsistency_synthea.py
connect.py		connect.py
dataset.py		dataset.py
dataset_synthea.py		dataset_synthea.py
encoding.py		encoding.py
graph.py		graph.py
inconsistencies.pkl		inconsistencies.pkl
inconsistency.py		inconsistency.py
llm.py		llm.py
llm_performance.png		llm_performance.png
load_synthea.py		load_synthea.py
machine_repair.py		machine_repair.py
repair_operations.png		repair_operations.png
requirements.txt		requirements.txt
response_statistics.py		response_statistics.py
rxnav_cache.sqlite		rxnav_cache.sqlite
utils.py		utils.py