jandres/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-05 22:15:52 +00:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

feat: llm intput only number and str

Browse Source
This commit is contained in:
Joel
2024-03-20 22:00:56 +08:00
parent 75e876b14e
commit 3d4d60a353
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web/app/components/workflow/nodes/llm/panel.tsx
View File

@@ -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>

5
web/app/components/workflow/nodes/llm/use-config.ts
View File

@@ -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,