vectorvein 0.2.31__tar.gz → 0.2.33__tar.gz

{vectorvein-0.2.31 → vectorvein-0.2.33}/PKG-INFO

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

{vectorvein-0.2.31 → vectorvein-0.2.33}/pyproject.toml

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

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/chat_clients/utils.py

@@ -595,7 +595,20 @@ def transform_from_openai_message(message: ChatCompletionMessageParam, backend:
                     )
             return {"role": role, "content": formatted_content}
         else:
-            return {"role": role, "content": content}
+            if tool_calls:
+                formatted_content = [{"type": "text", "text": content}] if content else []
+                for tool_call in tool_calls:
+                    formatted_content.append(
+                        {
+                            "type": "tool_use",
+                            "id": tool_call["id"],
+                            "name": tool_call["function"]["name"],
+                            "input": json.loads(tool_call["function"]["arguments"]),
+                        }
+                    )
+                return {"role": role, "content": formatted_content}
+            else:
+                return {"role": role, "content": content}
     else:
         return message  # 对于其他后端，保持原样
 

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/graph/node.py

@@ -17,6 +17,8 @@ class Node:
         seleted_workflow_title: str = "",
         is_template: bool = False,
         initialized: bool = False,
+        can_add_input_ports: bool = False,
+        can_add_output_ports: bool = False,
     ) -> None:
         self.id: str = node_id or str(uuid.uuid4())
         self.type: str = node_type
@@ -31,6 +33,8 @@ class Node:
         self.ignored: bool = False
         self.lock: bool = False
         self.shadow: bool = False
+        self.can_add_input_ports: bool = can_add_input_ports
+        self.can_add_output_ports: bool = can_add_output_ports
 
     def add_port(
         self,
@@ -43,10 +47,14 @@ class Node:
         **kwargs,
     ):
         if is_output:
+            if not self.can_add_output_ports:
+                raise ValueError(f"Node<{self.id}> '{self.type}' does not allow adding output ports")
             self.ports[name] = OutputPort(
                 name=name, port_type=port_type, show=show, value=value, options=options, **kwargs
             )
         else:
+            if not self.can_add_input_ports:
+                raise ValueError(f"Node<{self.id}> '{self.type}' does not allow adding input ports")
             self.ports[name] = InputPort(
                 name=name, port_type=port_type, show=show, value=value, options=options, **kwargs
             )

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/audio_generation.py

@@ -17,6 +17,7 @@ class MinimaxMusicGeneration(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".wav", ".mp3"],
+                    multiple=True,
                 ),
                 "purpose": InputPort(
                     name="purpose",
@@ -99,12 +100,14 @@ class SoundEffects(Node):
                     port_type=PortType.TEXTAREA,
                     value="",
                     max_length=50,
+                    multiple=True,
                 ),
                 "video": InputPort(
                     name="video",
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".mp4", ".mov", ".webm", ".m4v", ".gif"],
+                    multiple=True,
                 ),
                 "length": InputPort(
                     name="length",
@@ -139,6 +142,7 @@ class Tts(Node):
                     name="text",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "output_type": OutputPort(
                     name="output_type",

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/control_flows.py

@@ -149,6 +149,7 @@ class JsonProcess(Node):
                 ),
                 "output": OutputPort(),
             },
+            can_add_output_ports=True,
         )
 
 

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/image_generation.py

