changed uuid type

guestrin-lab · Jan 15, 2025 · 9f257f7 · 9f257f7
1 parent c76b658
commit 9f257f7
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[uuid4]) -> NDArray[np.float64]:
+    def get_vectors_from_index(self, collection_name: str, ids: list[str]) -> NDArray[np.float64]:
         """Retrieve vectors for specific document IDs"""
         collection = self.client.collections.get(collection_name)