21 lines
657 B
Python
21 lines
657 B
Python
#!/usr/bin/env python3
|
|
"""E2: Entity resolution diagnostic. Queries Graphiti's FalkorDB for the six test entities."""
|
|
import subprocess
|
|
import sys
|
|
|
|
TEST_ENTITIES = ["Aaron", "Kat", "HVAMC", "Bird", "Susan Hamlet", "Tulsa album"]
|
|
|
|
def run_cypher(query):
|
|
result = subprocess.run(
|
|
["docker", "exec", "falkordb", "redis-cli", "GRAPH.QUERY", "aaron", query],
|
|
capture_output=True, text=True
|
|
)
|
|
return result.stdout
|
|
|
|
for name in TEST_ENTITIES:
|
|
print(f"\n{'=' * 60}")
|
|
print(f"ENTITY: {name}")
|
|
print('=' * 60)
|
|
query = f"MATCH (n:Entity) WHERE n.name CONTAINS '{name}' RETURN n.name, n.summary"
|
|
print(run_cypher(query))
|