vectorvein 0.2.54__tar.gz → 0.2.55__tar.gz

{vectorvein-0.2.54 → vectorvein-0.2.55}/PKG-INFO

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

{vectorvein-0.2.54 → vectorvein-0.2.55}/pyproject.toml

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

{vectorvein-0.2.54 → vectorvein-0.2.55}/src/vectorvein/workflow/graph/node.py

@@ -118,6 +118,12 @@ class Node:
             "shadow": self.shadow,
         }
 
+    def __str__(self) -> str:
+        return f"Node<{self.id}:{self.type}>"
+
+    def __repr__(self) -> str:
+        return self.__str__()
+
     def has_inputs(self) -> bool:
         for port in self.ports.values():
             if isinstance(port, InputPort):

{vectorvein-0.2.54 → vectorvein-0.2.55}/src/vectorvein/workflow/graph/workflow.py

@@ -4,7 +4,7 @@ from typing import List, Union, Dict, Any, Optional
 from .node import Node
 from .edge import Edge
 from ..utils.layout import layout
-from ..utils.check import WorkflowCheckResult, check_dag, check_ui
+from ..utils.check import WorkflowCheckResult, check_dag, check_ui, check_useless_nodes
 
 
 class Workflow:
@@ -127,10 +127,15 @@ class Workflow:
         """
         dag_check = check_dag(self)  # 检查流程图是否为有向无环图，并检测是否存在孤立节点。
         ui_check = check_ui(self)
+        useless_nodes = check_useless_nodes(self)
 
         # 合并结果
-        result = dag_check
-        result["ui_warnings"] = ui_check
+        result: WorkflowCheckResult = {
+            "no_cycle": dag_check["no_cycle"],
+            "no_isolated_nodes": dag_check["no_isolated_nodes"],
+            "ui_warnings": ui_check,
+            "useless_nodes": useless_nodes,
+        }
 
         return result
 

{vectorvein-0.2.54 → vectorvein-0.2.55}/src/vectorvein/workflow/utils/check.py

@@ -4,6 +4,7 @@ from ..graph.port import InputPort
 
 if TYPE_CHECKING:
     from ..graph.workflow import Workflow
+    from ..graph.node import Node
 
 
 class UIWarning(TypedDict, total=False):
@@ -14,23 +15,24 @@ class UIWarning(TypedDict, total=False):
     has_output_nodes: bool  # 是否存在输出节点
 
 
-class WorkflowCheckResult(TypedDict, total=False):
+class WorkflowCheckResult(TypedDict):
     """工作流检查结果类型。"""
 
     no_cycle: bool  # 工作流是否不包含环
     no_isolated_nodes: bool  # 工作流是否不包含孤立节点
+    useless_nodes: list["Node"]  # 工作流中无用的节点
     ui_warnings: UIWarning  # UI相关警告
 
 
-def check_dag(workflow: "Workflow") -> WorkflowCheckResult:
+def check_dag(workflow: "Workflow"):
     """检查流程图是否为有向无环图，并检测是否存在孤立节点。
 
     Returns:
-        WorkflowCheckResult: 包含检查结果的字典
+        包含检查结果的字典
             - no_cycle (bool): 如果流程图是有向无环图返回 True，否则返回 False
             - no_isolated_nodes (bool): 如果不存在孤立节点返回 True，否则返回 False
     """
-    result: WorkflowCheckResult = {"no_cycle": True, "no_isolated_nodes": True}
+    result = {"no_cycle": True, "no_isolated_nodes": True}
 
     # 过滤掉触发器节点和辅助节点
     trigger_nodes = [
@@ -159,3 +161,22 @@ def check_ui(workflow: "Workflow") -> UIWarning:
     warnings["has_shown_input_ports"] = has_shown_input_ports
 
     return warnings
+
+
+def check_useless_nodes(workflow: "Workflow") -> list["Node"]:
+    """检查工作流中是否存在无用的节点。
+
+    无用的节点定义：
+    1. 节点非 output 类节点，并且节点的输出端口没有任何连线，说明该节点数据不会传给下一个节点或者显示出来。
+    """
+    useless_nodes = []
+    source_nodes = set(edge.source for edge in workflow.edges)
+
+    for node in workflow.nodes:
+        if hasattr(node, "category") and node.category == "outputs":
+            continue
+
+        if node.id not in source_nodes:
+            useless_nodes.append(node)
+
+    return useless_nodes