From 50b3df03c4b36a2b1bd74481e1bd10be5f6be3a7 Mon Sep 17 00:00:00 2001
From: Preston Rasmussen <109292228+prasmussen15@users.noreply.github.com>
Date: Wed, 30 Apr 2025 15:00:29 -0400
Subject: [PATCH] Lucene sanitize (#423)

* lucene sanitize

* bump version
---
 graphiti_core/search/search_utils.py | 16 ++++++++++------
 pyproject.toml                       |  2 +-
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/graphiti_core/search/search_utils.py b/graphiti_core/search/search_utils.py
index ac7f8d1c..22baf72e 100644
--- a/graphiti_core/search/search_utils.py
+++ b/graphiti_core/search/search_utils.py
@@ -341,10 +341,10 @@ async def node_fulltext_search(
 
     query = (
         """
-                                CALL db.index.fulltext.queryNodes("node_name_and_summary", $query, {limit: $limit}) 
-                                YIELD node AS n, score
-                                WHERE n:Entity
-                                """
+                                    CALL db.index.fulltext.queryNodes("node_name_and_summary", $query, {limit: $limit}) 
+                                    YIELD node AS n, score
+                                    WHERE n:Entity
+                                    """
         + filter_query
         + ENTITY_NODE_RETURN
         + """
@@ -709,10 +709,14 @@ async def get_relevant_nodes(
         query,
         query_params,
         nodes=[
-            {'uuid': node.uuid, 'name': node.name, 'name_embedding': node.name_embedding}
+            {
+                'uuid': node.uuid,
+                'name': lucene_sanitize(node.name),
+                'name_embedding': node.name_embedding,
+            }
             for node in nodes
         ],
-        group_id=group_id,
+        group_id=lucene_sanitize(group_id),
         limit=limit,
         min_score=min_score,
         database_=DEFAULT_DATABASE,
diff --git a/pyproject.toml b/pyproject.toml
index a0807d65..8438837e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.11.0"
+version = "0.11.1"
 authors = [
     { "name" = "Paul Paliychuk", "email" = "paul@getzep.com" },
     { "name" = "Preston Rasmussen", "email" = "preston@getzep.com" },