From f1b1a3919fcdb1b5af333bc94d958425a79c6b66 Mon Sep 17 00:00:00 2001
From: coolgo0811 <coolgo0811@gmail.com>
Date: Fri, 5 Jan 2024 10:04:41 +0800
Subject: [PATCH] Fix vectorstores/Redis to support output component "Chains"

---
 src/backend/langflow/components/vectorstores/Redis.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/components/vectorstores/Redis.py b/src/backend/langflow/components/vectorstores/Redis.py
index f13428829..fb6e0f2a3 100644
--- a/src/backend/langflow/components/vectorstores/Redis.py
+++ b/src/backend/langflow/components/vectorstores/Redis.py
@@ -1,11 +1,11 @@
-from typing import Optional
+from typing import Optional, List, Union
 from langflow import CustomComponent
 
 from langchain.vectorstores.redis import Redis
 from langchain.schema import Document
 from langchain.vectorstores.base import VectorStore
 from langchain.embeddings.base import Embeddings
-
+from langchain.schema import BaseRetriever
 
 class RedisComponent(CustomComponent):
     """
@@ -42,7 +42,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.