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

Refactor Template class in base.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-11 13:04:01 -03:00
commit 7953950f14
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,9 +1,8 @@
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):
@ -31,6 +30,7 @@ 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