Refactor LCVectorStoreComponent to set status when building retriever

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 20:46:00 -03:00
commit cdfed9c3c2

View file

@ -79,7 +79,10 @@ class LCVectorStoreComponent(Component):
"""
vector_store = self.build_vector_store()
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:
raise ValueError(f"Vector Store {vector_store.__class__.__name__} does not have an as_retriever method.")