From 069c27bda04ceb61af05acff8d09fda1f9e18a9c Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 21:58:07 -0300 Subject: [PATCH] refactor: Update RedisVectorStoreComponent to use Retriever type hinting in return value --- src/backend/base/langflow/components/vectorstores/Redis.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/vectorstores/Redis.py b/src/backend/base/langflow/components/vectorstores/Redis.py index a7d6ed7b9..e83fcfa4c 100644 --- a/src/backend/base/langflow/components/vectorstores/Redis.py +++ b/src/backend/base/langflow/components/vectorstores/Redis.py @@ -1,4 +1,4 @@ -from typing import Optional +from typing import Optional, cast from langchain_community.vectorstores.redis import Redis from langchain_core.embeddings import Embeddings @@ -80,5 +80,4 @@ class RedisVectorStoreComponent(CustomComponent): redis_url=redis_server_url, index_name=redis_index_name, ) - return redis_vs - return redis_vs + return cast(Retriever, redis_vs)