refactor: Remove unused build_base_retriever method from vector store components

src/backend/base/langflow/components/models/AmazonBedrockModel.py

@@ -78,7 +78,7 @@ class AmazonBedrockComponent(LCModelComponent):
         cache = self.cache
         stream = self.stream
         try:
-            output = ChatBedrock(
+            output = ChatBedrock(  # type: ignore
                 credentials_profile_name=credentials_profile_name,
                 model_id=model_id,
                 region_name=region_name,

src/backend/base/langflow/components/models/HuggingFaceModel.py

@@ -43,7 +43,7 @@ class HuggingFaceEndpointsComponent(LCModelComponent):
         model_kwargs = self.model_kwargs or {}
 
         try:
-            llm = HuggingFaceEndpoint(
+            llm = HuggingFaceEndpoint(  # type: ignore
                 endpoint_url=endpoint_url,
                 task=task,
                 huggingfacehub_api_token=huggingfacehub_api_token,

src/backend/base/langflow/components/models/VertexAiModel.py

@@ -1,5 +1,6 @@
 from langchain_google_vertexai import ChatVertexAI
 
+from langflow.base.constants import STREAM_INFO_TEXT
 from langflow.base.models.model import LCModelComponent
 from langflow.field_typing import LanguageModel
 from langflow.io import BoolInput, FileInput, FloatInput, IntInput, MessageInput, MultilineInput, Output, StrInput

src/backend/base/langflow/components/vectorstores/AstraDB.py

@@ -181,10 +181,6 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
         if self.add_to_vector_store:
             self._add_documents_to_vector_store(vector_store)
 
-        return vector_store
-
-    def build_base_retriever(self):
-        vector_store = self.build_vector_store()
         self.status = self._astradb_collection_to_data(vector_store.collection)
         return vector_store
 

src/backend/base/langflow/components/vectorstores/Cassandra.py

@@ -99,9 +99,6 @@ class CassandraVectorStoreComponent(Component):
     def build_vector_store(self) -> Cassandra:
         return self._build_cassandra()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_cassandra()
-
     def _build_cassandra(self) -> Cassandra:
         try:
             import cassio

src/backend/base/langflow/components/vectorstores/Couchbase.py

@@ -66,9 +66,6 @@ class CouchbaseVectorStoreComponent(Component):
     def build_vector_store(self) -> CouchbaseVectorStore:
         return self._build_couchbase()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_couchbase()
-
     def _build_couchbase(self) -> CouchbaseVectorStore:
         try:
             from couchbase.auth import PasswordAuthenticator  # type: ignore

src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py

@@ -63,9 +63,6 @@ class MongoVectorStoreComponent(Component):
     def build_vector_store(self) -> MongoDBAtlasVectorSearch:
         return self._build_mongodb_atlas()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_mongodb_atlas()
-
     def _build_mongodb_atlas(self) -> MongoDBAtlasVectorSearch:
         try:
             from pymongo import MongoClient

src/backend/base/langflow/components/vectorstores/Pinecone.py

@@ -67,10 +67,7 @@ class PineconeVectorStoreComponent(Component):
         Output(display_name="Search Results", name="search_results", method="search_documents"),
     ]
 
-    def build_vector_store(self) -> Pinecone:
-        return self._build_pinecone()
-
-    def build_base_retriever(self) -> Retriever:
+    def build_vector_store(self) -> Retriever:
         return self._build_pinecone()
 
     def _build_pinecone(self) -> Pinecone:

src/backend/base/langflow/components/vectorstores/Qdrant.py

@@ -71,9 +71,6 @@ class QdrantVectorStoreComponent(Component):
     def build_vector_store(self) -> Qdrant:
         return self._build_qdrant()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_qdrant()
-
     def _build_qdrant(self) -> Qdrant:
         qdrant_kwargs = {
             "collection_name": self.collection_name,

src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py

@@ -58,9 +58,6 @@ class SupabaseVectorStoreComponent(Component):
     def build_vector_store(self) -> SupabaseVectorStore:
         return self._build_supabase()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_supabase()
-
     def _build_supabase(self) -> SupabaseVectorStore:
         supabase: Client = create_client(self.supabase_url, supabase_key=self.supabase_service_key)
 

src/backend/base/langflow/components/vectorstores/Upstash.py

@@ -74,9 +74,6 @@ class UpstashVectorStoreComponent(Component):
     def build_vector_store(self) -> UpstashVectorStore:
         return self._build_upstash()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_upstash()
-
     def _build_upstash(self) -> UpstashVectorStore:
         use_upstash_embedding = self.embedding is None
 

src/backend/base/langflow/components/vectorstores/Weaviate.py

@@ -59,9 +59,6 @@ class WeaviateVectorStoreComponent(Component):
     def build_vector_store(self) -> Weaviate:
         return self._build_weaviate()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_weaviate()
-
     def _build_weaviate(self) -> Weaviate:
         if self.api_key:
             auth_config = weaviate.AuthApiKey(api_key=self.api_key)

src/backend/base/langflow/components/vectorstores/pgvector.py

@@ -56,9 +56,6 @@ class PGVectorStoreComponent(Component):
     def build_vector_store(self) -> PGVector:
         return self._build_pgvector()
 
-    def build_base_retriever(self) -> Retriever:
-        return self._build_pgvector()
-
     def _build_pgvector(self) -> PGVector:
         if self.add_to_vector_store:
             documents = []