Refactor LCVectorStoreComponent to set status when building retriever
This commit is contained in:
parent
184a6d1342
commit
cdfed9c3c2
1 changed files with 4 additions and 1 deletions
|
|
@ -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.")
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue