From cdfed9c3c2c4fffca516ea7f1e4886fa0a0f4e76 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 22 Jun 2024 20:46:00 -0300 Subject: [PATCH] Refactor LCVectorStoreComponent to set status when building retriever --- src/backend/base/langflow/base/vectorstores/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/base/vectorstores/model.py b/src/backend/base/langflow/base/vectorstores/model.py index e0e5bde91..2cc93cc60 100644 --- a/src/backend/base/langflow/base/vectorstores/model.py +++ b/src/backend/base/langflow/base/vectorstores/model.py @@ -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.")