From 6618efc4098c4d527e8988a33a1618cf3dcd55a0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 3 Oct 2023 17:59:48 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20refactor(base.py):=20remove=20un?= =?UTF-8?q?used=20import=20from=20langchain=20module=20in=20base.py=20?= =?UTF-8?q?=F0=9F=94=A8=20refactor(base.py):=20remove=20unused=20import=20?= =?UTF-8?q?from=20langchain.utilities=20module=20in=20base.py=20?= =?UTF-8?q?=F0=9F=94=A8=20refactor(base.py):=20remove=20unused=20import=20?= =?UTF-8?q?from=20langchain.requests=20module=20in=20base.py=20?= =?UTF-8?q?=F0=9F=94=A8=20refactor(base.py):=20remove=20unused=20import=20?= =?UTF-8?q?from=20langchain.sql=5Fdatabase=20module=20in=20base.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/utilities/base.py | 4 ++-- src/backend/langflow/interface/wrappers/base.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/interface/utilities/base.py b/src/backend/langflow/interface/utilities/base.py index 665143da2..bfc7cb11e 100644 --- a/src/backend/langflow/interface/utilities/base.py +++ b/src/backend/langflow/interface/utilities/base.py @@ -1,6 +1,6 @@ from typing import Dict, List, Optional, Type -from langchain import SQLDatabase, utilities +from langchain import utilities from langflow.custom.customs import get_custom_nodes from langflow.interface.base import LangChainTypeCreator @@ -32,7 +32,7 @@ class UtilityCreator(LangChainTypeCreator): utility_name: import_class(f"langchain.utilities.{utility_name}") for utility_name in utilities.__all__ } - self.type_dict["SQLDatabase"] = SQLDatabase + self.type_dict["SQLDatabase"] = utilities.SQLDatabase # Filter according to settings.utilities self.type_dict = { name: utility diff --git a/src/backend/langflow/interface/wrappers/base.py b/src/backend/langflow/interface/wrappers/base.py index c4399fb3e..de631101a 100644 --- a/src/backend/langflow/interface/wrappers/base.py +++ b/src/backend/langflow/interface/wrappers/base.py @@ -1,6 +1,6 @@ from typing import Dict, List, Optional -from langchain import requests, sql_database +from langchain.utilities import requests, sql_database from langflow.interface.base import LangChainTypeCreator from loguru import logger