From 0d393fff74716687877b14fdf4873d5134b50a34 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Boschi?= Date: Tue, 13 Feb 2024 10:37:50 +0100 Subject: [PATCH] fixes --- .../e2e_tests/test_utils/vector_store_handler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py b/ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py index b14f9d815..a90ed58b7 100644 --- a/ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py +++ b/ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py @@ -30,7 +30,8 @@ class EnhancedLangChainVectorStore(LangChainVectorStore, EnhancedVectorStore, AB """Enhanced LangChain vector store""" -class EnhancedLlamaIndexVectorStore(LLamaIndexVectorStore, EnhancedVectorStore, ABC): +# we can't use the VectorStore type here from llama_index.vector_stores.types because AstraDBVectorStore is based on BasePydanticVectorStore +class EnhancedLlamaIndexVectorStore(EnhancedVectorStore, ABC): """Enhanced Llama-Index vector store"""