Fix vectorstores/Redis to support output component "Chains" (#1292)

The previous version was missing  output component "Chains".
Before:

![image](https://github.com/logspace-ai/langflow/assets/7540749/079dca11-68da-4a0d-8d0e-80b97321d90e)
After:

![image](https://github.com/logspace-ai/langflow/assets/7540749/9331ceac-28be-4a4d-a37a-f3b20ae18ea8)
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-01 09:20:19 -03:00 committed by GitHub
commit a6877033f8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,6 +1,8 @@
from typing import Optional
from typing import Optional, List, Union
from langflow import CustomComponent
from langchain.embeddings.base import Embeddings
from langchain.schema import BaseRetriever
from langchain.schema import Document
from langchain_community.vectorstores import VectorStore
from langchain_community.vectorstores.redis import Redis
@ -42,7 +44,7 @@ class RedisComponent(CustomComponent):
redis_server_url: str,
redis_index_name: str,
documents: Optional[Document] = None,
) -> VectorStore:
) -> Union[VectorStore, BaseRetriever]:
"""
Builds the Vector Store or BaseRetriever object.