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

vectorvein/workflow/nodes/media_processing.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class ClaudeVision(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ClaudeVision",
             category="media_processing",
@@ -53,7 +51,7 @@ class ClaudeVision(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -73,7 +71,7 @@ class ClaudeVision(Node):
 
 
 class DeepseekVl(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="DeepseekVl",
             category="media_processing",
@@ -105,7 +103,7 @@ class DeepseekVl(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -125,7 +123,7 @@ class DeepseekVl(Node):
 
 
 class GeminiVision(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="GeminiVision",
             category="media_processing",
@@ -167,7 +165,7 @@ class GeminiVision(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -187,7 +185,7 @@ class GeminiVision(Node):
 
 
 class GlmVision(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="GlmVision",
             category="media_processing",
@@ -221,7 +219,7 @@ class GlmVision(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.images_or_urls.value == 'images'",
@@ -241,7 +239,7 @@ class GlmVision(Node):
 
 
 class GptVision(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="GptVision",
             category="media_processing",
@@ -277,7 +275,7 @@ class GptVision(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -307,7 +305,7 @@ class GptVision(Node):
 
 
 class InternVision(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="InternVision",
             category="media_processing",
@@ -340,7 +338,7 @@ class InternVision(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -360,7 +358,7 @@ class InternVision(Node):
 
 
 class Ocr(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Ocr",
             category="media_processing",
@@ -389,7 +387,7 @@ class Ocr(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -416,7 +414,7 @@ class Ocr(Node):
                 ),
                 "output_words_info": OutputPort(
                     name="output_words_info",
-                    value=list(),
+                    value=[],
                     condition="fields_data.get('ocr_type') in ['general', 'business_license']",
                     condition_python=lambda ports: ports["ocr_type"].value in ["general", "business_license"],
                     has_tooltip=True,
@@ -426,7 +424,7 @@ class Ocr(Node):
 
 
 class QwenVision(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="QwenVision",
             category="media_processing",
@@ -468,7 +466,7 @@ class QwenVision(Node):
                 "images": InputPort(
                     name="images",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".jpg", ".jpeg", ".png", ".webp"],
                     condition="fields_data.get('images_or_urls') == 'images'",
@@ -488,7 +486,7 @@ class QwenVision(Node):
 
 
 class SpeechRecognition(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="SpeechRecognition",
             category="media_processing",
@@ -507,7 +505,7 @@ class SpeechRecognition(Node):
                 "files": InputPort(
                     name="files",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     multiple=True,
                     support_file_types=[".wav", ".mp3", ".mp4", ".m4a", ".wma", ".aac", ".ogg", ".amr", ".flac"],
                     condition="fields_data.get('files_or_urls') == 'files'",

vectorvein/workflow/nodes/output.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class Audio(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Audio",
             category="outputs",
@@ -63,7 +61,7 @@ class Audio(Node):
 
 
 class Text(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Text",
             category="outputs",
@@ -93,7 +91,7 @@ class Text(Node):
 
 
 class Table(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Table",
             category="outputs",
@@ -131,7 +129,7 @@ class Table(Node):
 
 
 class Document(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Document",
             category="outputs",
@@ -186,7 +184,7 @@ class Document(Node):
 
 
 class Echarts(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Echarts",
             category="outputs",
@@ -208,7 +206,7 @@ class Echarts(Node):
 
 
 class Email(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Email",
             category="outputs",
@@ -240,7 +238,7 @@ class Email(Node):
 
 
 class Html(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Html",
             category="outputs",
@@ -258,7 +256,7 @@ class Html(Node):
 
 
 class Mermaid(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Mermaid",
             category="outputs",
@@ -280,7 +278,7 @@ class Mermaid(Node):
 
 
 class Mindmap(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Mindmap",
             category="outputs",
@@ -302,7 +300,7 @@ class Mindmap(Node):
 
 
 class PictureRender(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="PictureRender",
             category="outputs",

vectorvein/workflow/nodes/relational_db.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class GetTableInfo(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="GetTableInfo",
             category="relational_db",
@@ -39,7 +37,7 @@ class GetTableInfo(Node):
 
 
 class RunSql(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="RunSql",
             category="relational_db",
@@ -90,7 +88,7 @@ class RunSql(Node):
 
 
 class SmartQuery(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="SmartQuery",
             category="relational_db",

vectorvein/workflow/nodes/text_processing.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class TextInOut(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextInOut",
             category="text_processing",
@@ -33,7 +31,7 @@ class TextInOut(Node):
 
 
 class TextReplace(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextReplace",
             category="text_processing",
@@ -57,7 +55,7 @@ class TextReplace(Node):
 
 
 class TextSplitters(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextSplitters",
             category="text_processing",
@@ -106,7 +104,7 @@ class TextSplitters(Node):
 
 
 class TextTruncation(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextTruncation",
             category="text_processing",
@@ -144,7 +142,7 @@ class TextTruncation(Node):
 
 
 class MarkdownToHtml(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="MarkdownToHtml",
             category="text_processing",
@@ -165,7 +163,7 @@ class MarkdownToHtml(Node):
 
 
 class ListRender(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ListRender",
             category="text_processing",
@@ -199,7 +197,7 @@ class ListRender(Node):
 
 
 class TemplateCompose(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TemplateCompose",
             category="text_processing",
@@ -219,7 +217,7 @@ class TemplateCompose(Node):
 
 
 class RegexExtract(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="RegexExtract",
             category="text_processing",

vectorvein/workflow/nodes/tools.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class CodebaseAnalysis(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="CodebaseAnalysis",
             category="tools",
@@ -24,7 +22,7 @@ class CodebaseAnalysis(Node):
                 "codebase_file": InputPort(
                     name="codebase_file",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".zip"],
                     multiple=False,
                     condition="return fieldsData.input_type.value === 'file'",
@@ -65,7 +63,7 @@ class CodebaseAnalysis(Node):
                 "ignore_patterns": InputPort(
                     name="ignore_patterns",
                     port_type=PortType.INPUT,
-                    value=list(),
+                    value=[],
                     multiple=True,
                 ),
                 "output": OutputPort(),
@@ -74,7 +72,7 @@ class CodebaseAnalysis(Node):
 
 
 class TextTranslation(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextTranslation",
             category="tools",
@@ -142,7 +140,7 @@ class TextTranslation(Node):
 
 
 class TextSearch(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextSearch",
             category="tools",
@@ -264,7 +262,7 @@ class TextSearch(Node):
 
 
 class ProgrammingFunction(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ProgrammingFunction",
             category="tools",
@@ -330,7 +328,7 @@ class ProgrammingFunction(Node):
 
 
 class ImageSearch(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ImageSearch",
             category="tools",
@@ -375,7 +373,7 @@ class ImageSearch(Node):
 
 
 class WorkflowInvoke(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="WorkflowInvoke",
             category="tools",

vectorvein/workflow/nodes/triggers.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort
 
 
 class ButtonTrigger(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="ButtonTrigger",
             category="triggers",

vectorvein/workflow/nodes/vector_db.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class AddData(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="AddData",
             category="vector_db",
@@ -91,7 +89,7 @@ class AddData(Node):
 
 
 class DeleteData(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="DeleteData",
             category="vector_db",
@@ -115,7 +113,7 @@ class DeleteData(Node):
 
 
 class Search(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="Search",
             category="vector_db",

vectorvein/workflow/nodes/video_generation.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class KlingVideo(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="KlingVideo",
             category="video_generation",
@@ -20,7 +18,7 @@ class KlingVideo(Node):
                 "image": InputPort(
                     name="image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png"],
                     multiple=True,
                 ),
@@ -67,7 +65,7 @@ class KlingVideo(Node):
 
 
 class CogVideoX(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="CogVideoX",
             category="video_generation",
@@ -82,7 +80,7 @@ class CogVideoX(Node):
                 "image": InputPort(
                     name="image",
                     port_type=PortType.FILE,
-                    value=list(),
+                    value=[],
                     support_file_types=[".jpg", ".jpeg", ".png"],
                     multiple=True,
                 ),

vectorvein/workflow/nodes/web_crawlers.py

@@ -1,11 +1,9 @@
-from typing import Optional
-
 from ..graph.node import Node
 from ..graph.port import PortType, InputPort, OutputPort
 
 
 class TextCrawler(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="TextCrawler",
             category="web_crawlers",
@@ -43,7 +41,7 @@ class TextCrawler(Node):
 
 
 class BilibiliCrawler(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="BilibiliCrawler",
             category="web_crawlers",
@@ -84,7 +82,7 @@ class BilibiliCrawler(Node):
 
 
 class DouyinCrawler(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="DouyinCrawler",
             category="web_crawlers",
@@ -111,7 +109,7 @@ class DouyinCrawler(Node):
 
 
 class YoutubeCrawler(Node):
-    def __init__(self, id: Optional[str] = None):
+    def __init__(self, id: str | None = None):
         super().__init__(
             node_type="YoutubeCrawler",
             category="web_crawlers",

vectorvein/workflow/utils/analyse.py

@@ -34,9 +34,7 @@ class AnalyseResult(TypedDict):
     nodes: list[NodeRecord]
 
 
-def analyse_workflow_record(
-    json_str: str, connected_only: bool = False, reserver_programming_function_ports: bool = False
-) -> AnalyseResult:
+def analyse_workflow_record(json_str: str, connected_only: bool = False, reserver_programming_function_ports: bool = False) -> AnalyseResult:
     """
     分析工作流JSON字符串，提取节点和端口信息
 
@@ -274,9 +272,7 @@ def format_analysis_result(analysis_result: AnalyseResult, max_value_length: int
 
                     port_summary.append(port_info)
 
-                truncated_node["ports_summary"] = (
-                    f"{len(ports)}个端口，其中{sum(1 for p in ports if p.get('connected'))}个已连接"
-                )
+                truncated_node["ports_summary"] = f"{len(ports)}个端口，其中{sum(1 for p in ports if p.get('connected'))}个已连接"
 
             node_str = f"节点{idx + 1}: {json.dumps(truncated_node, ensure_ascii=False, indent=2)}"
             formatted_parts.append(node_str)
@@ -284,7 +280,7 @@ def format_analysis_result(analysis_result: AnalyseResult, max_value_length: int
     # 添加其他可能的顶级信息
     for key, value in analysis_result.items():
         if key != "nodes":
-            if isinstance(value, (dict, list)):
+            if isinstance(value, dict | list):
                 summary = f"{key}: 包含{len(value)}个项目"
             else:
                 summary = f"{key}: {prettify_value(value, max_value_length)}"
@@ -350,16 +346,12 @@ def format_workflow_analysis_for_llm(analysis_result: AnalyseResult, max_value_l
                             # 列表值，使用Python风格显示
                             truncated_list = []
                             item_max_length = max(max_value_length // len(value) if value else max_value_length, 10)
-                            node_info.append(
-                                f"      - value(list): {json.dumps(prettify_value(value, item_max_length), ensure_ascii=False)}"
-                            )
+                            node_info.append(f"      - value(list): {json.dumps(prettify_value(value, item_max_length), ensure_ascii=False)}")
                         elif isinstance(value, dict):
                             # 字典值，使用Python风格显示
                             dict_items = []
                             key_max_length = max(max_value_length // len(value) if value else max_value_length, 10)
-                            node_info.append(
-                                f"      - value(dict): {json.dumps(prettify_value(value, key_max_length), ensure_ascii=False)}"
-                            )
+                            node_info.append(f"      - value(dict): {json.dumps(prettify_value(value, key_max_length), ensure_ascii=False)}")
                         elif isinstance(value, str):
                             # 字符串值，带引号
                             truncated = prettify_value(value, max_value_length)

vectorvein/workflow/utils/check.py

@@ -39,17 +39,11 @@ def check_dag(workflow: "Workflow"):
     result = {"no_cycle": True, "no_isolated_nodes": True}
 
     # 过滤掉触发器节点和辅助节点
-    trigger_nodes = [
-        node.id
-        for node in workflow.nodes
-        if hasattr(node, "category") and (node.category == "triggers" or node.category == "assistedNodes")
-    ]
+    trigger_nodes = [node.id for node in workflow.nodes if hasattr(node, "category") and (node.category == "triggers" or node.category == "assistedNodes")]
 
     # 获取需要检查的节点和边
     regular_nodes = [node.id for node in workflow.nodes if node.id not in trigger_nodes]
-    regular_edges = [
-        edge for edge in workflow.edges if edge.source not in trigger_nodes and edge.target not in trigger_nodes
-    ]
+    regular_edges = [edge for edge in workflow.edges if edge.source not in trigger_nodes and edge.target not in trigger_nodes]
 
     # ---------- 检查有向图是否有环 ----------
     # 构建邻接表
@@ -59,7 +53,7 @@ def check_dag(workflow: "Workflow"):
             adjacency[edge.source].append(edge.target)
 
     # 三种状态: 0 = 未访问, 1 = 正在访问, 2 = 已访问完成
-    visited = {node_id: 0 for node_id in regular_nodes}
+    visited = dict.fromkeys(regular_nodes, 0)
 
     def dfs_cycle_detection(node_id):
         # 如果节点正在被访问，说明找到了环
@@ -157,9 +151,7 @@ def check_ui(workflow: "Workflow") -> UIWarning:
 
                 # 检查显示的端口是否也被连接
                 if (node.id, port_name) in connected_ports:
-                    warnings["input_ports_shown_but_connected"].append(
-                        {"node_id": node.id, "node_type": node.type, "port_name": port_name}
-                    )
+                    warnings["input_ports_shown_but_connected"].append({"node_id": node.id, "node_type": node.type, "port_name": port_name})
 
     # 如果没有任何显示的输入端口
     warnings["has_shown_input_ports"] = has_shown_input_ports
@@ -174,7 +166,7 @@ def check_useless_nodes(workflow: "Workflow") -> list["Node"]:
     1. 节点非 output 类节点，并且节点的输出端口没有任何连线，说明该节点数据不会传给下一个节点或者显示出来。
     """
     useless_nodes = []
-    source_nodes = set(edge.source for edge in workflow.edges)
+    source_nodes = {edge.source for edge in workflow.edges}
 
     for node in workflow.nodes:
         if hasattr(node, "category") and node.category == "outputs":
@@ -206,9 +198,7 @@ def check_required_ports(workflow: "Workflow") -> "list[tuple[Node, Port]]":
                         # 如果是函数，传入ports作为参数
                         condition_result = port.condition_python(ports)
                     else:
-                        raise ValueError(
-                            f"condition_python 必须是可调用函数，当前类型为 {type(port.condition_python)}"
-                        )
+                        raise ValueError(f"condition_python 必须是可调用函数，当前类型为 {type(port.condition_python)}")
 
                     if not condition_result:
                         condition_applies = False