From e79022c71f89909e9b29ca069db01f0920f706b3 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Tue, 16 Jan 2024 16:34:01 -0300 Subject: [PATCH] Update SupabaseVectorStore implementation --- .../vectorstores/SupabaseVectorStore.py | 20 ++++++++----------- 1 file changed, 8 insertions(+), 12 deletions(-) diff --git a/src/backend/langflow/components/vectorstores/SupabaseVectorStore.py b/src/backend/langflow/components/vectorstores/SupabaseVectorStore.py index e875fc06e..d8108250f 100644 --- a/src/backend/langflow/components/vectorstores/SupabaseVectorStore.py +++ b/src/backend/langflow/components/vectorstores/SupabaseVectorStore.py @@ -1,11 +1,14 @@ from langflow import CustomComponent -from typing import Optional, List -from langchain.vectorstores import SupabaseVectorStore +from typing import Optional, List,Union +from langchain_community.vectorstores.supabase import SupabaseVectorStore from langflow.field_typing import ( Document, Embeddings, NestedDict, ) +from langchain.schema import BaseRetriever +from langchain.vectorstores.base import VectorStore +from supabase.client import Client, create_client class SupabaseComponent(CustomComponent): @@ -32,13 +35,6 @@ class SupabaseComponent(CustomComponent): supabase_service_key: str = "", supabase_url: str = "", table_name: str = "", - ) -> SupabaseVectorStore: - return SupabaseVectorStore( - documents=documents, - embedding=embedding, - query_name=query_name, - search_kwargs=search_kwargs, - supabase_service_key=supabase_service_key, - supabase_url=supabase_url, - table_name=table_name, - ) + ) -> Union[VectorStore,SupabaseVectorStore,BaseRetriever]: + supabase: Client = create_client(supabase_url, supabase_key=supabase_service_key) + return SupabaseVectorStore.from_documents(documents=documents,embedding=embedding,query_name=query_name,search_kwargs=search_kwargs,client=supabase,table_name=table_name) \ No newline at end of file