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.