From 3b1205c16361d4e3dff6ea22316a9e2692ab6e26 Mon Sep 17 00:00:00 2001 From: prestonrasmussen Date: Fri, 6 Dec 2024 15:33:06 -0500 Subject: [PATCH 1/2] update lucene escaping --- graphiti_core/helpers.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/graphiti_core/helpers.py b/graphiti_core/helpers.py index 5455f9ff..1e7bec50 100644 --- a/graphiti_core/helpers.py +++ b/graphiti_core/helpers.py @@ -57,6 +57,12 @@ def lucene_sanitize(query: str) -> str: ':': r'\:', '\\': r'\\', '/': r'\/', + 'O': r'\O', + 'R': r'\R', + 'N': r'\N', + 'T': r'\T', + 'A': r'\A', + 'D': r'\D', } ) From 8b8b9a462b23c35ebc7dd18166643f5f76702d19 Mon Sep 17 00:00:00 2001 From: prestonrasmussen Date: Mon, 9 Dec 2024 10:16:16 -0500 Subject: [PATCH 2/2] update unit test --- tests/helpers_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/helpers_test.py b/tests/helpers_test.py index 20029bc0..d709a9a4 100644 --- a/tests/helpers_test.py +++ b/tests/helpers_test.py @@ -23,10 +23,10 @@ def test_lucene_sanitize(): # Call the function with test data queries = [ ( - 'This has every secape character + - && || ! ( ) { } [ ] ^ " ~ * ? : \\ /', - 'This has every secape character \+ \- \&\& \|\| \! \( \) \{ \} \[ \] \^ \\" \~ \* \? \: \\\ \/', + 'This has every escape character + - && || ! ( ) { } [ ] ^ " ~ * ? : \\ /', + '\This has every escape character \+ \- \&\& \|\| \! \( \) \{ \} \[ \] \^ \\" \~ \* \? \: \\\ \/', ), - ('This has no escape characters', 'This has no escape characters'), + ('this has no escape characters', 'this has no escape characters'), ] for query, assert_result in queries: