jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

chore: Updating Supabase Vector Store parameters format

Browse source
This commit is contained in:
joaoguilhermeS 2024-06-22 17:24:36 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit 02c8251727
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py
View file

@ -12,7 +12,7 @@ from langflow.schema import Data
class SupabaseVectorStoreComponent(LCVectorStoreComponent): class SupabaseVectorStoreComponent(LCVectorStoreComponent):
display_name = "Supabase" display_name = "Supabase"
description = "Supabase Vector Store with search capabilities" 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" icon = "Supabase"
inputs = [ inputs = [
@ -20,13 +20,13 @@ class SupabaseVectorStoreComponent(LCVectorStoreComponent):
SecretStrInput(name="supabase_service_key", display_name="Supabase Service Key", required=True), SecretStrInput(name="supabase_service_key", display_name="Supabase Service Key", required=True),
StrInput(name="table_name", display_name="Table Name", advanced=True), StrInput(name="table_name", display_name="Table Name", advanced=True),
StrInput(name="query_name", display_name="Query Name"), 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( DataInput(
name="vector_store_inputs", name="ingest_data",
display_name="Vector Store Inputs", display_name="Ingest Data",
is_list=True, is_list=True,
), ),
MultilineInput(name="search_input", display_name="Search Input"), HandleInput(name="embedding", display_name="Embedding", input_types=["Embeddings"]),
IntInput( IntInput(
name="number_of_results", name="number_of_results",
display_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) supabase: Client = create_client(self.supabase_url, supabase_key=self.supabase_service_key)
documents = [] documents = []
for _input in self.vector_store_inputs or []: for _input in self.ingest_data or []:
if isinstance(_input, Data): if isinstance(_input, Data):
documents.append(_input.to_lc_document()) documents.append(_input.to_lc_document())
else: else:
@ -70,9 +70,9 @@ class SupabaseVectorStoreComponent(LCVectorStoreComponent):
def search_documents(self) -> List[Data]: def search_documents(self) -> List[Data]:
vector_store = self._build_supabase() 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( docs = vector_store.similarity_search(
query=self.search_input, query=self.search_query,
k=self.number_of_results, k=self.number_of_results,
) )