diff --git a/src/backend/langflow/interface/wrappers/base.py b/src/backend/langflow/interface/wrappers/base.py index 469559ad9..bc42f1430 100644 --- a/src/backend/langflow/interface/wrappers/base.py +++ b/src/backend/langflow/interface/wrappers/base.py @@ -1,6 +1,6 @@ -from typing import ClassVar, Dict, List, Optional +from typing import Dict, List, Optional -from langchain_community.utilities import requests, sql_database +from langchain_community.utilities import requests from loguru import logger from langflow.interface.base import LangChainTypeCreator @@ -10,13 +10,11 @@ from langflow.utils.util import build_template_from_class, build_template_from_m class WrapperCreator(LangChainTypeCreator): type_name: str = "wrappers" - from_method_nodes: ClassVar[Dict] = {"SQLDatabase": "from_uri"} - @property def type_to_loader_dict(self) -> Dict: if self.type_dict is None: self.type_dict = { - wrapper.__name__: wrapper for wrapper in [requests.TextRequestsWrapper, sql_database.SQLDatabase] + wrapper.__name__: wrapper for wrapper in [requests.TextRequestsWrapper] } return self.type_dict