opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: support add files and vision

This commit is contained in:
Joel 2024-03-18 21:59:38 +08:00
parent a89287bf20
commit 7b9fbccf60
3 changed files with 30 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
web/app/components/workflow/nodes/llm/default.ts
View File

@ -26,7 +26,7 @@ const nodeDefault: NodeDefault<LLMNodeType> = {
text: '',
}],
context: {
enabled: true,
enabled: false,
variable_selector: [],
},
vision: {

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

@ -81,12 +81,12 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
label: t(`${i18nPrefix}.context`)!,
inputs: [{
label: '',
variable: 'contexts',
variable: '#context#',
type: InputVarType.contexts,
required: false,
}],
values: { contexts },
onChange: keyValue => setContexts((keyValue as any).contexts),
values: { '#context#': contexts },
onChange: keyValue => setContexts((keyValue as any)['#context#']),
},
)
}
@ -97,12 +97,12 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
label: t(`${i18nPrefix}.vision`)!,
inputs: [{
label: t(`${i18nPrefix}.files`)!,
variable: 'visionFiles',
variable: '#files#',
type: InputVarType.files,
required: false,
}],
values: { visionFiles },
onChange: keyValue => setVisionFiles((keyValue as any).visionFiles),
values: { '#files#': visionFiles },
onChange: keyValue => setVisionFiles((keyValue as any)['#files#']),
},
)
}

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

@ -148,6 +148,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
const handleContextVarChange = useCallback((newVar: ValueSelector | string) => {
const newInputs = produce(inputs, (draft) => {
draft.context.variable_selector = newVar as ValueSelector
draft.context.enabled = !!(newVar && newVar.length > 0)
})
setInputs(newInputs)
}, [inputs, setInputs])
@ -198,14 +199,31 @@ const useConfig = (id: string, payload: LLMNodeType) => {
data: inputs,
defaultRunInputData: {
'#context#': [RETRIEVAL_OUTPUT_STRUCT],
'#vision#': [],
'#files#': [],
},
})
// const handleRun = (submitData: Record<string, any>) => {
// console.log(submitData)
// const res = produce(submitData, (draft) => {
// debugger
// if (draft.contexts) {
// draft['#context#'] = draft.contexts
// delete draft.contexts
// }
// if (draft.visionFiles) {
// draft['#files#'] = draft.visionFiles
// delete draft.visionFiles
// }
// })
// doHandleRun(res)
// }
const inputVarValues = (() => {
const vars: Record<string, any> = {}
Object.keys(runInputData)
.filter(key => !['#context#', '#vision#'].includes(key))
.filter(key => !['#context#', '#files#'].includes(key))
.forEach((key) => {
vars[key] = runInputData[key]
})
@ -216,7 +234,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
const newVars = {
...newPayload,
'#context#': runInputData['#context#'],
'#vision#': runInputData['#vision#'],
'#files#': runInputData['#files#'],
}
setRunInputData(newVars)
}, [runInputData, setRunInputData])
@ -229,11 +247,11 @@ const useConfig = (id: string, payload: LLMNodeType) => {
})
}, [runInputData, setRunInputData])
const visionFiles = runInputData['#vision']
const visionFiles = runInputData['#files#']
const setVisionFiles = useCallback((newFiles: any[]) => {
setRunInputData({
...runInputData,
'#vision#': newFiles,
'#files#': newFiles,
})
}, [runInputData, setRunInputData])