From 8f965c753d2ac72f871095d6e30b3b325c166f4a Mon Sep 17 00:00:00 2001
From: Daniel Chalef <131175+danielchalef@users.noreply.github.com>
Date: Mon, 25 Aug 2025 16:58:14 -0700
Subject: [PATCH] fix missing async in context handler

---
 graphiti_core/utils/maintenance/graph_data_operations.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/graphiti_core/utils/maintenance/graph_data_operations.py b/graphiti_core/utils/maintenance/graph_data_operations.py
index 9c1b5976..66dc55e4 100644
--- a/graphiti_core/utils/maintenance/graph_data_operations.py
+++ b/graphiti_core/utils/maintenance/graph_data_operations.py
@@ -89,7 +89,7 @@ async def build_indices_and_constraints(driver: GraphDriver, delete_existing: bo
 
 
 async def clear_data(driver: GraphDriver, group_ids: list[str] | None = None):
-    with driver.session() as session:
+    async with driver.session() as session:
 
         async def delete_all(tx):
             await tx.run('MATCH (n) DETACH DELETE n')