Quellcode durchsuchen

fix(workflow): improve formatting and add checkbox type to input variable type mapping

tags/2.0.0-beta.1
twwu vor 2 Monaten
Ursprung
Commit
5e854238b0

+ 1
- 1
web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx Datei anzeigen

const data = await fetchDynamicOptions() const data = await fetchDynamicOptions()
setDynamicOptions(data?.options || []) setDynamicOptions(data?.options || [])
} }
finally {
finally {
setIsLoading(false) setIsLoading(false)
} }
} }

+ 14
- 14
web/app/components/workflow/nodes/_base/hooks/use-available-var-list.ts Datei anzeigen

hideChatVar, hideChatVar,
passedInAvailableNodes, passedInAvailableNodes,
}: Params = { }: Params = {
onlyLeafNodeVar: false,
filterVar: () => true,
}) => {
onlyLeafNodeVar: false,
filterVar: () => true,
}) => {
const { getTreeLeafNodes, getNodeById, getBeforeNodesInSameBranchIncludeParent } = useWorkflow() const { getTreeLeafNodes, getNodeById, getBeforeNodesInSameBranchIncludeParent } = useWorkflow()
const { getNodeAvailableVars } = useWorkflowVariables() const { getNodeAvailableVars } = useWorkflowVariables()
const isChatMode = useIsChatMode() const isChatMode = useIsChatMode()
const ragPipelineVariables = useWorkflowStore(s => s.ragPipelineVariables) const ragPipelineVariables = useWorkflowStore(s => s.ragPipelineVariables)
const isDataSourceNode = currNode?.data?.type === BlockEnum.DataSource const isDataSourceNode = currNode?.data?.type === BlockEnum.DataSource
const dataSourceRagVars: NodeOutPutVar[] = [] const dataSourceRagVars: NodeOutPutVar[] = []
if(isDataSourceNode) {
if (isDataSourceNode) {
const ragVariablesInDataSource = ragPipelineVariables?.filter(ragVariable => ragVariable.belong_to_node_id === nodeId) const ragVariablesInDataSource = ragPipelineVariables?.filter(ragVariable => ragVariable.belong_to_node_id === nodeId)
const filterVars = ragVariablesInDataSource?.filter(v => filterVar({ const filterVars = ragVariablesInDataSource?.filter(v => filterVar({
variable: v.variable, variable: v.variable,
nodeId, nodeId,
isRagVariable: true, isRagVariable: true,
}, ['rag', nodeId, v.variable])) }, ['rag', nodeId, v.variable]))
if(filterVars?.length) {
if (filterVars?.length) {
dataSourceRagVars.push({ dataSourceRagVars.push({
nodeId,
title: currNode.data?.title,
vars: filterVars.map((v) => {
return {
variable: `rag.${nodeId}.${v.variable}`,
type: inputVarTypeToVarType(v.type),
description: v.label,
isRagVariable: true,
nodeId,
title: currNode.data?.title,
vars: filterVars.map((v) => {
return {
variable: `rag.${nodeId}.${v.variable}`,
type: inputVarTypeToVarType(v.type),
description: v.label,
isRagVariable: true,
} as Var } as Var
}),
}),
}) })
} }
} }

+ 1
- 0
web/app/components/workflow/nodes/data-source/utils.ts Datei anzeigen

[PipelineInputVarType.number]: VarType.number, [PipelineInputVarType.number]: VarType.number,
[PipelineInputVarType.singleFile]: VarType.file, [PipelineInputVarType.singleFile]: VarType.file,
[PipelineInputVarType.multiFiles]: VarType.arrayFile, [PipelineInputVarType.multiFiles]: VarType.arrayFile,
[PipelineInputVarType.checkbox]: VarType.boolean,
} as any)[type] || VarType.string } as any)[type] || VarType.string
} }

Laden…
Abbrechen
Speichern