From 53a0566f8c38eb181c1199ed63392b10b94e4fa4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 9 Dec 2023 23:22:49 -0300 Subject: [PATCH] Refactor code to use model_dump method for adding new custom field --- src/backend/langflow/interface/types.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/interface/types.py b/src/backend/langflow/interface/types.py index c508816a0..478bf0ca0 100644 --- a/src/backend/langflow/interface/types.py +++ b/src/backend/langflow/interface/types.py @@ -8,8 +8,6 @@ from uuid import UUID from cachetools import LRUCache, cached from fastapi import HTTPException -from loguru import logger - from langflow.interface.agents.base import agent_creator from langflow.interface.chains.base import chain_creator from langflow.interface.custom.custom_component import CustomComponent @@ -33,6 +31,7 @@ from langflow.template.field.base import TemplateField from langflow.template.frontend_node.constants import CLASSES_TO_REMOVE from langflow.template.frontend_node.custom_components import CustomComponentFrontendNode from langflow.utils.util import get_base_classes +from loguru import logger # Used to get the base_classes list @@ -130,7 +129,7 @@ def add_new_custom_field( display_name=display_name, **field_config, ) - template.get("template")[field_name] = new_field.to_dict() + template.get("template")[field_name] = new_field.model_dump(by_alias=True, exclude_none=True) template.get("custom_fields")[field_name] = None return template