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

Remove unused code in WrapperCreator class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 15:51:58 -03:00
commit 5785f692ec
1 changed files with 1 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,7 +4,7 @@ from langchain_community.utilities import requests
from loguru import logger from loguru import logger
from langflow.interface.base import LangChainTypeCreator from langflow.interface.base import LangChainTypeCreator
from langflow.utils.util import build_template_from_class, build_template_from_method from langflow.utils.util import build_template_from_class
class WrapperCreator(LangChainTypeCreator): class WrapperCreator(LangChainTypeCreator):
@ -20,14 +20,6 @@ class WrapperCreator(LangChainTypeCreator):
def get_signature(self, name: str) -> Optional[Dict]: def get_signature(self, name: str) -> Optional[Dict]:
try: try:
if name in self.from_method_nodes:
return build_template_from_method(
name,
type_to_cls_dict=self.type_to_loader_dict,
add_function=True,
method_name=self.from_method_nodes[name],
)
return build_template_from_class(name, self.type_to_loader_dict) return build_template_from_class(name, self.type_to_loader_dict)
except ValueError as exc: except ValueError as exc:
raise ValueError("Wrapper not found") from exc raise ValueError("Wrapper not found") from exc