This commit is contained in:
StyleZhang
2024-03-15 11:01:47 +08:00
parent 05ac27dfa8
commit 985c07b25b
2 changed files with 6 additions and 3 deletions

View File

@@ -60,7 +60,7 @@ export const useWorkflow = () => {
setNodes(newNodes)
}, [store])
const getTreeLeafNodes = useCallback(() => {
const getTreeLeafNodes = useCallback((nodeId: string) => {
const {
getNodes,
edges,
@@ -73,6 +73,8 @@ export const useWorkflow = () => {
const list: Node[] = []
const preOrder = (root: Node, callback: (node: Node) => void) => {
if (root.id === nodeId)
return
const outgoers = getOutgoers(root, nodes, edges)
if (outgoers.length) {
@@ -81,7 +83,8 @@ export const useWorkflow = () => {
})
}
else {
callback(root)
if (root.id !== nodeId)
callback(root)
}
}
preOrder(startNode, (node) => {

View File

@@ -61,7 +61,7 @@ const VarReferencePicker: FC<Props> = ({
const [varKindType, setVarKindType] = useState<VarKindType>(defaultVarKindType)
const isConstant = isSupportConstantValue && varKindType === VarKindType.static
const { getTreeLeafNodes, getBeforeNodesInSameBranch } = useWorkflow()
const availableNodes = onlyLeafNodeVar ? getTreeLeafNodes() : getBeforeNodesInSameBranch(nodeId)
const availableNodes = onlyLeafNodeVar ? getTreeLeafNodes(nodeId) : getBeforeNodesInSameBranch(nodeId)
const outputVars = toNodeOutputVars(availableNodes, isChatMode, onlyVarType)
const [open, setOpen] = useState(false)
const hasValue = !isConstant && value.length > 0