Small fixes

Author: mariusconjeaud

neomodel/async_/match.py

@@ -578,7 +578,7 @@ def build_vector_query(self):
             )
 
         if type(vector_filter.threshold) not in [float, type(None)]:
-            raise ValueError(f"Vector Filter Threshold must be a float or None.")
+            raise ValueError("Vector Filter Threshold must be a float or None.")
 
         vector_filter.index_name = f"vector_index_{source_class.__label__}_{vector_filter.vector_attribute_name}"
         vector_filter.node_set_label = source_class.__label__.lower()
@@ -587,7 +587,6 @@ def build_vector_query(self):
         self._ast.return_clause = f"{vector_filter.node_set_label}, score"
         self._ast.result_class = source_class.__class__
 
-
     def build_fulltext_query(self):
         """
         Query a free text indexed property on the node.
@@ -609,7 +608,7 @@ def build_fulltext_query(self):
             )
 
         if type(full_text_filter.threshold) not in [float, type(None)]:
-            raise ValueError(f"Full Text Filter Threshold must be a float or None.")
+            raise ValueError("Full Text Filter Threshold must be a float or None.")
 
         full_text_filter.index_name = f"fulltext_index_{source_class.__label__}_{full_text_filter.fulltext_attribute_name}"
         full_text_filter.node_set_label = source_class.__label__.lower()

neomodel/semantic_filters.py

@@ -1,5 +1,3 @@
-from typing import Union
-
 class VectorFilter(object):
     """
     Represents a CALL db.index.vector.query* neo functions call within the OGM
@@ -16,7 +14,11 @@ class VectorFilter(object):
     """
 
     def __init__(
-            self, topk: int, vector_attribute_name: str, candidate_vector: list[float], threshold: Union[float, None] =  None
+        self,
+        topk: int,
+        vector_attribute_name: str,
+        candidate_vector: list[float],
+        threshold: float | None = None,
     ):
         self.topk = topk
         self.vector_attribute_name = vector_attribute_name
@@ -25,9 +27,10 @@ def __init__(
         self.node_set_label = None
         self.vector = candidate_vector
 
+
 class FulltextFilter(object):
     """
-    Represents a CALL db.index.fulltext.query* neo functon call within the OGM.
+    Represents a CALL db.index.fulltext.query* neo function call within the OGM.
     :param query_strng: The string you are finding the nearest
     :type query_string: str
     :param freetext_attribute_name: The property name for the free text indexed property.
@@ -39,7 +42,12 @@ class FulltextFilter(object):
 
     """
 
-    def __init__(self, query_string: str, fulltext_attribute_name: str, topk: int, threshold: Union[float, None] =  None
+    def __init__(
+        self,
+        query_string: str,
+        fulltext_attribute_name: str,
+        topk: int,
+        threshold: float | None = None,
     ):
         self.topk = topk
         self.query_string = query_string

neomodel/sync_/match.py

@@ -576,7 +576,7 @@ def build_vector_query(self):
             )
 
         if type(vector_filter.threshold) not in [float, type(None)]:
-            raise ValueError(f"Vector Filter Threshold must be a float or None.")
+            raise ValueError("Vector Filter Threshold must be a float or None.")
 
         vector_filter.index_name = f"vector_index_{source_class.__label__}_{vector_filter.vector_attribute_name}"
         vector_filter.node_set_label = source_class.__label__.lower()
@@ -606,7 +606,7 @@ def build_fulltext_query(self):
             )
 
         if type(full_text_filter.threshold) not in [float, type(None)]:
-            raise ValueError(f"Full Text Filter Threshold must be a float or None.")
+            raise ValueError("Full Text Filter Threshold must be a float or None.")
 
         full_text_filter.index_name = f"fulltext_index_{source_class.__label__}_{full_text_filter.fulltext_attribute_name}"
         full_text_filter.node_set_label = source_class.__label__.lower()