Frontend for complete if empty (#1676)

This commit is contained in:
Shuchang Zheng
2025-01-30 04:57:01 +08:00
committed by GitHub
parent 5e4b86c828
commit 04e2a2b063
5 changed files with 30 additions and 1 deletions

View File

@@ -20,6 +20,7 @@ import { WorkflowBlockIcon } from "../WorkflowBlockIcon";
import type { LoopNode } from "./types"; import type { LoopNode } from "./types";
import { useState } from "react"; import { useState } from "react";
import { useIsFirstBlockInWorkflow } from "../../hooks/useIsFirstNodeInWorkflow"; import { useIsFirstBlockInWorkflow } from "../../hooks/useIsFirstNodeInWorkflow";
import { Checkbox } from "@/components/ui/checkbox";
function LoopNode({ id, data }: NodeProps<LoopNode>) { function LoopNode({ id, data }: NodeProps<LoopNode>) {
const { updateNodeData } = useReactFlow(); const { updateNodeData } = useReactFlow();
@@ -125,6 +126,27 @@ function LoopNode({ id, data }: NodeProps<LoopNode>) {
}} }}
/> />
</div> </div>
<div className="space-y-2">
<div className="space-y-2">
<div className="flex gap-4">
<div className="flex gap-2">
<Label className="text-xs text-slate-300">
Complete if Empty
</Label>
<HelpTooltip content="When checked, this block will successfully complete when the loop value is an empty list" />
</div>
<Checkbox
checked={data.completeIfEmpty}
disabled={!data.editable}
onCheckedChange={(checked) => {
updateNodeData(id, {
completeIfEmpty: checked,
});
}}
/>
</div>
</div>
</div>
</div> </div>
</div> </div>
</div> </div>

View File

@@ -4,6 +4,7 @@ import { NodeBaseData } from "../types";
export type LoopNodeData = NodeBaseData & { export type LoopNodeData = NodeBaseData & {
loopValue: string; loopValue: string;
loopVariableReference: string; loopVariableReference: string;
completeIfEmpty: boolean;
}; };
export type LoopNode = Node<LoopNodeData, "loop">; export type LoopNode = Node<LoopNodeData, "loop">;
@@ -13,6 +14,7 @@ export const loopNodeDefaultData: LoopNodeData = {
label: "", label: "",
loopValue: "", loopValue: "",
loopVariableReference: "", loopVariableReference: "",
completeIfEmpty: false,
continueOnFailure: false, continueOnFailure: false,
} as const; } as const;

View File

@@ -144,7 +144,7 @@ function layout(
const loopNodeWidth = 600; // 600 px const loopNodeWidth = 600; // 600 px
const layouted = layoutUtil(childNodes, childEdges, { const layouted = layoutUtil(childNodes, childEdges, {
marginx: (loopNodeWidth - maxChildWidth) / 2, marginx: (loopNodeWidth - maxChildWidth) / 2,
marginy: 200, marginy: 225,
}); });
loopNodeChildren[index] = layouted.nodes; loopNodeChildren[index] = layouted.nodes;
}); });
@@ -381,6 +381,7 @@ function convertToNode(
...commonData, ...commonData,
loopValue: block.loop_over?.key ?? "", loopValue: block.loop_over?.key ?? "",
loopVariableReference: loopVariableReference, loopVariableReference: loopVariableReference,
completeIfEmpty: block.complete_if_empty,
}, },
}; };
} }
@@ -1112,6 +1113,7 @@ function getWorkflowBlocksUtil(
continue_on_failure: node.data.continueOnFailure, continue_on_failure: node.data.continueOnFailure,
loop_blocks: getOrderedChildrenBlocks(nodes, edges, node.id), loop_blocks: getOrderedChildrenBlocks(nodes, edges, node.id),
loop_variable_reference: node.data.loopVariableReference, loop_variable_reference: node.data.loopVariableReference,
complete_if_empty: node.data.completeIfEmpty,
}, },
]; ];
} }
@@ -1628,6 +1630,7 @@ function convertBlocksToBlockYAML(
loop_over_parameter_key: block.loop_over?.key ?? "", loop_over_parameter_key: block.loop_over?.key ?? "",
loop_blocks: convertBlocksToBlockYAML(block.loop_blocks), loop_blocks: convertBlocksToBlockYAML(block.loop_blocks),
loop_variable_reference: block.loop_variable_reference, loop_variable_reference: block.loop_variable_reference,
complete_if_empty: block.complete_if_empty,
}; };
return blockYaml; return blockYaml;
} }

View File

@@ -236,6 +236,7 @@ export type ForLoopBlock = WorkflowBlockBase & {
loop_over: WorkflowParameter; loop_over: WorkflowParameter;
loop_blocks: Array<WorkflowBlock>; loop_blocks: Array<WorkflowBlock>;
loop_variable_reference: string | null; loop_variable_reference: string | null;
complete_if_empty: boolean;
}; };
export type CodeBlock = WorkflowBlockBase & { export type CodeBlock = WorkflowBlockBase & {

View File

@@ -265,6 +265,7 @@ export type ForLoopBlockYAML = BlockYAMLBase & {
loop_over_parameter_key?: string; loop_over_parameter_key?: string;
loop_blocks: Array<BlockYAML>; loop_blocks: Array<BlockYAML>;
loop_variable_reference: string | null; loop_variable_reference: string | null;
complete_if_empty: boolean;
}; };
export type PDFParserBlockYAML = BlockYAMLBase & { export type PDFParserBlockYAML = BlockYAMLBase & {