diff --git a/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py b/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py index f517411c1..4c9d89669 100644 --- a/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py +++ b/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py @@ -12,7 +12,7 @@ from langflow.schema import Data class SupabaseVectorStoreComponent(LCVectorStoreComponent): display_name = "Supabase" description = "Supabase Vector Store with search capabilities" - documentation = "https://python.langchain.com/docs/modules/data_connection/vectorstores/integrations/supabase" + documentation = "https://python.langchain.com/v0.2/docs/integrations/vectorstores/supabase/" icon = "Supabase" inputs = [ @@ -20,13 +20,13 @@ class SupabaseVectorStoreComponent(LCVectorStoreComponent): SecretStrInput(name="supabase_service_key", display_name="Supabase Service Key", required=True), StrInput(name="table_name", display_name="Table Name", advanced=True), StrInput(name="query_name", display_name="Query Name"), - HandleInput(name="embedding", display_name="Embedding", input_types=["Embeddings"]), + MultilineInput(name="search_query", display_name="Search Query"), DataInput( - name="vector_store_inputs", - display_name="Vector Store Inputs", + name="ingest_data", + display_name="Ingest Data", is_list=True, ), - MultilineInput(name="search_input", display_name="Search Input"), + HandleInput(name="embedding", display_name="Embedding", input_types=["Embeddings"]), IntInput( name="number_of_results", display_name="Number of Results", @@ -43,7 +43,7 @@ class SupabaseVectorStoreComponent(LCVectorStoreComponent): supabase: Client = create_client(self.supabase_url, supabase_key=self.supabase_service_key) documents = [] - for _input in self.vector_store_inputs or []: + for _input in self.ingest_data or []: if isinstance(_input, Data): documents.append(_input.to_lc_document()) else: @@ -70,9 +70,9 @@ class SupabaseVectorStoreComponent(LCVectorStoreComponent): def search_documents(self) -> List[Data]: vector_store = self._build_supabase() - if self.search_input and isinstance(self.search_input, str) and self.search_input.strip(): + if self.search_query and isinstance(self.search_query, str) and self.search_query.strip(): docs = vector_store.similarity_search( - query=self.search_input, + query=self.search_query, k=self.number_of_results, )