From b02462f0921794420564d0d861a15b5d963dac48 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 24 Jun 2024 17:15:16 -0300 Subject: [PATCH] fix: add call to _add_documents_to_vector_store in AstraDB component --- src/backend/base/langflow/components/vectorstores/AstraDB.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/backend/base/langflow/components/vectorstores/AstraDB.py b/src/backend/base/langflow/components/vectorstores/AstraDB.py index 3d146c510..aabbb91a7 100644 --- a/src/backend/base/langflow/components/vectorstores/AstraDB.py +++ b/src/backend/base/langflow/components/vectorstores/AstraDB.py @@ -196,6 +196,10 @@ class AstraVectorStoreComponent(LCVectorStoreComponent): except Exception as e: raise ValueError(f"Error initializing AstraDBVectorStore: {str(e)}") from e + if hasattr(self, "ingest_data") and self.ingest_data: + logger.debug("Ingesting data into the Vector Store.") + self._add_documents_to_vector_store(vector_store) + self.status = self._astradb_collection_to_data(vector_store.collection) return vector_store