vectorvein 0.2.72__tar.gz → 0.2.74__tar.gz

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

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

{vectorvein-0.2.72 → 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.72"
+version = "0.2.74"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.72 → 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.72 → vectorvein-0.2.74}/src/vectorvein/workflow/graph/workflow.py

@@ -12,6 +12,7 @@ from ..utils.check import (
     check_useless_nodes,
     check_required_ports,
     check_override_ports,
+    check_output_nodes_with_no_inputs,
 )
 
 
@@ -68,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:
@@ -138,6 +146,7 @@ class Workflow:
         useless_nodes = check_useless_nodes(self)
         required_ports = check_required_ports(self)
         override_ports = check_override_ports(self)
+        output_nodes_with_no_inputs = check_output_nodes_with_no_inputs(self)
 
         # 合并结果
         result: WorkflowCheckResult = {
@@ -147,6 +156,7 @@ class Workflow:
             "useless_nodes": useless_nodes,
             "required_ports": required_ports,
             "override_ports": override_ports,
+            "output_nodes_with_no_inputs": output_nodes_with_no_inputs,
         }
 
         return result
@@ -325,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"),
             )

{vectorvein-0.2.72 → vectorvein-0.2.74}/src/vectorvein/workflow/utils/check.py

@@ -25,6 +25,7 @@ class WorkflowCheckResult(TypedDict):
     ui_warnings: UIWarning  # UI相关警告
     required_ports: list[tuple["Node", "Port"]]  # 未连接的必填端口
     override_ports: list[tuple["Node", "Port"]]  # 被覆盖的端口
+    output_nodes_with_no_inputs: list["Node"]  # 输出端口没有输入端口连接的节点
 
 
 def check_dag(workflow: "Workflow"):
@@ -259,3 +260,30 @@ def check_override_ports(workflow: "Workflow") -> list[tuple["Node", "Port"]]:
                     override_ports.append((node, port))
 
     return override_ports
+
+
+def check_output_nodes_with_no_inputs(workflow: "Workflow") -> list["Node"]:
+    """检查工作流中是否存在输出端口没有输入端口连接的节点。"""
+    output_nodes_with_no_inputs = []
+
+    # 找出所有连接的目标节点和端口
+    connected_targets = {(edge.target, edge.target_handle) for edge in workflow.edges}
+
+    # 遍历所有节点
+    for node in workflow.nodes:
+        # 只检查输出类节点
+        if hasattr(node, "category") and node.category == "outputs":
+            # 检查该节点是否有任何连接到其输入端口的边
+            has_input_connection = False
+
+            for port_name, port in node.ports.items():
+                if isinstance(port, InputPort):
+                    if (node.id, port_name) in connected_targets:
+                        has_input_connection = True
+                        break
+
+            # 如果该输出节点没有任何输入连接，则添加到结果列表
+            if not has_input_connection:
+                output_nodes_with_no_inputs.append(node)
+
+    return output_nodes_with_no_inputs