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

✨ feat(base.py): add custom serialization method to FrontendNodeRequest class to remove unnecessary name field in the response

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-11 08:24:18 -03:00
commit e2fe2364ea
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/langflow/api/v1/base.py
View file

@ -1,9 +1,10 @@
from typing import Optional 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 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): class CacheResponse(BaseModel):
@ -17,6 +18,12 @@ class Code(BaseModel):
class FrontendNodeRequest(FrontendNode): class FrontendNodeRequest(FrontendNode):
template: dict # type: ignore 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): class ValidatePromptRequest(BaseModel):
name: str name: str