feat: add BaseRetriever as base class for VectorStores

2023-06-07 19:45:15 -03:00 · 2023-06-07 19:45:15 -03:00 · 2ca750f654
commit 2ca750f654
parent af6a8b902a
1 changed files with 3 additions and 0 deletions
--- a/src/backend/langflow/template/frontend_node/vectorstores.py
+++ b/src/backend/langflow/template/frontend_node/vectorstores.py
@ -20,6 +20,9 @@ class VectorStoreFrontendNode(FrontendNode):

            self.template.add_field(extra_field)

+    def add_extra_base_classes(self) -> None:
+        self.base_classes.append("BaseRetriever")
+
    @staticmethod
    def format_field(field: TemplateField, name: Optional[str] = None) -> None:
        FrontendNode.format_field(field, name)