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

Refactor format_template field in FrontendNode

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-11 21:45:32 -03:00
commit debf495cf9
2 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/template/frontend_node/base.py
View file

@ -40,7 +40,7 @@ class FieldFormatters(BaseModel):
class FrontendNode(BaseModel): class FrontendNode(BaseModel):
format_template: bool = Field(default=True, exclude=True) _format_template: bool = True
template: Template template: Template
description: Optional[str] = None description: Optional[str] = None
base_classes: List[str] base_classes: List[str]
@ -81,7 +81,7 @@ class FrontendNode(BaseModel):
def serialize_model(self, handler): def serialize_model(self, handler):
result = handler(self) result = handler(self)
if hasattr(self, "template") and hasattr(self.template, "to_dict"): 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) result["template"] = self.template.to_dict(format_func)
name = result.pop("name") name = result.pop("name")

4
src/backend/langflow/template/frontend_node/custom_components.py
View file

@ -1,7 +1,5 @@
from typing import Optional from typing import Optional
from pydantic import Field
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
from langflow.template.template.base import Template from langflow.template.template.base import Template
@ -47,7 +45,7 @@ class Component(CustomComponent):
class CustomComponentFrontendNode(FrontendNode): class CustomComponentFrontendNode(FrontendNode):
format_template: bool = Field(default=False, exclude=True) _format_template: bool = False
name: str = "CustomComponent" name: str = "CustomComponent"
display_name: Optional[str] = "CustomComponent" display_name: Optional[str] = "CustomComponent"
beta: bool = True beta: bool = True