opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 07:46:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update VAR_TYPE_MAP and initialData handling in useConfigurations for improved variable processing

Browse Source
This commit is contained in:
twwu 2025-06-13 15:57:16 +08:00
parent 5c228bca4f
commit 4b11d29ede
2 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
web/app/components/datasets/documents/create-from-pipeline/process-documents/hooks.ts
View File

@ -4,12 +4,13 @@ import { usePublishedPipelineProcessingParams } from '@/service/use-pipeline'
import { PipelineInputVarType } from '@/models/pipeline' import { PipelineInputVarType } from '@/models/pipeline'
import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail' import { useDatasetDetailContextWithSelector } from '@/context/dataset-detail'
type PartialInputVarType = PipelineInputVarType.textInput | PipelineInputVarType.number | PipelineInputVarType.select | PipelineInputVarType.checkbox const VAR_TYPE_MAP: Record<PipelineInputVarType, BaseFieldType> = {
const VAR_TYPE_MAP: Record<PartialInputVarType, BaseFieldType> = {
[PipelineInputVarType.textInput]: BaseFieldType.textInput, [PipelineInputVarType.textInput]: BaseFieldType.textInput,
[PipelineInputVarType.number]: BaseFieldType.numberInput, [PipelineInputVarType.paragraph]: BaseFieldType.paragraph,
[PipelineInputVarType.select]: BaseFieldType.select, [PipelineInputVarType.select]: BaseFieldType.select,
[PipelineInputVarType.singleFile]: BaseFieldType.file,
[PipelineInputVarType.multiFiles]: BaseFieldType.fileList,
[PipelineInputVarType.number]: BaseFieldType.numberInput,
[PipelineInputVarType.checkbox]: BaseFieldType.checkbox, [PipelineInputVarType.checkbox]: BaseFieldType.checkbox,
} }
@ -23,15 +24,15 @@ export const useConfigurations = (datasourceNodeId: string) => {
const initialData = useMemo(() => { const initialData = useMemo(() => {
const variables = paramsConfig?.variables || [] const variables = paramsConfig?.variables || []
return variables.reduce((acc, item) => { return variables.reduce((acc, item) => {
const type = VAR_TYPE_MAP[item.type as PartialInputVarType] const type = VAR_TYPE_MAP[item.type]
if (type === BaseFieldType.textInput) if ([BaseFieldType.textInput, BaseFieldType.paragraph, BaseFieldType.select].includes(type))
acc[item.variable] = '' acc[item.variable] = item.default_value ?? ''
if (type === BaseFieldType.numberInput) if (type === BaseFieldType.numberInput)
acc[item.variable] = 0 acc[item.variable] = item.default_value ?? 0
if (type === BaseFieldType.select)
acc[item.variable] = item.options?.[0] || ''
if (type === BaseFieldType.checkbox) if (type === BaseFieldType.checkbox)
acc[item.variable] = true acc[item.variable] = true
if ([BaseFieldType.file, BaseFieldType.fileList].includes(type))
acc[item.variable] = []
return acc return acc
}, {} as Record<string, any>) }, {} as Record<string, any>)
}, [paramsConfig]) }, [paramsConfig])
@ -39,7 +40,7 @@ export const useConfigurations = (datasourceNodeId: string) => {
const configurations = useMemo(() => { const configurations = useMemo(() => {
const variables = paramsConfig?.variables || [] const variables = paramsConfig?.variables || []
const configs = variables.map(item => ({ const configs = variables.map(item => ({
type: VAR_TYPE_MAP[item.type as PartialInputVarType], type: VAR_TYPE_MAP[item.type],
variable: item.variable, variable: item.variable,
label: item.label, label: item.label,
required: item.required, required: item.required,
@ -49,7 +50,12 @@ export const useConfigurations = (datasourceNodeId: string) => {
value: option, value: option,
})), })),
showConditions: [], showConditions: [],
default: item.default_value, placeholder: item.placeholder,
tooltip: item.tooltips,
unit: item.unit,
allowedFileTypes: item.allowed_file_types,
allowedFileExtensions: item.allowed_file_extensions,
allowedFileUploadMethods: item.allowed_file_upload_methods,
})) }))
return configs return configs
}, [paramsConfig]) }, [paramsConfig])

2
web/app/components/rag-pipeline/components/panel/test-run/document-processing/hooks.ts
View File

@ -14,6 +14,7 @@ const VAR_TYPE_MAP: Record<PipelineInputVarType, BaseFieldType> = {
[PipelineInputVarType.number]: BaseFieldType.numberInput, [PipelineInputVarType.number]: BaseFieldType.numberInput,
[PipelineInputVarType.checkbox]: BaseFieldType.checkbox, [PipelineInputVarType.checkbox]: BaseFieldType.checkbox,
} }
export const useConfigurations = (datasourceNodeId: string) => { export const useConfigurations = (datasourceNodeId: string) => {
const pipelineId = useStore(state => state.pipelineId) const pipelineId = useStore(state => state.pipelineId)
const { data: paramsConfig } = useDraftPipelineProcessingParams({ const { data: paramsConfig } = useDraftPipelineProcessingParams({
@ -50,7 +51,6 @@ export const useConfigurations = (datasourceNodeId: string) => {
value: option, value: option,
})), })),
showConditions: [], showConditions: [],
default: item.default_value,
placeholder: item.placeholder, placeholder: item.placeholder,
tooltip: item.tooltips, tooltip: item.tooltips,
unit: item.unit, unit: item.unit,