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

Remove field_order from template

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-06 12:56:32 -03:00
commit 199b656f19
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/template/template/base.py
View file

@ -1,14 +1,14 @@
from typing import Callable, Union from typing import Callable, Union
from pydantic import BaseModel, model_serializer
from langflow.template.field.base import TemplateField from langflow.template.field.base import TemplateField
from langflow.utils.constants import DIRECT_TYPES from langflow.utils.constants import DIRECT_TYPES
from pydantic import BaseModel, model_serializer
class Template(BaseModel): class Template(BaseModel):
type_name: str type_name: str
fields: list[TemplateField] fields: list[TemplateField]
field_order: list[str] = []
def process_fields( def process_fields(
self, self,
@ -30,7 +30,6 @@ class Template(BaseModel):
for field in self.fields: for field in self.fields:
result[field.name] = field.model_dump(by_alias=True, exclude_none=True) result[field.name] = field.model_dump(by_alias=True, exclude_none=True)
result["_type"] = result.pop("type_name") result["_type"] = result.pop("type_name")
result.pop("field_order", None)
return result return result
# For backwards compatibility # For backwards compatibility
@ -46,7 +45,9 @@ class Template(BaseModel):
"""Returns the field with the given name.""" """Returns the field with the given name."""
field = next((field for field in self.fields if field.name == field_name), None) field = next((field for field in self.fields if field.name == field_name), None)
if field is None: if field is None:
raise ValueError(f"Field {field_name} not found in template {self.type_name}") raise ValueError(
f"Field {field_name} not found in template {self.type_name}"
)
return field return field
def update_field(self, field_name: str, field: TemplateField) -> None: def update_field(self, field_name: str, field: TemplateField) -> None: