From d2eb36bef2f0c46e1adb8f5211588281a6d2d682 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 21:52:20 -0300 Subject: [PATCH] refactor: Update RedisVectorStoreComponent to use Retriever type hinting in return value --- .../base/langflow/components/vectorstores/Redis.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/components/vectorstores/Redis.py b/src/backend/base/langflow/components/vectorstores/Redis.py index 65b1b6f70..a7d6ed7b9 100644 --- a/src/backend/base/langflow/components/vectorstores/Redis.py +++ b/src/backend/base/langflow/components/vectorstores/Redis.py @@ -1,11 +1,10 @@ -from typing import Optional, Union +from typing import Optional from langchain_community.vectorstores.redis import Redis from langchain_core.embeddings import Embeddings -from langchain_core.retrievers import BaseRetriever -from langchain_core.vectorstores import VectorStore from langflow.custom import CustomComponent +from langflow.field_typing import Retriever from langflow.schema import Data @@ -45,7 +44,7 @@ class RedisVectorStoreComponent(CustomComponent): redis_index_name: str, schema: Optional[str] = None, inputs: Optional[Data] = None, - ) -> Union[VectorStore, BaseRetriever]: + ) -> Retriever: """ Builds the Vector Store or BaseRetriever object. @@ -82,3 +81,4 @@ class RedisVectorStoreComponent(CustomComponent): index_name=redis_index_name, ) return redis_vs + return redis_vs