From a45947c01beb0da9c91e410e4122ca0882bca30b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 25 Jun 2023 18:37:05 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(vectorstores.py):=20add=20V?= =?UTF-8?q?ectorStoreRetriever=20as=20an=20extra=20base=20class=20to=20Vec?= =?UTF-8?q?torStoreFrontendNode=20The=20VectorStoreFrontendNode=20class=20?= =?UTF-8?q?now=20has=20VectorStoreRetriever=20as=20an=20extra=20base=20cla?= =?UTF-8?q?ss=20in=20addition=20to=20BaseRetriever.=20This=20change=20was?= =?UTF-8?q?=20made=20to=20improve=20the=20functionality=20of=20the=20class?= =?UTF-8?q?=20by=20allowing=20it=20to=20inherit=20from=20VectorStoreRetrie?= =?UTF-8?q?ver.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/template/frontend_node/vectorstores.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/template/frontend_node/vectorstores.py b/src/backend/langflow/template/frontend_node/vectorstores.py index 01b6bfe53..58cfcdc34 100644 --- a/src/backend/langflow/template/frontend_node/vectorstores.py +++ b/src/backend/langflow/template/frontend_node/vectorstores.py @@ -200,7 +200,7 @@ class VectorStoreFrontendNode(FrontendNode): self.template.add_field(field) def add_extra_base_classes(self) -> None: - self.base_classes.append("BaseRetriever") + self.base_classes.extend(("BaseRetriever", "VectorStoreRetriever")) @staticmethod def format_field(field: TemplateField, name: Optional[str] = None) -> None: