🔧 (FAISS.py): Remove unused import and methods, refactor class name to improve consistency and clarity

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 20:23:01 -03:00
commit f26dd89dfa

View file

@ -1,17 +1,16 @@
from typing import List from typing import List
from langchain_community.vectorstores import FAISS from langchain_community.vectorstores import FAISS
from langchain_core.retrievers import BaseRetriever
from loguru import logger from loguru import logger
from langflow.custom import Component from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.field_typing import Text from langflow.field_typing import Text
from langflow.helpers.data import docs_to_data from langflow.helpers.data import docs_to_data
from langflow.io import BoolInput, HandleInput, IntInput, Output, StrInput from langflow.io import BoolInput, HandleInput, IntInput, Output, StrInput
from langflow.schema import Data from langflow.schema import Data
class FaissVectorStoreComponent(Component): class FaissVectorStoreComponent(LCVectorStoreComponent):
""" """
FAISS Vector Store with search capabilities FAISS Vector Store with search capabilities
""" """
@ -83,18 +82,6 @@ class FaissVectorStoreComponent(Component):
] ]
def build_vector_store(self) -> FAISS: def build_vector_store(self) -> FAISS:
"""
Builds the Vector Store object.
"""
return self._build_faiss()
def build_base_retriever(self) -> BaseRetriever:
"""
Builds the BaseRetriever object.
"""
return self._build_faiss()
def _build_faiss(self) -> FAISS:
""" """
Builds the FAISS object. Builds the FAISS object.
""" """