refactor for field types

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

@@ -22,7 +22,7 @@ import PromptAreaComponent from "../../../../components/promptComponent";
 import TextAreaComponent from "../../../../components/textAreaComponent";
 import ToggleShadComponent from "../../../../components/toggleShadComponent";
 import { Button } from "../../../../components/ui/button";
-import { TOOLTIP_EMPTY } from "../../../../constants/constants";
+import { LANGFLOW_SUPPORTED_TYPES, TOOLTIP_EMPTY } from "../../../../constants/constants";
 import { FlowsContext } from "../../../../contexts/flowsContext";
 import { typesContext } from "../../../../contexts/typesContext";
 import { ParameterComponentType } from "../../../../types/components";
@@ -223,15 +223,7 @@ export default function ParameterComponent({
 
   return !showNode ? (
     left &&
-    (type === "str" ||
+    (LANGFLOW_SUPPORTED_TYPES.has(type??"")) &&
-      type === "bool" ||
-      type === "float" ||
-      type === "code" ||
-      type === "prompt" ||
-      type === "file" ||
-      type === "int" ||
-      type === "dict" ||
-      type === "NestedDict") &&
     !optionalHandle ? (
       <></>
     ) : (
@@ -306,15 +298,7 @@ export default function ParameterComponent({
           </div>
         </div>
         {left &&
-        (type === "str" ||
+        (LANGFLOW_SUPPORTED_TYPES.has(type??"")) &&
-          type === "bool" ||
-          type === "float" ||
-          type === "code" ||
-          type === "prompt" ||
-          type === "file" ||
-          type === "int" ||
-          type === "dict" ||
-          type === "NestedDict") &&
         !optionalHandle ? (
           <></>
         ) : (

src/frontend/src/constants/constants.ts

@@ -663,3 +663,6 @@ export const LAST_USED_SPAN_1 = "The last time this key was used.";
 
 export const LAST_USED_SPAN_2 =
   "Accurate to within the hour from the most recent usage.";
+
+export const LANGFLOW_SUPPORTED_TYPES= new Set(["str","bool","float","code",
+"prompt","file","int","dict","NestedDict"])

src/frontend/src/modals/ApiModal/index.tsx

@@ -13,7 +13,7 @@ import {
 // import "ace-builds/webpack-resolver";
 import CodeTabsComponent from "../../components/codeTabsComponent";
 import IconComponent from "../../components/genericIconComponent";
-import { EXPORT_CODE_DIALOG } from "../../constants/constants";
+import { EXPORT_CODE_DIALOG, LANGFLOW_SUPPORTED_TYPES } from "../../constants/constants";
 import { AuthContext } from "../../contexts/authContext";
 import { FlowsContext } from "../../contexts/flowsContext";
 import { TemplateVariableType } from "../../types/api";
@@ -99,15 +99,7 @@ const ApiModal = forwardRef(
             (templateField) =>
               templateField.charAt(0) !== "_" &&
               node.data.node.template[templateField].show &&
-              (node.data.node.template[templateField].type === "str" ||
+              (LANGFLOW_SUPPORTED_TYPES.has(node.data.node.template[templateField].type))
-                node.data.node.template[templateField].type === "bool" ||
-                node.data.node.template[templateField].type === "float" ||
-                node.data.node.template[templateField].type === "code" ||
-                node.data.node.template[templateField].type === "prompt" ||
-                node.data.node.template[templateField].type === "file" ||
-                node.data.node.template[templateField].type === "int" ||
-                node.data.node.template[templateField].type === "dict" ||
-                node.data.node.template[templateField].type === "NestedDict")
           )
           .map((n, i) => {
             arrNodesWithValues.push(node["id"]);
@@ -146,9 +138,9 @@ const ApiModal = forwardRef(
       );
 
       if (existingTweak) {
-        existingTweak[tw][template["name"]] = changes as string;
+        existingTweak[tw][template["name"]!] = changes as string;
 
-        if (existingTweak[tw][template["name"]] == template.value) {
+        if (existingTweak[tw][template["name"]!] == template.value) {
           tweak.current.forEach((element) => {
             if (element[tw] && Object.keys(element[tw])?.length === 0) {
               tweak.current = tweak.current.filter((obj) => {
@@ -161,7 +153,7 @@ const ApiModal = forwardRef(
       } else {
         const newTweak = {
           [tw]: {
-            [template["name"]]: changes,
+            [template["name"]!]: changes,
           },
         } as uniqueTweakType;
         tweak.current.push(newTweak);

src/frontend/src/modals/EditNodeModal/index.tsx

@@ -25,7 +25,7 @@ import {
   TableHeader,
   TableRow,
 } from "../../components/ui/table";
-import { limitScrollFieldsModal } from "../../constants/constants";
+import { LANGFLOW_SUPPORTED_TYPES, limitScrollFieldsModal } from "../../constants/constants";
 import { FlowsContext } from "../../contexts/flowsContext";
 import { typesContext } from "../../contexts/typesContext";
 import { NodeDataType } from "../../types/flow";
@@ -145,24 +145,7 @@ const EditNodeModal = forwardRef(
                           (templateParam) =>
                             templateParam.charAt(0) !== "_" &&
                             myData.current.node?.template[templateParam].show &&
-                            (myData.current.node.template[templateParam]
+                            (LANGFLOW_SUPPORTED_TYPES.has(myData.current.node.template[templateParam].type))
-                              .type === "str" ||
-                              myData.current.node.template[templateParam]
-                                .type === "bool" ||
-                              myData.current.node.template[templateParam]
-                                .type === "float" ||
-                              myData.current.node.template[templateParam]
-                                .type === "code" ||
-                              myData.current.node.template[templateParam]
-                                .type === "prompt" ||
-                              myData.current.node.template[templateParam]
-                                .type === "file" ||
-                              myData.current.node.template[templateParam]
-                                .type === "int" ||
-                              myData.current.node.template[templateParam]
-                                .type === "dict" ||
-                              myData.current.node.template[templateParam]
-                                .type === "NestedDict")
                         )
                         .map((templateParam, index) => (
                           <TableRow key={index} className="h-10">

src/frontend/src/utils/reactflowUtils.ts

@@ -9,7 +9,7 @@ import {
   XYPosition,
 } from "reactflow";
 import ShortUniqueId from "short-unique-id";
-import { specialCharsRegex } from "../constants/constants";
+import { LANGFLOW_SUPPORTED_TYPES, specialCharsRegex } from "../constants/constants";
 import { APITemplateType, TemplateVariableType } from "../types/api";
 import {
   FlowType,
@@ -634,15 +634,7 @@ function updateGroupNodeTemplate(template: APITemplateType) {
     let type = template[key].type;
     let input_types = template[key].input_types;
     if (
-      (type === "str" ||
+      LANGFLOW_SUPPORTED_TYPES.has(type) &&
-        type === "bool" ||
-        type === "float" ||
-        type === "code" ||
-        type === "prompt" ||
-        type === "file" ||
-        type === "int" ||
-        type === "dict" ||
-        type === "NestedDict") &&
       !template[key].required &&
       !input_types
     ) {