jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: astradb ingests twice (#2573)

Browse source
This commit is contained in:
Nicolò Boschi 2024-07-08 13:21:34 +02:00 • committed by GitHub
commit 8339fcf0d8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/base/langflow/components/vectorstores/AstraDB.py
View file

@ -162,7 +162,8 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
), ),
] ]
def _build_vector_store_no_ingest(self): def _build_vector_store(self):
# cache the vector store to avoid re-initializing and ingest data again
if self._cached_vectorstore: if self._cached_vectorstore:
return self._cached_vectorstore return self._cached_vectorstore
@ -224,6 +225,8 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
except Exception as e: except Exception as e:
raise ValueError(f"Error initializing AstraDBVectorStore: {str(e)}") from e raise ValueError(f"Error initializing AstraDBVectorStore: {str(e)}") from e
self._add_documents_to_vector_store(vector_store)
self._cached_vectorstore = vector_store self._cached_vectorstore = vector_store
return vector_store return vector_store
@ -266,8 +269,7 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
return args return args
def search_documents(self) -> list[Data]: def search_documents(self) -> list[Data]:
vector_store = self._build_vector_store_no_ingest() vector_store = self._build_vector_store()
self._add_documents_to_vector_store(vector_store)
logger.debug(f"Search input: {self.search_input}") logger.debug(f"Search input: {self.search_input}")
logger.debug(f"Search type: {self.search_type}") logger.debug(f"Search type: {self.search_type}")
@ -300,6 +302,5 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
} }
def build_vector_store(self): def build_vector_store(self):
vector_store = self._build_vector_store_no_ingest() vector_store = self._build_vector_store()
self._add_documents_to_vector_store(vector_store)
return vector_store return vector_store