diff --git a/src/backend/langflow/components/vectorstores/Redis.py b/src/backend/langflow/components/vectorstores/Redis.py
index 6a4e9db2a..95125095c 100644
--- a/src/backend/langflow/components/vectorstores/Redis.py
+++ b/src/backend/langflow/components/vectorstores/Redis.py
@@ -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.