vectorvein 0.1.85__tar.gz → 0.1.86__tar.gz

{vectorvein-0.1.85 → vectorvein-0.1.86}/PKG-INFO

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

{vectorvein-0.1.85 → vectorvein-0.1.86}/pyproject.toml

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

{vectorvein-0.1.85 → vectorvein-0.1.86}/src/vectorvein/workflow/graph/node.py

@@ -80,3 +80,12 @@ class Node:
             if isinstance(port, InputPort):
                 return True
         return False
+
+    def has_port(self, port_name: str) -> bool:
+        return port_name in self.ports
+
+    def has_input_port(self, port_name: str) -> bool:
+        return port_name in self.ports and isinstance(self.ports[port_name], InputPort)
+
+    def has_output_port(self, port_name: str) -> bool:
+        return port_name in self.ports and isinstance(self.ports[port_name], OutputPort)

{vectorvein-0.1.85 → vectorvein-0.1.86}/src/vectorvein/workflow/graph/workflow.py

@@ -26,15 +26,39 @@ class Workflow:
         target_node: Union[str, Node],
         target_port: str,
     ):
+        # 获取源节点ID
         if isinstance(source_node, Node):
             source_node_id = source_node.id
         else:
             source_node_id = source_node
+
+        # 获取目标节点ID
         if isinstance(target_node, Node):
             target_node_id = target_node.id
         else:
             target_node_id = target_node
 
+        # 检查源节点是否存在
+        source_node_exists = any(node.id == source_node_id for node in self.nodes)
+        if not source_node_exists:
+            raise ValueError(f"源节点不存在: {source_node_id}")
+
+        # 检查目标节点是否存在
+        target_node_exists = any(node.id == target_node_id for node in self.nodes)
+        if not target_node_exists:
+            raise ValueError(f"目标节点不存在: {target_node_id}")
+
+        # 检查源节点的端口是否存在
+        source_node_obj = next(node for node in self.nodes if node.id == source_node_id)
+        if not source_node_obj.has_output_port(source_port):
+            raise ValueError(f"源节点 {source_node_id} 不存在输出端口: {source_port}")
+
+        # 检查目标节点的端口是否存在
+        target_node_obj = next(node for node in self.nodes if node.id == target_node_id)
+        if not target_node_obj.has_input_port(target_port):
+            raise ValueError(f"目标节点 {target_node_id} 不存在输入端口: {target_port}")
+
+        # 创建并添加边
         edge_id = f"vueflow__edge-{source_node_id}{source_port}-{target_node_id}{target_port}"
         edge = Edge(edge_id, source_node_id, source_port, target_node_id, target_port)
         self.add_edge(edge)