@@ -40,22 +40,26 @@ class BackgroundGeneration(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
+                    multiple=True,
                 ),
                 "ref_prompt": InputPort(
                     name="ref_prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
                     max_length=200,
+                    multiple=True,
                 ),
                 "image_title": InputPort(
                     name="title",
                     port_type=PortType.INPUT,
                     value="",
+                    multiple=True,
                 ),
                 "image_sub_title": InputPort(
                     name="sub_title",
                     port_type=PortType.INPUT,
                     value="",
+                    multiple=True,
                 ),
                 "n": InputPort(
                     name="n",
@@ -109,6 +113,7 @@ class DallE(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "model": InputPort(
                     name="model",
@@ -173,6 +178,7 @@ class Flux1(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "model": InputPort(
                     name="model",
@@ -184,6 +190,7 @@ class Flux1(Node):
                         {"value": "FLUX.1 [pro]", "label": "FLUX.1 [pro]"},
                         {"value": "FLUX.1 [pro] ultra", "label": "FLUX.1 [pro] ultra"},
                     ],
+                    multiple=True,
                 ),
                 "width": InputPort(
                     name="width",
@@ -258,6 +265,7 @@ class Inpainting(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
+                    multiple=True,
                 ),
                 "inpainting_method": InputPort(
                     name="inpainting_method",
@@ -274,6 +282,7 @@ class Inpainting(Node):
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="return fieldsData.inpainting_method.value === 'custom'",
+                    multiple=True,
                 ),
                 "prompt": InputPort(
                     name="prompt",
@@ -320,11 +329,13 @@ class Kolors(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "negative_prompt": InputPort(
                     name="negative_prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "width": InputPort(
                     name="width",
@@ -390,11 +401,13 @@ class Pulid(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
+                    multiple=True,
                 ),
                 "prompt": InputPort(
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "negative_prompt": InputPort(
                     name="negative_prompt",
@@ -499,7 +512,7 @@ class Recraft(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
-                    multiple=False,
+                    multiple=True,
                     condition="return fieldsData.generation_type.value === 'image_to_vector'",
                 ),
                 "prompt": InputPort(
@@ -507,6 +520,7 @@ class Recraft(Node):
                     port_type=PortType.TEXTAREA,
                     value="",
                     condition="return fieldsData.generation_type.value === 'text_to_image'",
+                    multiple=True,
                 ),
                 "base_style": InputPort(
                     name="base_style",
@@ -519,6 +533,7 @@ class Recraft(Node):
                         {"value": "vector_illustration", "label": "vector_illustration"},
                     ],
                     condition="return fieldsData.generation_type.value === 'text_to_image'",
+                    multiple=True,
                 ),
                 "substyle_realistic_image": InputPort(
                     name="substyle_realistic_image",
@@ -535,6 +550,7 @@ class Recraft(Node):
                         {"value": "motion_blur", "label": "motion_blur"},
                     ],
                     condition="return fieldsData.generation_type.value === 'text_to_image' && fieldsData.base_style.value === 'realistic_image'",
+                    multiple=True,
                 ),
                 "substyle_digital_illustration": InputPort(
                     name="substyle_digital_illustration",
@@ -553,6 +569,7 @@ class Recraft(Node):
                         {"value": "2d_art_poster_2", "label": "2d_art_poster_2"},
                     ],
                     condition="return fieldsData.generation_type.value === 'text_to_image' && fieldsData.base_style.value === 'digital_illustration'",
+                    multiple=True,
                 ),
                 "substyle_vector_illustration": InputPort(
                     name="substyle_vector_illustration",
@@ -566,6 +583,7 @@ class Recraft(Node):
                         {"value": "linocut", "label": "linocut"},
                     ],
                     condition="return fieldsData.generation_type.value === 'text_to_image' && fieldsData.base_style.value === 'vector_illustration'",
+                    multiple=True,
                 ),
                 "size": InputPort(
                     name="size",
@@ -647,11 +665,13 @@ class StableDiffusion(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "negative_prompt": InputPort(
                     name="negative_prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "model": InputPort(
                     name="model",

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/llms.py

@@ -16,6 +16,7 @@ class AliyunQwen(Node):
                     name="prompt",
                     port_type=PortType.TEXT,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -77,6 +78,7 @@ class Baichuan(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -165,6 +167,7 @@ class BaiduWenxin(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -204,6 +207,7 @@ class ChatGLM(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -286,6 +290,7 @@ class Claude(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -333,6 +338,7 @@ class Deepseek(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -423,6 +429,7 @@ class Gemini(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -519,6 +526,7 @@ class LingYiWanWu(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -563,6 +571,7 @@ class MiniMax(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -648,6 +657,7 @@ class Moonshot(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -734,6 +744,7 @@ class OpenAI(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -824,6 +835,7 @@ class XAi(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -908,6 +920,7 @@ class CustomModel(Node):
                     name="prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "model_family": InputPort(
                     name="model_family",

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/media_editing.py

@@ -17,6 +17,7 @@ class AudioEditing(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".mp3", ".wav", ".ogg", ".m4a"],
+                    multiple=True,
                 ),
                 "audio_processing_logic": InputPort(
                     name="audio_processing_logic",
@@ -221,6 +222,7 @@ class ImageEditing(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
+                    multiple=True,
                 ),
                 "crop": InputPort(
                     name="crop",
@@ -362,6 +364,7 @@ class ImageSegmentation(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
+                    multiple=True,
                 ),
                 "selection_method": InputPort(
                     name="selection_method",
@@ -422,6 +425,7 @@ class ImageWatermark(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
+                    multiple=True,
                 ),
                 "image_or_text": InputPort(
                     name="image_or_text",
@@ -554,6 +558,7 @@ class VideoEditing(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=["video/*"],
+                    multiple=True,
                 ),
                 "video_processing_logic": InputPort(
                     name="video_processing_logic",
@@ -631,6 +636,7 @@ class VideoScreenshot(Node):
                     port_type=PortType.FILE,
                     value=list(),
                     support_file_types=["video/*"],
+                    multiple=True,
                 ),
                 "screenshot_method": InputPort(
                     name="screenshot_method",

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/media_processing.py

@@ -16,6 +16,7 @@ class ClaudeVision(Node):
                     name="text_prompt",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "llm_model": InputPort(
                     name="llm_model",
@@ -28,6 +29,12 @@ class ClaudeVision(Node):
                         {"value": "claude-3-haiku", "label": "claude-3-haiku"},
                     ],
                 ),
+                "multiple_input": InputPort(
+                    name="multiple_input",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                    has_tooltip=True,
+                ),
                 "images_or_urls": InputPort(
                     name="images_or_urls",
                     port_type=PortType.RADIO,
@@ -130,6 +137,12 @@ class GeminiVision(Node):
                         {"value": "gemini-exp-1206", "label": "gemini-exp-1206"},
                     ],
                 ),
+                "multiple_input": InputPort(
+                    name="multiple_input",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                    has_tooltip=True,
+                ),
                 "images_or_urls": InputPort(
                     name="images_or_urls",
                     port_type=PortType.RADIO,
@@ -404,6 +417,12 @@ class QwenVision(Node):
                         {"value": "qwen-vl-plus", "label": "qwen-vl-plus"},
                     ],
                 ),
+                "multiple_input": InputPort(
+                    name="multiple_input",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                    has_tooltip=True,
+                ),
                 "images_or_urls": InputPort(
                     name="images_or_urls",
                     port_type=PortType.RADIO,

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/text_processing.py

@@ -16,6 +16,7 @@ class TextInOut(Node):
                     name="text",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "input_type": InputPort(
                     name="input_type",
@@ -26,10 +27,7 @@ class TextInOut(Node):
                         {"value": "number", "label": "number"},
                     ],
                 ),
-                "output": OutputPort(
-                    name="output",
-                    port_type=PortType.TEXT,
-                ),
+                "output": OutputPort(),
             },
         )
 
@@ -46,16 +44,14 @@ class TextReplace(Node):
                     name="text",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "replace_items": InputPort(
                     name="replace_items",
                     port_type=PortType.INPUT,
                     value=[],
                 ),
-                "output": OutputPort(
-                    name="output",
-                    port_type=PortType.TEXT,
-                ),
+                "output": OutputPort(),
             },
         )
 
@@ -101,10 +97,7 @@ class TextSplitters(Node):
                     value="\\n",
                     condition="return fieldsData.split_method.value == 'delimiter'",
                 ),
-                "output": OutputPort(
-                    name="output",
-                    port_type=PortType.TEXT,
-                ),
+                "output": OutputPort(),
             },
         )
 
@@ -121,6 +114,7 @@ class TextTruncation(Node):
                     name="text",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "truncate_method": InputPort(
                     name="truncate_method",
@@ -212,9 +206,11 @@ class TemplateCompose(Node):
                     name="template",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "output": OutputPort(),
             },
+            can_add_input_ports=True,
         )
 
 
@@ -230,6 +226,7 @@ class RegexExtract(Node):
                     name="text",
                     port_type=PortType.TEXTAREA,
                     value="",
+                    multiple=True,
                 ),
                 "pattern": InputPort(
                     name="pattern",

{vectorvein-0.2.31 → vectorvein-0.2.33}/src/vectorvein/workflow/nodes/web_crawlers.py

@@ -16,6 +16,7 @@ class TextCrawler(Node):
                     name="url",
                     port_type=PortType.INPUT,
                     value="",
+                    multiple=True,
                 ),
                 "output_type": InputPort(
                     name="output_type",
@@ -53,6 +54,7 @@ class BilibiliCrawler(Node):
                     name="url_or_bvid",
                     port_type=PortType.INPUT,
                     value="",
+                    multiple=True,
                 ),
                 "download_video": InputPort(
                     name="download_video",
@@ -93,6 +95,7 @@ class DouyinCrawler(Node):
                     name="url",
                     port_type=PortType.INPUT,
                     value="",
+                    multiple=True,
                 ),
                 "output_title": OutputPort(
                     name="output_title",
@@ -119,6 +122,7 @@ class YoutubeCrawler(Node):
                     name="url_or_video_id",
                     port_type=PortType.INPUT,
                     value="",
+                    multiple=True,
                 ),
                 "get_comments": InputPort(
                     name="get_comments",