refactor: Remove unused build_base_retriever method from vector store components

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:09:18 -03:00
commit 75836a478c
13 changed files with 4 additions and 34 deletions

View file

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

View file

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

View file

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

View file

@ -181,10 +181,6 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
if self.add_to_vector_store: if self.add_to_vector_store:
self._add_documents_to_vector_store(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) self.status = self._astradb_collection_to_data(vector_store.collection)
return vector_store return vector_store

View file

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

View file

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

View file

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

View file

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

View file

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

View file

@ -58,9 +58,6 @@ class SupabaseVectorStoreComponent(Component):
def build_vector_store(self) -> SupabaseVectorStore: def build_vector_store(self) -> SupabaseVectorStore:
return self._build_supabase() return self._build_supabase()
def build_base_retriever(self) -> Retriever:
return self._build_supabase()
def _build_supabase(self) -> SupabaseVectorStore: def _build_supabase(self) -> SupabaseVectorStore:
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)

View file

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

View file

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

View file

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