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

🐛 fix(types.py): prevent adding base classes that are in CLASSES_TO_REMOVE list to frontend_node

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-25 20:54:57 -03:00
commit 683d0ffcc8
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/types.py
View file

@ -18,6 +18,7 @@ from langflow.interface.custom.base import custom_component_creator
from langflow.interface.custom.custom_component import CustomComponent
from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.constants import CLASSES_TO_REMOVE
from langflow.template.frontend_node.custom_components import (
CustomComponentFrontendNode,
)
@ -247,7 +248,8 @@ def add_base_classes(frontend_node, return_type):
base_classes = get_base_classes(return_type_instance)
for base_class in base_classes:
frontend_node.get("base_classes").append(base_class)
if base_class not in CLASSES_TO_REMOVE:
frontend_node.get("base_classes").append(base_class)
def build_langchain_template_custom_component(custom_component: CustomComponent):