Qdrant delete ID from training data minor mistake

heresandyboy · May 3, 2024 · fb6401e · fb6401e
1 parent 7b962d5
commit fb6401e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/vanna/qdrant/qdrant.py b/src/vanna/qdrant/qdrant.py
@@ -315,8 +315,8 @@ def _format_point_id(self, id: str, collection_name: str) -> str:
         return "{0}-{1}".format(id, ID_SUFFIXES[collection_name])
 
     def _parse_point_id(self, id: str) -> Tuple[str, str]:
-        id, suffix = id.rsplit("-", 1)
+        id, curr_suffix = id.rsplit("-", 1)
         for collection_name, suffix in ID_SUFFIXES.items():
-            if type == suffix:
+            if curr_suffix == suffix:
                 return id, collection_name
         raise ValueError(f"Invalid id {id}")