From 41666cf38e3f418368c451fd9eaa2d6b282c9c0e Mon Sep 17 00:00:00 2001
From: Gal Shubeli <galshubeli93@gmail.com>
Date: Thu, 14 Aug 2025 15:41:56 +0300
Subject: [PATCH] improve-delete

---
 graphiti_core/driver/falkordb_driver.py | 67 +++++++++++--------------
 1 file changed, 30 insertions(+), 37 deletions(-)

diff --git a/graphiti_core/driver/falkordb_driver.py b/graphiti_core/driver/falkordb_driver.py
index 1fc355fa..590f34ef 100644
--- a/graphiti_core/driver/falkordb_driver.py
+++ b/graphiti_core/driver/falkordb_driver.py
@@ -14,6 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
+import asyncio
 import logging
 from typing import TYPE_CHECKING, Any
 
@@ -191,46 +192,38 @@ class FalkorDriver(GraphDriver):
             await self.client.connection.close()
 
     async def delete_all_indexes(self) -> None:
-        from collections import defaultdict
-        
-        result = await self.execute_query('CALL db.indexes()')
-        if result is None:
+        result = await self.execute_query("CALL db.indexes()")
+        if not result:
             return
-        
-        records, _, _ = result
-        
-        # Organize indexes by type and label
-        range_indexes = defaultdict(list)
-        fulltext_indexes = defaultdict(list)
-        entity_types = {}
-        
-        for record in records:
-            label = record['label']
-            entity_types[label] = record['entitytype']
-            
-            for field_name, index_type in record['types'].items():
-                if 'RANGE' in index_type:
-                    range_indexes[label].append(field_name)
-                if 'FULLTEXT' in index_type:
-                    fulltext_indexes[label].append(field_name)
-        
-        # Drop all range indexes
-        for label, fields in range_indexes.items():
-            for field in fields:
-                await self.execute_query(f'DROP INDEX ON :{label}({field})')
 
-        # Drop all fulltext indexes
-        for label, fields in fulltext_indexes.items():
-            entity_type = entity_types[label]
-            for field in fields:
-                if entity_type == 'NODE':
-                    await self.execute_query(
-                        f'DROP FULLTEXT INDEX FOR (n:{label}) ON (n.{field})'
-                    )
-                elif entity_type == 'RELATIONSHIP':
-                    await self.execute_query(
-                        f'DROP FULLTEXT INDEX FOR ()-[e:{label}]-() ON (e.{field})'
+        records, _, _ = result
+        drop_tasks = []
+
+        for record in records:
+            label = record["label"]
+            entity_type = record["entitytype"]
+
+            for field_name, index_type in record["types"].items():
+                if "RANGE" in index_type:
+                    drop_tasks.append(
+                        self.execute_query(f"DROP INDEX ON :{label}({field_name})")
                     )
+                elif "FULLTEXT" in index_type:
+                    if entity_type == "NODE":
+                        drop_tasks.append(
+                            self.execute_query(
+                                f"DROP FULLTEXT INDEX FOR (n:{label}) ON (n.{field_name})"
+                            )
+                        )
+                    elif entity_type == "RELATIONSHIP":
+                        drop_tasks.append(
+                            self.execute_query(
+                                f"DROP FULLTEXT INDEX FOR ()-[e:{label}]-() ON (e.{field_name})"
+                            )
+                        )
+
+        if drop_tasks:
+            await asyncio.gather(*drop_tasks)
 
     def clone(self, database: str) -> 'GraphDriver':
         """