From a64f3764be96441dc2b4e645c99e60bee0afab4f Mon Sep 17 00:00:00 2001 From: Lucas Oliveira <62335616+lucaseduoli@users.noreply.github.com> Date: Wed, 12 Feb 2025 17:31:24 -0300 Subject: [PATCH] fix: made tool_mode components minimizable (#6237) Fixed count handles function to allow tool_mode to minimize Co-authored-by: anovazzi1 --- src/frontend/src/CustomNodes/helpers/count-handles.ts | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/frontend/src/CustomNodes/helpers/count-handles.ts b/src/frontend/src/CustomNodes/helpers/count-handles.ts index f529d21dc..a0acf78f6 100644 --- a/src/frontend/src/CustomNodes/helpers/count-handles.ts +++ b/src/frontend/src/CustomNodes/helpers/count-handles.ts @@ -9,8 +9,10 @@ export function countHandlesFn(data: NodeDataType): number { ) .map((templateCamp) => { const { template } = data.node!; - if (template[templateCamp]?.input_types) return true; + if (template[templateCamp]?.tool_mode && data.node?.tool_mode) + return false; if (!template[templateCamp]?.show) return false; + if (template[templateCamp]?.input_types) return true; switch (template[templateCamp]?.type) { case "str": case "bool": @@ -18,8 +20,10 @@ export function countHandlesFn(data: NodeDataType): number { case "code": case "prompt": case "file": + case "table": case "int": return false; + default: return true; }