From e760f228478facb655a02faf12fdc8fed05f3df8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 27 Jun 2023 19:17:46 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(base.py):=20change=20fronten?= =?UTF-8?q?d=5Fnode=20field=20in=20ValidatePromptRequest=20to=20use=20Fron?= =?UTF-8?q?tendNodeRequest=20instead=20of=20FrontendNode=20=F0=9F=94=A5=20?= =?UTF-8?q?chore(base.py):=20remove=20unused=20FrontendNodeRequest=20class?= =?UTF-8?q?=20=F0=9F=94=A8=20refactor(validate.py):=20modify=20post=5Fvali?= =?UTF-8?q?date=5Fprompt=20function=20to=20update=20the=20template=20field?= =?UTF-8?q?=20in=20the=20frontend=5Fnode=20object=20The=20ValidatePromptRe?= =?UTF-8?q?quest=20class=20was=20using=20the=20FrontendNode=20class=20inst?= =?UTF-8?q?ead=20of=20the=20FrontendNodeRequest=20class,=20which=20caused?= =?UTF-8?q?=20an=20error.=20The=20unused=20FrontendNodeRequest=20class=20w?= =?UTF-8?q?as=20removed.=20The=20post=5Fvalidate=5Fprompt=20function=20was?= =?UTF-8?q?=20modified=20to=20update=20the=20template=20field=20in=20the?= =?UTF-8?q?=20frontend=5Fnode=20object=20instead=20of=20appending=20to=20t?= =?UTF-8?q?he=20fields=20list.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/base.py | 8 ++++++-- src/backend/langflow/api/v1/validate.py | 3 ++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/api/v1/base.py b/src/backend/langflow/api/v1/base.py index 1a4936a2f..5c32536c8 100644 --- a/src/backend/langflow/api/v1/base.py +++ b/src/backend/langflow/api/v1/base.py @@ -13,9 +13,13 @@ class Code(BaseModel): code: str +class FrontendNodeRequest(FrontendNode): + template: dict + + class ValidatePromptRequest(BaseModel): template: str - frontend_node: FrontendNode + frontend_node: FrontendNodeRequest # Build ValidationResponse class for {"imports": {"errors": []}, "function": {"errors": []}} @@ -34,7 +38,7 @@ class CodeValidationResponse(BaseModel): class PromptValidationResponse(BaseModel): input_variables: list - frontend_node: FrontendNode + frontend_node: FrontendNodeRequest INVALID_CHARACTERS = { diff --git a/src/backend/langflow/api/v1/validate.py b/src/backend/langflow/api/v1/validate.py index 3046ad9ba..658e5b274 100644 --- a/src/backend/langflow/api/v1/validate.py +++ b/src/backend/langflow/api/v1/validate.py @@ -36,7 +36,8 @@ def post_validate_prompt(prompt: ValidatePromptRequest): template_field = TemplateField( name=variable, field_type="str", show=True, advanced=False ) - prompt.frontend_node.template.fields.append(template_field) + + prompt.frontend_node.template[variable] = template_field.dict() prompt.frontend_node.custom_fields.append(variable) except Exception as exc: logger.exception(exc)