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

Refactor LCVectorStoreComponent to set status when building retriever

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 20:46:00 -03:00
commit cdfed9c3c2
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/base/vectorstores/model.py
View file

@ -79,7 +79,10 @@ class LCVectorStoreComponent(Component):
""" """
vector_store = self.build_vector_store() vector_store = self.build_vector_store()
if hasattr(vector_store, "as_retriever"): if hasattr(vector_store, "as_retriever"):
return vector_store.as_retriever() retriever = vector_store.as_retriever()
if self.status is None:
self.status = "Retriever built successfully."
return retriever
else: else:
raise ValueError(f"Vector Store {vector_store.__class__.__name__} does not have an as_retriever method.") raise ValueError(f"Vector Store {vector_store.__class__.__name__} does not have an as_retriever method.")