gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update benchmark.py

Browse source
This commit is contained in:
hajdul88 2026-01-12 18:50:25 +01:00
parent 9d519d6421
commit 6992c639af
1 changed files with 1 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
age_adapter/benchmark.py
View file

@ -43,7 +43,7 @@ async def main():
await age_adapter.connect()
await neo4j_adapter.initialize()
batch_size = 3000
batch_size = 500
node_ids = [f"node_{i}" for i in range(batch_size)]
nodes = [(nid, ["TestNode"], {"name": f"Node {i}", "value": i})
for i, nid in enumerate(node_ids)]
@ -339,36 +339,6 @@ async def main():
print(f"Count Edges: AGE={age_time_count_edges:.4f}s, Neo4j={neo4j_time_count_edges:.4f}s")
start = time.perf_counter()
query = f"""
MATCH (a {{id: '{query_node_ids[0]}'}})-[*1..3]-(b {{id: '{query_node_ids[5]}'}})
RETURN a, b
LIMIT 1
"""
try:
await age_adapter.execute_cypher(query)
age_time_path_query = time.perf_counter() - start
except Exception as e:
age_time_path_query = time.perf_counter() - start
start = time.perf_counter()
try:
src_uuid = UUID(query_node_ids[0]) if '-' in query_node_ids[0] else UUID(int=hash(query_node_ids[0]) & ((1 << 128) - 1))
except:
src_uuid = UUID(int=hash(query_node_ids[0]) & ((1 << 128) - 1))
try:
tgt_uuid = UUID(query_node_ids[5]) if '-' in query_node_ids[5] else UUID(int=hash(query_node_ids[5]) & ((1 << 128) - 1))
except:
tgt_uuid = UUID(int=hash(query_node_ids[5]) & ((1 << 128) - 1))
query = f"""
MATCH (a:`__Node__`{{id: $src_id}})-[*1..3]-(b:`__Node__`{{id: $tgt_id}})
RETURN a, b
LIMIT 1
"""
await neo4j_adapter.query(query, {"src_id": str(src_uuid), "tgt_id": str(tgt_uuid)})
neo4j_time_path_query = time.perf_counter() - start
print(f"Path Query (1-3 hops): AGE={age_time_path_query:.4f}s, Neo4j={neo4j_time_path_query:.4f}s")
await age_adapter.close()
await neo4j_adapter.driver.close()