fix-ruff

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
     

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 ''

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