Merge branch 'form_io' of github.com:logspace-ai/langflow into form_io

src/backend/langflow/api/v1/validate.py

@@ -31,14 +31,28 @@ def post_validate_code(code: Code):
 def post_validate_prompt(prompt: ValidatePromptRequest):
     try:
         input_variables = validate_prompt(prompt.template)
+        # Reinitialize custom_fields
+        old_custom_fields = prompt.frontend_node.custom_fields.copy()
+        prompt.frontend_node.custom_fields = []
+        # Add new variables to the template
         for variable in input_variables:
             try:
                 template_field = TemplateField(
                     name=variable, field_type="str", show=True, advanced=False
                 )
 
-                prompt.frontend_node.template[variable] = template_field.dict()
+                prompt.frontend_node.template[variable] = template_field.to_dict()
                 prompt.frontend_node.custom_fields.append(variable)
+
+            except Exception as exc:
+                logger.exception(exc)
+                raise HTTPException(status_code=500, detail=str(exc)) from exc
+
+        # Remove variables that are not in the template anymore
+        for variable in old_custom_fields:
+            if variable not in input_variables:
+                try:
+                    prompt.frontend_node.template.pop(variable, None)
                 except Exception as exc:
                     logger.exception(exc)
                     raise HTTPException(status_code=500, detail=str(exc)) from exc

src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx

@@ -260,6 +260,9 @@ export default function ParameterComponent({
           </div>
         ) : left === true && type === "prompt" ? (
           <PromptAreaComponent
+            setNodeClass={(nodeClass) => {
+              data.node = nodeClass;
+            }}
             nodeClass={data.node}
             disabled={disabled}
             value={data.node.template[name].value ?? ""}

src/frontend/src/components/promptComponent/index.tsx

@@ -7,8 +7,8 @@ import { INPUT_STYLE } from "../../constants";
 import { ExternalLink } from "lucide-react";
 
 export default function PromptAreaComponent({
-  nodeClass,
   setNodeClass,
+  nodeClass,
   value,
   onChange,
   disabled,

src/frontend/src/utils.ts

@@ -830,11 +830,17 @@ export function groupByFamily(data, baseClasses) {
 
   Object.keys(data).map((d) => {
     Object.keys(data[d]).map((n) => {
+      try {
         if (
-        data[d][n].base_classes.some((r) => baseClasses.split("\n").includes(r))
+          data[d][n].base_classes.some((r) =>
+            baseClasses.split("\n").includes(r)
+          )
         ) {
           arrOfParent.push(d);
         }
+      } catch (e) {
+        console.log(e);
+      }
     });
   });
 
@@ -844,6 +850,7 @@ export function groupByFamily(data, baseClasses) {
 
   Object.keys(data).map((d) => {
     Object.keys(data[d]).map((n) => {
+      try {
         baseClasses.split("\n").forEach((tol) => {
           data[d][n].base_classes.forEach((data) => {
             if (tol == data) {
@@ -854,6 +861,9 @@ export function groupByFamily(data, baseClasses) {
             }
           });
         });
+      } catch (e) {
+        console.log(e);
+      }
     });
   });