From cfd4a229ea464ab7ac9b9786aba7b11ec50d0b69 Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Sat, 8 Apr 2023 08:55:22 -0300 Subject: [PATCH] fix: adding chainfrontendnode --- src/backend/langflow/interface/chains/base.py | 7 ++++++- src/backend/langflow/template/nodes.py | 10 ++++++++++ 2 files changed, 16 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/interface/chains/base.py b/src/backend/langflow/interface/chains/base.py index 0a46d3474..6f90b0e62 100644 --- a/src/backend/langflow/interface/chains/base.py +++ b/src/backend/langflow/interface/chains/base.py @@ -1,9 +1,10 @@ -from typing import Dict, List, Optional +from typing import Dict, List, Optional, Type from langflow.custom.customs import get_custom_nodes from langflow.interface.base import LangChainTypeCreator from langflow.interface.custom_lists import chain_type_to_cls_dict from langflow.settings import settings +from langflow.template.nodes import ChainFrontendNode from langflow.utils.util import build_template_from_class # Assuming necessary imports for Field, Template, and FrontendNode classes @@ -12,6 +13,10 @@ from langflow.utils.util import build_template_from_class class ChainCreator(LangChainTypeCreator): type_name: str = "chains" + @property + def frontend_node_class(self) -> Type[ChainFrontendNode]: + return ChainFrontendNode + @property def type_to_loader_dict(self) -> Dict: if self.type_dict is None: diff --git a/src/backend/langflow/template/nodes.py b/src/backend/langflow/template/nodes.py index c76c56b14..c5d9bf23d 100644 --- a/src/backend/langflow/template/nodes.py +++ b/src/backend/langflow/template/nodes.py @@ -289,3 +289,13 @@ class MemoryFrontendNode(FrontendNode): field.field_type = "int" field.value = 10 field.display_name = "Memory Size" + + +class ChainFrontendNode(FrontendNode): + @staticmethod + def format_field(field: TemplateField, name: Optional[str] = None) -> None: + FrontendNode.format_field(field, name) + + if "key" in field.name: + field.password = False + field.show = False