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)