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

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 21:52:20 -03:00
commit d2eb36bef2

View file

@ -1,11 +1,10 @@
from typing import Optional, Union
from typing import Optional
from langchain_community.vectorstores.redis import Redis
from langchain_core.embeddings import Embeddings
from langchain_core.retrievers import BaseRetriever
from langchain_core.vectorstores import VectorStore
from langflow.custom import CustomComponent
from langflow.field_typing import Retriever
from langflow.schema import Data
@ -45,7 +44,7 @@ class RedisVectorStoreComponent(CustomComponent):
redis_index_name: str,
schema: Optional[str] = None,
inputs: Optional[Data] = None,
) -> Union[VectorStore, BaseRetriever]:
) -> Retriever:
"""
Builds the Vector Store or BaseRetriever object.
@ -82,3 +81,4 @@ class RedisVectorStoreComponent(CustomComponent):
index_name=redis_index_name,
)
return redis_vs
return redis_vs