refactor: Update RedisVectorStoreComponent to use Retriever type hinting in return value
This commit is contained in:
parent
5561b4ec52
commit
d2eb36bef2
1 changed files with 4 additions and 4 deletions
|
|
@ -1,11 +1,10 @@
|
||||||
from typing import Optional, Union
|
from typing import Optional
|
||||||
|
|
||||||
from langchain_community.vectorstores.redis import Redis
|
from langchain_community.vectorstores.redis import Redis
|
||||||
from langchain_core.embeddings import Embeddings
|
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.custom import CustomComponent
|
||||||
|
from langflow.field_typing import Retriever
|
||||||
from langflow.schema import Data
|
from langflow.schema import Data
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -45,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,
|
||||||
) -> Union[VectorStore, BaseRetriever]:
|
) -> Retriever:
|
||||||
"""
|
"""
|
||||||
Builds the Vector Store or BaseRetriever object.
|
Builds the Vector Store or BaseRetriever object.
|
||||||
|
|
||||||
|
|
@ -82,3 +81,4 @@ class RedisVectorStoreComponent(CustomComponent):
|
||||||
index_name=redis_index_name,
|
index_name=redis_index_name,
|
||||||
)
|
)
|
||||||
return redis_vs
|
return redis_vs
|
||||||
|
return redis_vs
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue