Remove multi input check and support multi input for agent node (mode…

…lscope#29)
zhijianma · Sep 11, 2024 · 3e709eb · 3e709eb
1 parent d3186c4
commit 3e709eb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 21 deletions.
diff --git a/src/agentscope/studio/static/js/workstation.js b/src/agentscope/studio/static/js/workstation.js
@@ -1441,24 +1441,6 @@ function checkConditions() {
         console.log("node", node);
         console.log("node.inputs", node.inputs);
 
-        if (node.inputs) {
-            for (let inputKey in node.inputs) {
-                if (node.class !== "ImageComposition" &&
-                    node.class !== "VideoComposition" &&
-                    node.inputs[inputKey].connections &&
-                    node.inputs[inputKey].connections.length > 1) {
-                    Swal.fire({
-                        title: 'Invalid Connections',
-                        text:
-                            `${node.name} has more than one connection in inputs.`,
-                        icon: 'error',
-                        confirmButtonText: 'Ok'
-                    });
-                    return false;
-                }
-            }
-        }
-
         let nodeElement = document.getElementById('node-' + nodeId);
         const requiredInputs = nodeElement.querySelectorAll('input[data-required="true"]');
 

diff --git a/src/agentscope/web/workstation/workflow_node.py b/src/agentscope/web/workstation/workflow_node.py
@@ -232,7 +232,7 @@ def compile(self) -> dict:
             "inits": f"{self.var_name} = DialogAgent("
             f"{kwarg_converter(self.opt_kwargs)})",
             "execs": f"{DEFAULT_FLOW_VAR} = {self.var_name}"
-            f"({DEFAULT_FLOW_VAR})",
+            f"([{DEFAULT_FLOW_VAR}])",
         }
 
 
@@ -343,7 +343,7 @@ def compile(self) -> dict:
             "inits": f"{self.var_name} = DictDialogAgent("
             f"{kwarg_converter(self.opt_kwargs)})",
             "execs": f"{DEFAULT_FLOW_VAR} = {self.var_name}"
-            f"({DEFAULT_FLOW_VAR})",
+            f"([{DEFAULT_FLOW_VAR}])",
         }
 
 
@@ -397,7 +397,7 @@ def compile(self) -> dict:
             f"({kwarg_converter(self.opt_kwargs)}, service_toolkit"
             f"={self.var_name}_service_toolkit)",
             "execs": f"{DEFAULT_FLOW_VAR} = {self.var_name}"
-            f"({DEFAULT_FLOW_VAR})",
+            f"([{DEFAULT_FLOW_VAR}])",
         }