jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Remove unused import and class variable in base.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 13:25:32 -03:00
commit 1c0cf20830
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/interface/wrappers/base.py
View file

@ -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