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

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

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 21:58:07 -03:00
commit 069c27bda0
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

@ -1,4 +1,4 @@
from typing import Optional from typing import Optional, cast
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
@ -80,5 +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 redis_vs return cast(Retriever, redis_vs)
return redis_vs