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

chore: Adjust FAISS Vector Store parameters format

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-22 13:52:41 -03:00
commit 84d4772d5e
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/components/vectorstores/FAISS.py
View file

@ -4,9 +4,8 @@ from langchain_community.vectorstores import FAISS
from loguru import logger
from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.field_typing import Text
from langflow.helpers.data import docs_to_data
from langflow.io import BoolInput, HandleInput, IntInput, StrInput, DataInput, MultilineInput
from langflow.io import BoolInput, DataInput, HandleInput, IntInput, MultilineInput, StrInput
from langflow.schema import Data
@ -74,7 +73,7 @@ class FaissVectorStoreComponent(LCVectorStoreComponent):
documents.append(_input)
faiss = FAISS.from_documents(documents=documents, embedding=self.embedding)
faiss.save_local(Text(path), self.index_name)
faiss.save_local(str(path), self.index_name)
return faiss
@ -87,7 +86,7 @@ class FaissVectorStoreComponent(LCVectorStoreComponent):
path = self.resolve_path(self.persist_directory)
vector_store = FAISS.load_local(
persist_directory=Text(path),
folder_path=path,
embeddings=self.embedding,
index_name=self.index_name,
allow_dangerous_deserialization=self.allow_dangerous_deserialization,