From 9f257f77eab80be60ea30a876414f6294287b5ed Mon Sep 17 00:00:00 2001 From: Amogh Tantradi Date: Tue, 14 Jan 2025 18:57:31 -0800 Subject: [PATCH] changed uuid type --- lotus/vector_store/weaviate_vs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lotus/vector_store/weaviate_vs.py b/lotus/vector_store/weaviate_vs.py index eab46d0..82fff0b 100644 --- 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)