From 3d4d60a3532e3ef872bcf79a237bb6e961bf0c6e Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Wed, 20 Mar 2024 22:00:56 +0800
Subject: [PATCH] feat: llm intput only number and str

---
 web/app/components/workflow/nodes/llm/panel.tsx     | 2 ++
 web/app/components/workflow/nodes/llm/use-config.ts | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/web/app/components/workflow/nodes/llm/panel.tsx b/web/app/components/workflow/nodes/llm/panel.tsx
index 84dd9dee55..ee280ac05d 100644
--- a/web/app/components/workflow/nodes/llm/panel.tsx
+++ b/web/app/components/workflow/nodes/llm/panel.tsx
@@ -40,6 +40,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
     handleVarListChange,
     handleAddVariable,
     handleContextVarChange,
+    filterInputVar,
     filterVar,
     handlePromptChange,
     handleMemoryChange,
@@ -142,6 +143,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
             nodeId={id}
             list={inputs.variables}
             onChange={handleVarListChange}
+            filterVar={filterInputVar}
           />
         </Field>
 
diff --git a/web/app/components/workflow/nodes/llm/use-config.ts b/web/app/components/workflow/nodes/llm/use-config.ts
index 88ea68e3b4..7e083e307d 100644
--- a/web/app/components/workflow/nodes/llm/use-config.ts
+++ b/web/app/components/workflow/nodes/llm/use-config.ts
@@ -183,6 +183,10 @@ const useConfig = (id: string, payload: LLMNodeType) => {
     setInputs(newInputs)
   }, [inputs, setInputs])
 
+  const filterInputVar = useCallback((varPayload: Var) => {
+    return [VarType.number, VarType.string].includes(varPayload.type)
+  }, [])
+
   const filterVar = useCallback((varPayload: Var) => {
     return [VarType.arrayObject, VarType.array, VarType.string].includes(varPayload.type)
   }, [])
@@ -274,6 +278,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
     handleVarListChange,
     handleAddVariable,
     handleContextVarChange,
+    filterInputVar,
     filterVar,
     handlePromptChange,
     handleMemoryChange,