vectorvein 0.1.85__py3-none-any.whl → 0.1.87__py3-none-any.whl

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/workflow/graph/port.py

@@ -43,7 +43,7 @@ class Port:
         self.port_type = port_type
         self.required = required
         self.show = show
-        self.value = value
+        self._value = value
         self.options = options
         self.field_type = field_type
         self.is_output = is_output
@@ -66,7 +66,7 @@ class Port:
             "field_type": self.port_type.value if isinstance(self.port_type, PortType) else self.port_type,
             "required": self.required,
             "show": self.show,
-            "value": self.value,
+            "value": self._value,
             "options": self.options,
             "type": self.field_type,
             "is_output": self.is_output,
@@ -82,6 +82,17 @@ class Port:
             "list": self.list,
         }
 
+    @property
+    def value(self) -> Any:
+        return self._value
+
+    @value.setter
+    def value(self, value: Any) -> None:
+        if self.options:
+            if value not in map(lambda x: x["value"], self.options):
+                raise ValueError(f"Value `{value}` is not in Port `{self.name}` options {self.options}")
+        self._value = value
+
 
 class InputPort(Port):
     def __init__(

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)

{vectorvein-0.1.85.dist-info → vectorvein-0.1.87.dist-info}/METADATA

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

{vectorvein-0.1.85.dist-info → vectorvein-0.1.87.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.1.85.dist-info/METADATA,sha256=liaNevO0Gdtzyor4mS1Wuo5Ma6LmkgexsYSKNNhWf3M,641
-vectorvein-0.1.85.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.1.85.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.1.87.dist-info/METADATA,sha256=qaCzwLhxB8o0buhZ-_94iqCMM8tITVVbZQMGt_xi_Lk,641
+vectorvein-0.1.87.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.1.87.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/__init__.py,sha256=omQuG4PRRPNflSAgtdU--rwsWG6vMpwMEyIGZyFVHVQ,18596
 vectorvein/chat_clients/anthropic_client.py,sha256=PGIKldH4FnGrqozoY_FZ6LqhDHC-jY7NF5J1F1zT2Ok,38257
@@ -34,9 +34,9 @@ vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/utilities/media_processing.py,sha256=CTRq-lGlFkFgP_FSRhNwF_qUgmOrXPf2_1Ok9HY42_g,5887
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
 vectorvein/workflow/graph/edge.py,sha256=xLZEJmBjAfVB53cd7CuRcKhgE6QqXv9nz32wJn8cmyk,1064
-vectorvein/workflow/graph/node.py,sha256=nxDq4pTFcsm2x9MmNmZxghhc-HduSLRwH3Ct2i8bZ3o,2871
-vectorvein/workflow/graph/port.py,sha256=kv0YbAPqe9Hh9QPSoiB2PUTTWQT1xIdNeVD50OkWTzA,5684
-vectorvein/workflow/graph/workflow.py,sha256=BmS351-WSeHQ02Kh0jOQpxJbNgZ-AHwW0mSQLXcKceU,2805
+vectorvein/workflow/graph/node.py,sha256=A3M_GghrSju1D3xc_HtPdGyr-7XSkplGPKJveOUiIF4,3256
+vectorvein/workflow/graph/port.py,sha256=Q6HmI2cUi6viJ98ec6-MmMPMRtKS1-OgaudP3LMwVLA,6054
+vectorvein/workflow/graph/workflow.py,sha256=uh8JzTbvaucKryvFUj3nAswtdOtaOw2Z4YfD9Q6r77s,3973
 vectorvein/workflow/nodes/__init__.py,sha256=jd4O27kIJdOtkij1FYZ6aJnJy2OQa7xtL1r-Yv8ylO0,3103
 vectorvein/workflow/nodes/audio_generation.py,sha256=ht2S0vnd0mIAt6FBaSWlADGbb7f_1DAySYrgYnvZT1Q,5726
 vectorvein/workflow/nodes/control_flows.py,sha256=Zc_uWuroYznLrU-BZCncyzvejC-zFl6EuN_VP8oq5mY,6573
@@ -54,4 +54,4 @@ vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs0
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=LsqomfXfqrXfHJDO1cl0Ox48f4St7X_SL12DSbAMSOw,5415
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.1.85.dist-info/RECORD,,
+vectorvein-0.1.87.dist-info/RECORD,,