pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-26 00:28:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 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
parent f486c27a2f
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):