Refactor Template class in base.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-11 13:04:01 -03:00
commit 7953950f14

View file

@ -1,9 +1,8 @@
from typing import Callable, Union
from pydantic import BaseModel, model_serializer
from langflow.template.field.base import TemplateField
from langflow.utils.constants import DIRECT_TYPES
from pydantic import BaseModel, model_serializer
class Template(BaseModel):
@ -31,6 +30,7 @@ class Template(BaseModel):
for field in self.fields:
result[field.name] = field.model_dump(by_alias=True, exclude_none=True)
result["_type"] = result.pop("type_name")
result.pop("field_order", None)
return result
# For backwards compatibility