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

<chore>: adjustments on redis vector store

Browse source
This commit is contained in:
berrytern 2024-06-21 17:27:21 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit 75cb932e7b
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/components/vectorstores/Redis.py
View file

@ -4,7 +4,7 @@ from langchain_community.vectorstores.redis import Redis
from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.helpers.data import docs_to_data
from langflow.io import HandleInput, IntInput, StrInput, SecretStrInput, MultilineInput, DataInput
from langflow.io import HandleInput, IntInput, StrInput, SecretStrInput, DataInput, MultilineInput
from langflow.schema import Data
from langchain.text_splitter import CharacterTextSplitter
@ -28,7 +28,6 @@ class RedisVectorStoreComponent(LCVectorStoreComponent):
StrInput(
name="schema",
display_name="Schema",
field_type=[".yaml"],
),
DataInput(
name="vector_store_inputs",
@ -79,7 +78,7 @@ class RedisVectorStoreComponent(LCVectorStoreComponent):
return redis_vs
def search_documents(self) -> List[Data]:
vector_store = self._build_redis()
vector_store = self.build_vector_store()
if self.search_input and isinstance(self.search_input, str) and self.search_input.strip():
docs = vector_store.similarity_search(