fixes

datastax · Feb 13, 2024 · 0d393ff · 0d393ff
1 parent 0560dc4
commit 0d393ff
Showing 1 changed file with 2 additions and 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
@@ -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"""