vectorvein 0.2.73__tar.gz → 0.2.74__tar.gz

{vectorvein-0.2.73 → vectorvein-0.2.74}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.2.73
+Version: 0.2.74
 Summary: VectorVein Python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.2.73 → vectorvein-0.2.74}/pyproject.toml

@@ -17,7 +17,7 @@ description = "VectorVein Python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.2.73"
+version = "0.2.74"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.73 → vectorvein-0.2.74}/src/vectorvein/workflow/graph/node.py

@@ -88,12 +88,18 @@ class Node:
         if is_output:
             if not self.can_add_output_ports:
                 raise ValueError(f"Node<{self.id}> '{self.type}' does not allow adding output ports")
+
+            if self.has_port(name):
+                raise ValueError(f"Node<{self.id}> '{self.type}' already has a port named '{name}'")
             self.ports[name] = OutputPort(
                 name=name, port_type=port_type, show=show, value=value, options=options, **kwargs
             )
         else:
             if not self.can_add_input_ports:
                 raise ValueError(f"Node<{self.id}> '{self.type}' does not allow adding input ports")
+
+            if self.has_port(name):
+                raise ValueError(f"Node<{self.id}> '{self.type}' already has a port named '{name}'")
             self.ports[name] = InputPort(
                 name=name, port_type=port_type, show=show, value=value, options=options, **kwargs
             )

{vectorvein-0.2.73 → vectorvein-0.2.74}/src/vectorvein/workflow/graph/workflow.py

@@ -69,6 +69,13 @@ class Workflow:
         if not target_node_obj.has_input_port(target_port):
             raise ValueError(f"Target node {target_node_id} has no input port: {target_port}")
 
+        # 确保目标端口是InputPort而不是OutputPort
+        target_port_obj = target_node_obj.ports[target_port]
+        if isinstance(target_port_obj, OutputPort):
+            raise ValueError(
+                f"The target port {target_port} of node {target_node_id} is an output port. OutputPort cannot be a connection target."
+            )
+
         # 检查目标端口是否已有被连接的线
         for edge in self.edges:
             if edge.target == target_node_id and edge.target_handle == target_port:
@@ -328,12 +335,29 @@ class Workflow:
 
         # 创建边
         for edge_data in data.get("edges", []):
+            # 获取目标节点和端口
+            target_node_id = edge_data["target"]
+            target_port_name = edge_data["targetHandle"]
+
+            # 查找目标节点
+            target_node = next((node for node in workflow.nodes if node.id == target_node_id), None)
+            if target_node is None:
+                raise ValueError(f"Target node not found: {target_node_id}")
+
+            # 检查目标端口是否是OutputPort
+            if target_node.has_port(target_port_name):
+                target_port = target_node.ports[target_port_name]
+                if isinstance(target_port, OutputPort):
+                    raise ValueError(
+                        f"The target port {target_port_name} of node {target_node_id} is an output port. OutputPort cannot be a connection target."
+                    )
+
             edge = Edge(
                 id=edge_data["id"],
                 source=edge_data["source"],
                 source_handle=edge_data["sourceHandle"],
-                target=edge_data["target"],
-                target_handle=edge_data["targetHandle"],
+                target=target_node_id,
+                target_handle=target_port_name,
                 animated=edge_data.get("animated", True),
                 type=edge_data.get("type", "default"),
             )