From e2fe2364ea0d5572c98579ca420be8dc943eaaf2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Dec 2023 08:24:18 -0300 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat(base.py):=20add=20custom=20ser?= =?UTF-8?q?ialization=20method=20to=20FrontendNodeRequest=20class=20to=20r?= =?UTF-8?q?emove=20unnecessary=20name=20field=20in=20the=20response?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/base.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/api/v1/base.py b/src/backend/langflow/api/v1/base.py index 15bb4e17a..01e4e9c79 100644 --- a/src/backend/langflow/api/v1/base.py +++ b/src/backend/langflow/api/v1/base.py @@ -1,9 +1,10 @@ from typing import Optional -from langflow.template.frontend_node.base import FrontendNode -from pydantic import field_validator, BaseModel + +from langchain.prompts import PromptTemplate +from pydantic import BaseModel, field_validator, model_serializer from langflow.interface.utils import extract_input_variables_from_prompt -from langchain.prompts import PromptTemplate +from langflow.template.frontend_node.base import FrontendNode class CacheResponse(BaseModel): @@ -17,6 +18,12 @@ class Code(BaseModel): class FrontendNodeRequest(FrontendNode): template: dict # type: ignore + @model_serializer(mode="wrap") + def serialize_model(self, handler): + # Override the default serialization method in FrontendNode + # because we don't need the name in the response (i.e. {name: {}}) + return handler(self) + class ValidatePromptRequest(BaseModel): name: str