vectorvein 0.2.35__py3-none-any.whl → 0.2.37__py3-none-any.whl

vectorvein/workflow/graph/node.py

@@ -10,10 +10,11 @@ class PortsDict(Dict[str, Port]):
     def __init__(self, owner_node: "Node", *args, **kwargs):
         super().__init__(*args, **kwargs)
         self._owner_node = owner_node
+        self._initializing = True  # 初始化标志
 
     def __setitem__(self, key: str, value: Port) -> None:
-        # 如果端口已存在，允许更新
-        if key in self:
+        # 初始化阶段或端口已存在时，允许直接添加/更新
+        if self._initializing or key in self:
             super().__setitem__(key, value)
             return
 
@@ -29,6 +30,10 @@ class PortsDict(Dict[str, Port]):
 
         super().__setitem__(key, value)
 
+    def finish_initialization(self):
+        """结束初始化阶段"""
+        self._initializing = False
+
 
 class Node:
     def __init__(
@@ -59,6 +64,9 @@ class Node:
         if ports:
             for name, port in ports.items():
                 self.ports[name] = port
+        # 结束初始化阶段
+        self.ports.finish_initialization()
+
         self.position: Dict[str, float] = position or {"x": 0, "y": 0}
         self.seleted_workflow_title: str = seleted_workflow_title
         self.is_template: bool = is_template

vectorvein/workflow/graph/workflow.py

@@ -48,28 +48,28 @@ class Workflow:
         # 检查源节点是否存在
         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}")
+            raise ValueError(f"Source node not found: {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}")
+            raise ValueError(f"Target node not found: {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}")
+            raise ValueError(f"Source node {source_node_id} has no output port: {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}")
+            raise ValueError(f"Target node {target_node_id} has no input port: {target_port}")
 
         # 检查目标端口是否已有被连接的线
         for edge in self.edges:
             if edge.target == target_node_id and edge.target_handle == target_port:
                 raise ValueError(
-                    f"目标节点 {target_node_id} 的输入端口 {target_port} 已经被连接: {edge.source}({edge.source_handle}) → {edge.target}({edge.target_handle})"
+                    f"The input port {target_port} of the target node {target_node_id} is already connected: {edge.source}({edge.source_handle}) → {edge.target}({edge.target_handle})"
                 )
 
         # 创建并添加边

vectorvein/workflow/nodes/tools.py

@@ -290,6 +290,7 @@ class ProgrammingFunction(Node):
                     field_type="textarea",
                 ),
             },
+            can_add_input_ports=True,
         )
 
 

{vectorvein-0.2.35.dist-info → vectorvein-0.2.37.dist-info}/METADATA

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

{vectorvein-0.2.35.dist-info → vectorvein-0.2.37.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.35.dist-info/METADATA,sha256=cIifHIbhWkmoBq64isEQVHa0BoZKKCp1aXuR3IAyKCw,4570
-vectorvein-0.2.35.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.35.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.37.dist-info/METADATA,sha256=fJpDR72JlBhl26sqEpD_Ejvr6ukbHTXBPjUiMfH1PAQ,4570
+vectorvein-0.2.37.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.37.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/api/__init__.py,sha256=lfY-XA46fgD2iIZTU0VYP8i07AwA03Egj4Qua0vUKrQ,738
 vectorvein/api/client.py,sha256=xF-leKDQzVyyy9FnIRaz0k4eElYW1XbbzeRLcpnyk90,33047
@@ -42,9 +42,9 @@ vectorvein/utilities/media_processing.py,sha256=7KtbLFzOYIn1e9QTN9G6C76NH8CBlV9k
 vectorvein/utilities/rate_limiter.py,sha256=dwolIUVw2wP83Odqpx0AAaE77de1GzxkYDGH4tM_u_4,10300
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
 vectorvein/workflow/graph/edge.py,sha256=1ckyyjCue_PLm7P1ItUfKOy6AKkemOpZ9m1WJ8UXIHQ,1072
-vectorvein/workflow/graph/node.py,sha256=9eBS4hOjpiIOdO9ZmG6ctbRtsZeDO29SJqsu_3ELFFc,5033
+vectorvein/workflow/graph/node.py,sha256=U3LAq01wOaDANVUuYWc2BhIq-Pl2-33FirJZkkNL_wA,5329
 vectorvein/workflow/graph/port.py,sha256=Q6HmI2cUi6viJ98ec6-MmMPMRtKS1-OgaudP3LMwVLA,6054
-vectorvein/workflow/graph/workflow.py,sha256=l0rSDZeSd6OtHMwxzhIENfOKS0fJKKB4JUow-dV-LUI,8610
+vectorvein/workflow/graph/workflow.py,sha256=K-1-FMfHO49Ww3sHKDOr_sD8GH-i36ylS7qtyJ-Vivg,8617
 vectorvein/workflow/nodes/__init__.py,sha256=a1fsitA0mf1UlDOSEz6QjSGjiFsvsQcuDE-p2q9SlhM,3181
 vectorvein/workflow/nodes/audio_generation.py,sha256=ZRFZ_ycMTSJ2LKmekctagQdJYTl-3q4TNOIKETpS9AM,5870
 vectorvein/workflow/nodes/control_flows.py,sha256=l8CjFQlsGV3fNGM6SVzS1Kz361K1xDv1fGT7acuDXuU,6613
@@ -56,10 +56,10 @@ vectorvein/workflow/nodes/media_processing.py,sha256=jkSP287gyhvF7hqY8Nohdc1Rdn-
 vectorvein/workflow/nodes/output.py,sha256=_UQxiddHtGv2rkjhUFE-KDgrjnh0AGJQJyq9-4Aji5A,12567
 vectorvein/workflow/nodes/relational_db.py,sha256=zfzUhV25TpZGhkIzO18PmAT5xhcsJC4AXKy0zyA05w8,5408
 vectorvein/workflow/nodes/text_processing.py,sha256=MRo_-oaC65hbzMxm7TYoeiS3rgvqh9y_Rny5RCfoATE,8342
-vectorvein/workflow/nodes/tools.py,sha256=uMlqJQI9oel3pG4WYrZdpSmWWy2pDQkAIzqOv5z3tB4,13811
+vectorvein/workflow/nodes/tools.py,sha256=n3gF7r5FqNz4Xg0nXYo5kl4ai_g4tWSRHqfnGrKJAls,13850
 vectorvein/workflow/nodes/triggers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs076z-7dlQQ,5712
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=BhJBX1AZH7-22Gu95Ox4qJqmH5DU-m4dbUb5N5DTA-M,5559
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.2.35.dist-info/RECORD,,
+vectorvein-0.2.37.dist-info/RECORD,,