edit uuid type

lotus-data · Jan 15, 2025 · c76b658 · c76b658
1 parent 1357fb3
commit c76b658
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lotus/vector_store/weaviate_vs.py b/lotus/vector_store/weaviate_vs.py
@@ -133,7 +133,7 @@ def __call__(self,
             indices=np.array(all_indices, dtype=np.int64).tolist()
         )
 
-    def get_vectors_from_index(self, collection_name: str, ids: list[int]) -> NDArray[np.float64]:
+    def get_vectors_from_index(self, collection_name: str, ids: list[uuid4]) -> NDArray[np.float64]:
         """Retrieve vectors for specific document IDs"""
         collection = self.client.collections.get(collection_name)