vectorvein 0.2.61__py3-none-any.whl → 0.2.63__py3-none-any.whl

vectorvein/workflow/graph/node.py

@@ -1,10 +1,14 @@
 import uuid
-from typing import Dict, Any, Optional, List, Union
+from typing import Dict, Any, Optional, List, Union, TYPE_CHECKING
 
-from .port import PortType, Port, InputPort, OutputPort
+from .port import InputPort, OutputPort
 
 
-class PortsDict(Dict[str, Port]):
+if TYPE_CHECKING:
+    from .port import PortType, Port
+
+
+class PortsDict(Dict[str, "Port"]):
     """自定义字典类，用于管理节点的端口，并强制执行端口添加权限。"""
 
     def __init__(self, owner_node: "Node", *args, **kwargs):
@@ -12,7 +16,7 @@ class PortsDict(Dict[str, Port]):
         self._owner_node = owner_node
         self._initializing = True
 
-    def __setitem__(self, key: str, value: Port) -> None:
+    def __setitem__(self, key: str, value: "Port") -> None:
         if self._initializing or key in self:
             super().__setitem__(key, value)
             return
@@ -40,7 +44,7 @@ class Node:
         category: str,
         task_name: str,
         description: str = "",
-        ports: Optional[Dict[str, Any]] = None,
+        ports: Optional[Dict[str, "Port"]] = None,
         node_id: Optional[str] = None,
         position: Optional[Dict[str, float]] = None,
         seleted_workflow_title: str = "",
@@ -74,7 +78,7 @@ class Node:
     def add_port(
         self,
         name: str,
-        port_type: Union[PortType, str],
+        port_type: Union["PortType", str],
         show: bool = False,
         value: Any = None,
         options: Optional[List[Any]] = None,

vectorvein/workflow/utils/json_to_code.py

@@ -142,8 +142,8 @@ def generate_python_code(
         code.append("")
 
     # 生成节点实例化代码
-    for info in node_instances.values():
-        code.append(f"{info['var_name']} = {info['type']}()")
+    for node_info in node_instances.values():
+        code.append(f"{node_info['var_name']} = {node_info['type']}()")
 
     code.append("")
     code.append("workflow = Workflow()")
@@ -153,8 +153,8 @@ def generate_python_code(
     code.append(f"workflow.add_nodes([\n{'\n'.join(node_list)}\n])")
 
     code.append("")
-    for info in node_instances.values():
-        for port in info["add_ports"]:
+    for node_info in node_instances.values():
+        for port in node_info["add_ports"]:
             params = [
                 f"name={to_python_str(port['name'])}",
                 f"port_type={to_python_str(port['field_type'])}",
@@ -166,13 +166,15 @@ def generate_python_code(
                 params.append(f"options={port['options']}")
             if port.get("is_output"):
                 params.append(f"is_output={bool(port['is_output'])}")
-            code.append(f"{info['var_name']}.add_port({', '.join(params)})")
+            if node_info["type"] == "ProgrammingFunction":
+                params.append(f'field_type="{port["type"]}"')
+            code.append(f"{node_info['var_name']}.add_port({', '.join(params)})")
 
-        for port_name in info["show_ports"]:
-            code.append(f"{info['var_name']}.ports['{port_name}'].show = True")
+        for port_name in node_info["show_ports"]:
+            code.append(f"{node_info['var_name']}.ports['{port_name}'].show = True")
 
-        for port in info["values"]:
-            code.append(f"{info['var_name']}.ports['{port['name']}'].value = {to_python_str(port['value'])}")
+        for port in node_info["values"]:
+            code.append(f"{node_info['var_name']}.ports['{port['name']}'].value = {to_python_str(port['value'])}")
 
     code.append("")
 

{vectorvein-0.2.61.dist-info → vectorvein-0.2.63.dist-info}/METADATA

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

{vectorvein-0.2.61.dist-info → vectorvein-0.2.63.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.61.dist-info/METADATA,sha256=BjRmW_3pGZXQGqEXCr6ga7o-jHKxUFyJ_Vg56AmNXdY,4570
-vectorvein-0.2.61.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.61.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.63.dist-info/METADATA,sha256=bp_MfR1VH9x50C2uHRYOeHF77enWRykynnuWFX8eJe4,4570
+vectorvein-0.2.63.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.63.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,7 +42,7 @@ 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=P3iL5kJ0Nabd9NPOcbbznD71OBUtsSUcZ_zJ6a2cnIc,5559
+vectorvein/workflow/graph/node.py,sha256=WtO4MkeMQr59zNl9N2ZoYiCZcVKxebSp1rKMPONmfqk,5628
 vectorvein/workflow/graph/port.py,sha256=HcinzQqNP7ysTvBmi3c4iaWne8nV6m-BpFFX0jTrMIE,7122
 vectorvein/workflow/graph/workflow.py,sha256=f5KAQUXTKii7UFYmSZeZZQ7JCKKPmJlUg0iFAfBEJQE,6366
 vectorvein/workflow/nodes/__init__.py,sha256=dWrWtL3q0Vsn-MLgJ7gNgLCrwZ5BrqjrN2QFPNeBMuc,3240
@@ -62,6 +62,6 @@ vectorvein/workflow/nodes/vector_db.py,sha256=p9AT_E8ASbcYHZqHYTCIGvqkIqzxaFM4Ux
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=FB0bTimkk___p3Z5UwQx2YarJyQCc45jjnbXbgGA_qw,5640
 vectorvein/workflow/utils/check.py,sha256=Oj_S5WQf4_Fr_ro3ipjZt9unKFSFcuwZrrSmrS9kVLE,10193
-vectorvein/workflow/utils/json_to_code.py,sha256=-RqISoBXIuVapWeafS4u55GWeUUuIGqCM64D4VIb-9E,6084
+vectorvein/workflow/utils/json_to_code.py,sha256=U9EF62D270Ff1D8JKH3Am_jwsiDkSLeC41MZz7rZ8hg,6257
 vectorvein/workflow/utils/layout.py,sha256=j0bRD3uaXu40xCS6U6BGahBI8FrHa5MiF55GbTrZ1LM,4565
-vectorvein-0.2.61.dist-info/RECORD,,
+vectorvein-0.2.63.dist-info/RECORD,,