From debf495cf9e0d4bbe6cfa58845b93154b8ef5523 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Dec 2023 21:45:32 -0300 Subject: [PATCH] Refactor format_template field in FrontendNode --- src/backend/langflow/template/frontend_node/base.py | 4 ++-- .../langflow/template/frontend_node/custom_components.py | 4 +--- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/template/frontend_node/base.py b/src/backend/langflow/template/frontend_node/base.py index ce1fe79a0..6379dd358 100644 --- a/src/backend/langflow/template/frontend_node/base.py +++ b/src/backend/langflow/template/frontend_node/base.py @@ -40,7 +40,7 @@ class FieldFormatters(BaseModel): class FrontendNode(BaseModel): - format_template: bool = Field(default=True, exclude=True) + _format_template: bool = True template: Template description: Optional[str] = None base_classes: List[str] @@ -81,7 +81,7 @@ class FrontendNode(BaseModel): def serialize_model(self, handler): result = handler(self) if hasattr(self, "template") and hasattr(self.template, "to_dict"): - format_func = self.format_field if self.format_template else None + format_func = self.format_field if self._format_template else None result["template"] = self.template.to_dict(format_func) name = result.pop("name") diff --git a/src/backend/langflow/template/frontend_node/custom_components.py b/src/backend/langflow/template/frontend_node/custom_components.py index c9f80beae..d604ae055 100644 --- a/src/backend/langflow/template/frontend_node/custom_components.py +++ b/src/backend/langflow/template/frontend_node/custom_components.py @@ -1,7 +1,5 @@ from typing import Optional -from pydantic import Field - from langflow.template.field.base import TemplateField from langflow.template.frontend_node.base import FrontendNode from langflow.template.template.base import Template @@ -47,7 +45,7 @@ class Component(CustomComponent): class CustomComponentFrontendNode(FrontendNode): - format_template: bool = Field(default=False, exclude=True) + _format_template: bool = False name: str = "CustomComponent" display_name: Optional[str] = "CustomComponent" beta: bool = True