From 339029cd5bc2c96ce424427a291860e56f541cca Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 31 Jan 2024 17:33:32 -0300 Subject: [PATCH] Update import statement for retrievers module --- src/backend/langflow/interface/retrievers/base.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/interface/retrievers/base.py b/src/backend/langflow/interface/retrievers/base.py index 63d0ef915..a6813e76c 100644 --- a/src/backend/langflow/interface/retrievers/base.py +++ b/src/backend/langflow/interface/retrievers/base.py @@ -1,13 +1,12 @@ from typing import Any, ClassVar, Dict, List, Optional, Type -from langchain import retrievers -from loguru import logger - +from langchain_community import retrievers from langflow.interface.base import LangChainTypeCreator from langflow.interface.importing.utils import import_class from langflow.services.deps import get_settings_service from langflow.template.frontend_node.retrievers import RetrieverFrontendNode from langflow.utils.util import build_template_from_class, build_template_from_method +from loguru import logger class RetrieverCreator(LangChainTypeCreator):