vectorvein 0.2.72__tar.gz → 0.2.73__tar.gz

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

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

{vectorvein-0.2.72 → vectorvein-0.2.73}/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.73"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.72 → vectorvein-0.2.73}/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,
 )
 
 
@@ -138,6 +139,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 +149,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

{vectorvein-0.2.72 → vectorvein-0.2.73}/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