Update db.py

datastax · Nov 10, 2023 · 52cdb72 · 52cdb72
1 parent 28527cd
commit 52cdb72
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/astrapy/db.py b/astrapy/db.py
@@ -81,6 +81,10 @@ def _pre_process_find(self, vector, fields=None):
         if not vector:
             return ValueError("Must pass a vector")
 
+        # Edge case for field selection
+        if fields and "$similarity" in fields:
+            raise ValueError("Please use the `include_similarity` parameter")
+
         # Build the new vector parameter
         sort = {"$vector": vector}
 
@@ -342,10 +346,6 @@ def vector_find_one(
         fields=None,
         include_similarity=True,
     ):
-        # Edge case for field selection
-        if fields and "$similarity" in fields:
-            raise ValueError("Please use the `include_similarity` parameter")
-
         # Pre-process the included arguments
         sort, projection = self._pre_process_find(
             vector,