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

refactor: Update RedisVectorStoreComponent to use VectorStore type hinting in return value

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 22:02:02 -03:00
commit 924df4bdad
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,7 +4,7 @@ from langchain_community.vectorstores.redis import Redis
from langchain_core.embeddings import Embeddings from langchain_core.embeddings import Embeddings
from langflow.custom import CustomComponent from langflow.custom import CustomComponent
from langflow.field_typing import Retriever from langflow.field_typing import VectorStore
from langflow.schema import Data from langflow.schema import Data
@ -44,7 +44,7 @@ class RedisVectorStoreComponent(CustomComponent):
redis_index_name: str, redis_index_name: str,
schema: Optional[str] = None, schema: Optional[str] = None,
inputs: Optional[Data] = None, inputs: Optional[Data] = None,
) -> Retriever: ) -> VectorStore:
""" """
Builds the Vector Store or BaseRetriever object. Builds the Vector Store or BaseRetriever object.
@ -80,4 +80,4 @@ class RedisVectorStoreComponent(CustomComponent):
redis_url=redis_server_url, redis_url=redis_server_url,
index_name=redis_index_name, index_name=redis_index_name,
) )
return cast(Retriever, redis_vs) return cast(VectorStore, redis_vs)