From c5f25a32fa18bf5d719bdd0d40cd573b275c8566 Mon Sep 17 00:00:00 2001
From: Gal Shubeli <galshubeli93@gmail.com>
Date: Wed, 17 Sep 2025 15:23:44 +0300
Subject: [PATCH] fix-ruff

---
 graphiti_core/driver/falkordb_driver.py | 14 --------------
 graphiti_core/search/search_utils.py    |  2 +-
 tests/helpers_test.py                   |  7 ++++++-
 3 files changed, 7 insertions(+), 16 deletions(-)

diff --git a/graphiti_core/driver/falkordb_driver.py b/graphiti_core/driver/falkordb_driver.py
index 057da1c5..87db533d 100644
--- a/graphiti_core/driver/falkordb_driver.py
+++ b/graphiti_core/driver/falkordb_driver.py
@@ -245,18 +245,4 @@ class FalkorDriver(GraphDriver):
         full_query = group_filter + ' (' + sanitized_query + ')'
 
         return full_query
-
-
-
-def convert_datetimes_to_strings(obj):
-    if isinstance(obj, dict):
-        return {k: convert_datetimes_to_strings(v) for k, v in obj.items()}
-    elif isinstance(obj, list):
-        return [convert_datetimes_to_strings(item) for item in obj]
-    elif isinstance(obj, tuple):
-        return tuple(convert_datetimes_to_strings(item) for item in obj)
-    elif isinstance(obj, datetime):
-        return obj.isoformat()
-    else:
-        return obj
     
\ No newline at end of file
diff --git a/graphiti_core/search/search_utils.py b/graphiti_core/search/search_utils.py
index 18ee57ce..87596d7c 100644
--- a/graphiti_core/search/search_utils.py
+++ b/graphiti_core/search/search_utils.py
@@ -102,7 +102,7 @@ def fulltext_query(query: str, group_ids: list[str] | None, driver: GraphDriver)
 
     group_ids_filter += ' AND ' if group_ids_filter else ''
 
-    lucene_query = lucene_sanitize(query)
+    lucene_query = driver.sanitize(query)
     # If the lucene query is too long return no query
     if len(lucene_query.split(' ')) + len(group_ids or '') >= MAX_QUERY_LENGTH:
         return ''
diff --git a/tests/helpers_test.py b/tests/helpers_test.py
index 024729e7..01f509e0 100644
--- a/tests/helpers_test.py
+++ b/tests/helpers_test.py
@@ -174,6 +174,11 @@ def mock_embedder():
 
 
 def test_lucene_sanitize():
+    from graphiti_core.driver.neo4j_driver import Neo4jDriver
+    
+    # Create a driver instance to test the sanitize method
+    driver = Neo4jDriver(uri="bolt://localhost:7687", user="neo4j", password="password")
+    
     # Call the function with test data
     queries = [
         (
@@ -184,7 +189,7 @@ def test_lucene_sanitize():
     ]
 
     for query, assert_result in queries:
-        result = lucene_sanitize(query)
+        result = driver.sanitize(query)
         assert assert_result == result