Spaces:
Running
Running
File size: 2,611 Bytes
33516f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | import os
import logging
from dotenv import load_dotenv
from langchain_neo4j import Neo4jGraph
# Load environment variables
load_dotenv()
# Configure logging
logging.basicConfig(level=logging.INFO, format='%(message)s')
logger = logging.getLogger(__name__)
def inspect_graph():
from src.config import NEO4J_URI, NEO4J_USERNAME, NEO4J_PASSWORD
if not all([NEO4J_URI, NEO4J_USERNAME, NEO4J_PASSWORD]):
logger.error("Missing Neo4j configuration.")
return
try:
graph = Neo4jGraph(
url=NEO4J_URI,
username=NEO4J_USERNAME,
password=NEO4J_PASSWORD
)
# Refresh schema to ensure we have the latest view
graph.refresh_schema()
print("\n=== Neo4j Graph Summary ===\n")
# 1. Schema Info
print("--- Node Labels & Relationships ---")
# graph.schema is a string representation of the schema
print(graph.schema)
# 2. Counts
print("\n--- Counts ---")
count_query = """
MATCH (n)
OPTIONAL MATCH (n)-[r]->()
RETURN count(distinct n) as nodes, count(r) as relationships
"""
counts = graph.query(count_query)
if counts:
print(f"Nodes: {counts[0]['nodes']}")
print(f"Relationships: {counts[0]['relationships']}")
else:
print("No data found.")
# 3. Top Connected Entities
print("\n--- Top 5 Most Connected Entities ---")
top_nodes_query = """
MATCH (n)-[r]-()
RETURN n.id as id, labels(n)[0] as type, count(r) as degree
ORDER BY degree DESC
LIMIT 5
"""
top_nodes = graph.query(top_nodes_query)
if top_nodes:
for node in top_nodes:
node_id = node.get('id', 'Unknown ID')
node_type = node.get('type', 'Unknown Type')
degree = node.get('degree', 0)
print(f"{node_id} ({node_type}): {degree} connections")
else:
print("No connected nodes found.")
# 4. Sample Relationship
print("\n--- Sample Relationship ---")
sample_query = """
MATCH (a)-[r]->(b)
RETURN a.id, type(r), b.id
LIMIT 1
"""
sample = graph.query(sample_query)
if sample:
s = sample[0]
print(f"{s['a.id']} --[{s['type(r)']}]--> {s['b.id']}")
else:
print("No relationships found.")
except Exception as e:
logger.error(f"Failed to inspect graph: {e}")
if __name__ == "__main__":
inspect_graph()
|