From fe9918cb8cf6c6b1cfdc8931a187efb24942e75f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Jun 2023 21:01:25 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(vectorstores.py):=20add=20s?= =?UTF-8?q?upport=20for=20SupabaseVectorStore=20type=20The=20SupabaseVecto?= =?UTF-8?q?rStore=20type=20requires=20additional=20fields=20such=20as=20ta?= =?UTF-8?q?ble=5Fname,=20query=5Fname,=20supabase=5Furl,=20and=20supabase?= =?UTF-8?q?=5Fservice=5Fkey.=20These=20fields=20are=20added=20to=20the=20e?= =?UTF-8?q?xtra=5Ffields=20list=20and=20are=20then=20added=20to=20the=20li?= =?UTF-8?q?st=20of=20fields=20that=20are=20displayed=20to=20the=20user.=20?= =?UTF-8?q?This=20change=20allows=20the=20user=20to=20create=20a=20Supabas?= =?UTF-8?q?eVectorStore=20with=20the=20required=20fields.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../template/frontend_node/vectorstores.py | 48 +++++++++++++++++++ 1 file changed, 48 insertions(+) diff --git a/src/backend/langflow/template/frontend_node/vectorstores.py b/src/backend/langflow/template/frontend_node/vectorstores.py index ce2c35434..0fcb76975 100644 --- a/src/backend/langflow/template/frontend_node/vectorstores.py +++ b/src/backend/langflow/template/frontend_node/vectorstores.py @@ -88,6 +88,50 @@ class VectorStoreFrontendNode(FrontendNode): display_name="Index Name", ) extra_fields.extend((extra_field, extra_field2)) + elif self.template.type_name == "SupabaseVectorStore": + # Add table_name and query_name + extra_field = TemplateField( + name="table_name", + field_type="str", + required=False, + placeholder="", + show=True, + advanced=True, + multiline=False, + value="", + ) + extra_field2 = TemplateField( + name="query_name", + field_type="str", + required=False, + placeholder="", + show=True, + advanced=True, + multiline=False, + value="", + ) + # Add supabase_url and supabase_service_key + extra_field3 = TemplateField( + name="supabase_url", + field_type="str", + required=False, + placeholder="", + show=True, + advanced=True, + multiline=False, + value="", + ) + extra_field4 = TemplateField( + name="supabase_service_key", + field_type="str", + required=False, + placeholder="", + show=True, + advanced=True, + multiline=False, + value="", + ) + extra_fields.extend((extra_field, extra_field2, extra_field3, extra_field4)) if extra_fields: for field in extra_fields: @@ -111,6 +155,10 @@ class VectorStoreFrontendNode(FrontendNode): "index_name", "namespace", "folder_path", + "table_name", + "query_name", + "supabase_url", + "supabase_service_key", ] advanced_fields = [ "n_dim",