Fix display bug in workstation for group nodes (#325)

modelscope · Jul 8, 2024 · 8af3524 · 8af3524
1 parent e73b9b8
commit 8af3524
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/agentscope/studio/static/js/workstation.js b/src/agentscope/studio/static/js/workstation.js
@@ -962,11 +962,15 @@ function hideShowGroupNodes(groupId, show) {
         groupInfo.data.elements.forEach(elementNodeId => {
             const elementNode = document.getElementById(`node-${elementNodeId}`);
             const childNodeInfo = editor.getNodeFromId(elementNodeId);
+            const contentBox = elementNode.querySelector('.box') ||
+                elementNode.querySelector('.box-highlight');
             if (elementNode) {
                 elementNode.style.display = show ? '' : 'none';
             }
             if (childNodeInfo.class === 'GROUP') {
-                hideShowGroupNodes(elementNodeId, show);
+                if (!show || (contentBox && !contentBox.classList.contains('hidden'))) {
+                    hideShowGroupNodes(elementNodeId, show);
+                }
             }
         });
     }
@@ -2037,4 +2041,4 @@ function showSurveyModal() {
 
 function hideSurveyModal() {
     document.getElementById("surveyModal").style.display = "none";
-}
+}