vectorvein 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

vectorvein/workflow/nodes/image_generation.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class BackgroundGeneration(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="BackgroundGeneration",
             category="image_generation",
@@ -15,7 +13,7 @@ class BackgroundGeneration(Node):
                 "base_image_url": InputPort(
                     name="base_image_url",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -40,7 +38,7 @@ class BackgroundGeneration(Node):
                 "ref_image_url": InputPort(
                     name="ref_image_url",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                 ),
@@ -104,7 +102,7 @@ class BackgroundGeneration(Node):
 
 
 class DallE(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="DallE",
             category="image_generation",
@@ -169,7 +167,7 @@ class DallE(Node):
 
 
 class Flux1(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Flux1",
             category="image_generation",
@@ -185,7 +183,7 @@ class Flux1(Node):
                 "input_image": InputPort(
                     name="input_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -272,7 +270,7 @@ class Flux1(Node):
 
 
 class Inpainting(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Inpainting",
             category="image_generation",
@@ -282,7 +280,7 @@ class Inpainting(Node):
                 "input_image": InputPort(
                     name="input_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -299,7 +297,7 @@ class Inpainting(Node):
                 "mask_image": InputPort(
                     name="mask_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="return fieldsData.inpainting_method.value === 'custom'",
                     condition_python=lambda ports: ports["inpainting_method"].value == "custom",
@@ -340,7 +338,7 @@ class Inpainting(Node):
 
 
 class Kolors(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Kolors",
             category="image_generation",
@@ -408,7 +406,7 @@ class Kolors(Node):
 
 
 class Pulid(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Pulid",
             category="image_generation",
@@ -418,7 +416,7 @@ class Pulid(Node):
                 "reference_image": InputPort(
                     name="reference_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -514,7 +512,7 @@ class Pulid(Node):
 
 
 class Recraft(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Recraft",
             category="image_generation",
@@ -533,7 +531,7 @@ class Recraft(Node):
                 "image_url": InputPort(
                     name="image_url",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     condition="return fieldsData.generation_type.value === 'image_to_vector'",
@@ -641,7 +639,7 @@ class Recraft(Node):
                 "colors": InputPort(
                     name="colors",
                     port_type=PortType.COLOR,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     condition="return fieldsData.generation_type.value === 'text_to_image'",
                     condition_python=lambda ports: ports["generation_type"].value == "text_to_image",
@@ -649,7 +647,7 @@ class Recraft(Node):
                 "background_color": InputPort(
                     name="background_color",
                     port_type=PortType.COLOR,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     max_count=1,
                     condition="return fieldsData.generation_type.value === 'text_to_image'",
@@ -671,7 +669,7 @@ class Recraft(Node):
 
 
 class GptImage(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="GptImage",
             category="image_generation",
@@ -704,7 +702,7 @@ class GptImage(Node):
                 "image": InputPort(
                     name="image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=False,
                     condition="return fieldsData.action.value === 'edit'",
@@ -713,7 +711,7 @@ class GptImage(Node):
                 "mask": InputPort(
                     name="mask",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".png"],
                     condition="return fieldsData.action.value === 'edit'",
                     condition_python=lambda ports: ports["action"].value == "edit",
@@ -772,7 +770,7 @@ class GptImage(Node):
 
 
 class StableDiffusion(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         special_width_height_models = [
             "stable-diffusion-xl-1024-v0-9",
             "stable-diffusion-xl-1024-v1-0",

vectorvein/workflow/nodes/llms.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class AliyunQwen(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="AliyunQwen",
             category="llms",
@@ -83,7 +81,7 @@ class AliyunQwen(Node):
 
 
 class Baichuan(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Baichuan",
             category="llms",
@@ -174,7 +172,7 @@ class Baichuan(Node):
 
 
 class BaiduWenxin(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="BaiduWenxin",
             category="llms",
@@ -215,7 +213,7 @@ class BaiduWenxin(Node):
 
 
 class ChatGLM(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ChatGLM",
             category="llms",
@@ -303,7 +301,7 @@ class ChatGLM(Node):
 
 
 class Claude(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Claude",
             category="llms",
@@ -355,7 +353,7 @@ class Claude(Node):
 
 
 class Deepseek(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Deepseek",
             category="llms",
@@ -449,7 +447,7 @@ class Deepseek(Node):
 
 
 class Gemini(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Gemini",
             category="llms",
@@ -540,7 +538,7 @@ class Gemini(Node):
 
 
 class LingYiWanWu(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="LingYiWanWu",
             category="llms",
@@ -585,7 +583,7 @@ class LingYiWanWu(Node):
 
 
 class MiniMax(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="MiniMax",
             category="llms",
@@ -673,7 +671,7 @@ class MiniMax(Node):
 
 
 class Moonshot(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Moonshot",
             category="llms",
@@ -763,7 +761,7 @@ class Moonshot(Node):
 
 
 class OpenAI(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="OpenAI",
             category="llms",
@@ -860,7 +858,7 @@ class OpenAI(Node):
 
 
 class XAi(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="XAi",
             category="llms",
@@ -951,7 +949,7 @@ class XAi(Node):
 
 
 class CustomModel(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="CustomModel",
             category="llms",

vectorvein/workflow/nodes/media_editing.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class AudioEditing(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="AudioEditing",
             category="media_editing",
@@ -15,7 +13,7 @@ class AudioEditing(Node):
                 "input_audio": InputPort(
                     name="input_audio",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".mp3", ".wav", ".ogg", ".m4a"],
                     multiple=True,
                     show=True,
@@ -52,26 +50,21 @@ class AudioEditing(Node):
                     port_type=PortType.NUMBER,
                     value=0,
                     condition="return fieldsData.trim.value && (fieldsData.trim_method.value === 'start_duration' || fieldsData.trim_method.value === 'end_duration')",
-                    condition_python=lambda ports: ports["trim"].value
-                    and (
-                        ports["trim_method"].value == "start_duration" or ports["trim_method"].value == "end_duration"
-                    ),
+                    condition_python=lambda ports: ports["trim"].value and (ports["trim_method"].value == "start_duration" or ports["trim_method"].value == "end_duration"),
                 ),
                 "trim_start_time": InputPort(
                     name="trim_start_time",
                     port_type=PortType.INPUT,
                     value="00:00:00",
                     condition="return fieldsData.trim.value && fieldsData.trim_method.value === 'start_end_time'",
-                    condition_python=lambda ports: ports["trim"].value
-                    and ports["trim_method"].value == "start_end_time",
+                    condition_python=lambda ports: ports["trim"].value and ports["trim_method"].value == "start_end_time",
                 ),
                 "trim_end_time": InputPort(
                     name="trim_end_time",
                     port_type=PortType.INPUT,
                     value="00:01:00",
                     condition="return fieldsData.trim.value && fieldsData.trim_method.value === 'start_end_time'",
-                    condition_python=lambda ports: ports["trim"].value
-                    and ports["trim_method"].value == "start_end_time",
+                    condition_python=lambda ports: ports["trim"].value and ports["trim_method"].value == "start_end_time",
                 ),
                 "adjust_volume": InputPort(
                     name="adjust_volume",
@@ -122,16 +115,14 @@ class AudioEditing(Node):
                     port_type=PortType.NUMBER,
                     value=1.0,
                     condition="return fieldsData.adjust_speed.value && fieldsData.speed_adjustment_method.value === 'specified_speed'",
-                    condition_python=lambda ports: ports["adjust_speed"].value
-                    and ports["speed_adjustment_method"].value == "specified_speed",
+                    condition_python=lambda ports: ports["adjust_speed"].value and ports["speed_adjustment_method"].value == "specified_speed",
                 ),
                 "specified_final_length": InputPort(
                     name="specified_final_length",
                     port_type=PortType.NUMBER,
                     value=10,
                     condition="return fieldsData.adjust_speed.value && fieldsData.speed_adjustment_method.value === 'specified_final_length'",
-                    condition_python=lambda ports: ports["adjust_speed"].value
-                    and ports["speed_adjustment_method"].value == "specified_final_length",
+                    condition_python=lambda ports: ports["adjust_speed"].value and ports["speed_adjustment_method"].value == "specified_final_length",
                 ),
                 "adjust_channels": InputPort(
                     name="adjust_channels",
@@ -175,7 +166,7 @@ class AudioEditing(Node):
 
 
 class ImageBackgroundRemoval(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ImageBackgroundRemoval",
             category="media_editing",
@@ -185,7 +176,7 @@ class ImageBackgroundRemoval(Node):
                 "input_image": InputPort(
                     name="input_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -234,7 +225,7 @@ class ImageBackgroundRemoval(Node):
 
 
 class ImageEditing(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ImageEditing",
             category="media_editing",
@@ -244,7 +235,7 @@ class ImageEditing(Node):
                 "input_image": InputPort(
                     name="input_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -317,16 +308,14 @@ class ImageEditing(Node):
                     port_type=PortType.NUMBER,
                     value=1,
                     condition="return fieldsData.crop.value && fieldsData.crop_method.value == 'proportional'",
-                    condition_python=lambda ports: ports["crop"].value
-                    and ports["crop_method"].value == "proportional",
+                    condition_python=lambda ports: ports["crop"].value and ports["crop_method"].value == "proportional",
                 ),
                 "crop_height_ratio": InputPort(
                     name="crop_height_ratio",
                     port_type=PortType.NUMBER,
                     value=1,
                     condition="return fieldsData.crop.value && fieldsData.crop_method.value == 'proportional'",
-                    condition_python=lambda ports: ports["crop"].value
-                    and ports["crop_method"].value == "proportional",
+                    condition_python=lambda ports: ports["crop"].value and ports["crop_method"].value == "proportional",
                 ),
                 "scale": InputPort(
                     name="scale",
@@ -349,16 +338,14 @@ class ImageEditing(Node):
                     port_type=PortType.NUMBER,
                     value=1,
                     condition="return fieldsData.scale.value && fieldsData.scale_method.value == 'proportional_scale'",
-                    condition_python=lambda ports: ports["scale"].value
-                    and ports["scale_method"].value == "proportional_scale",
+                    condition_python=lambda ports: ports["scale"].value and ports["scale_method"].value == "proportional_scale",
                 ),
                 "scale_width": InputPort(
                     name="scale_width",
                     port_type=PortType.NUMBER,
                     value=0,
                     condition="return fieldsData.scale.value && fieldsData.scale_method.value == 'fixed_width_height'",
-                    condition_python=lambda ports: ports["scale"].value
-                    and ports["scale_method"].value == "fixed_width_height",
+                    condition_python=lambda ports: ports["scale"].value and ports["scale_method"].value == "fixed_width_height",
                 ),
                 "scale_height": InputPort(
                     name="scale_height",
@@ -393,7 +380,7 @@ class ImageEditing(Node):
 
 
 class ImageSegmentation(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ImageSegmentation",
             category="media_editing",
@@ -403,7 +390,7 @@ class ImageSegmentation(Node):
                 "input_image": InputPort(
                     name="input_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -457,7 +444,7 @@ class ImageSegmentation(Node):
 
 
 class ImageWatermark(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ImageWatermark",
             category="media_editing",
@@ -467,7 +454,7 @@ class ImageWatermark(Node):
                 "input_image": InputPort(
                     name="input_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     multiple=True,
                     show=True,
@@ -484,7 +471,7 @@ class ImageWatermark(Node):
                 "watermark_image": InputPort(
                     name="watermark_image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="return fieldsData.image_or_text.value == 'image'",
                     condition_python=lambda ports: ports["image_or_text"].value == "image",
@@ -533,11 +520,10 @@ class ImageWatermark(Node):
                 "watermark_text_font_custom": InputPort(
                     name="watermark_text_font_custom",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".otf", ".ttf", ".ttc", ".otc"],
                     condition="return fieldsData.image_or_text.value == 'text' && fieldsData.watermark_text_font.value == 'custom'",
-                    condition_python=lambda ports: ports["image_or_text"].value == "text"
-                    and ports["watermark_text_font"].value == "custom",
+                    condition_python=lambda ports: ports["image_or_text"].value == "text" and ports["watermark_text_font"].value == "custom",
                 ),
                 "watermark_text_font_size": InputPort(
                     name="watermark_text_font_size",
@@ -600,7 +586,7 @@ class ImageWatermark(Node):
 
 
 class VideoEditing(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="VideoEditing",
             category="media_editing",
@@ -610,7 +596,7 @@ class VideoEditing(Node):
                 "input_video": InputPort(
                     name="input_video",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=["video/*"],
                     multiple=True,
                     show=True,
@@ -682,7 +668,7 @@ class VideoEditing(Node):
 
 
 class VideoScreenshot(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="VideoScreenshot",
             category="media_editing",
@@ -692,7 +678,7 @@ class VideoScreenshot(Node):
                 "input_video": InputPort(
                     name="input_video",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=["video/*"],
                     multiple=True,
                     show=True,