vectorvein 0.2.48__py3-none-any.whl → 0.2.50__py3-none-any.whl

vectorvein/types/defaults.py

@@ -698,7 +698,7 @@ GEMINI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "id": "gemini-2.0-flash-thinking-exp-01-21",
         "context_length": 1048576,
         "max_output_tokens": 8192,
-        "function_call_available": True,
+        "function_call_available": False,
         "response_format_available": True,
         "native_multimodal": False,
     },

vectorvein/workflow/graph/workflow.py

@@ -1,9 +1,10 @@
 import json
-from typing import List, Union, TypedDict
+from typing import List, Union, TypedDict, Dict, Any, Optional
 
 from .node import Node
 from .edge import Edge
 from .port import InputPort
+from ..utils.layout import layout
 
 
 class UIWarning(TypedDict, total=False):
@@ -283,3 +284,22 @@ class Workflow:
         result["ui_warnings"] = ui_check
 
         return result
+
+    def layout(self, options: Optional[Dict[str, Any]] = None) -> "Workflow":
+        """对工作流中的节点进行自动布局，计算并更新每个节点的位置。
+
+        此方法实现了一个简单的分层布局算法，将节点按照有向图的拓扑结构进行排列。
+
+        Args:
+            options: 布局选项，包括:
+                - direction: 布局方向 ('TB', 'BT', 'LR', 'RL')，默认 'LR'
+                - node_spacing: 同一层级节点间的间距，默认 500
+                - layer_spacing: 不同层级间的间距，默认 400
+                - margin_x: 图形左右边距，默认 20
+                - margin_y: 图形上下边距，默认 20
+
+        Returns:
+            布局后的工作流对象
+        """
+        layout(self.nodes, self.edges, options)
+        return self

vectorvein/workflow/utils/layout.py

@@ -0,0 +1,114 @@
+from typing import Optional, Dict, Any, List, TYPE_CHECKING
+
+
+if TYPE_CHECKING:
+    from vectorvein.workflow.graph.node import Node
+    from vectorvein.workflow.graph.edge import Edge
+
+
+def layout(nodes: List["Node"], edges: List["Edge"], options: Optional[Dict[str, Any]] = None):
+    """对工作流中的节点进行自动布局，计算并更新每个节点的位置。
+
+    此方法实现了一个简单的分层布局算法，将节点按照有向图的拓扑结构进行排列。
+
+    Args:
+        options: 布局选项，包括:
+            - direction: 布局方向 ('TB', 'BT', 'LR', 'RL')，默认 'TB'
+            - node_spacing: 同一层级节点间的间距，默认 150
+            - layer_spacing: 不同层级间的间距，默认 100
+            - margin_x: 图形左右边距，默认 20
+            - margin_y: 图形上下边距，默认 20
+
+    Returns:
+        布局后的工作流对象
+    """
+    # 设置默认选项
+    default_options = {
+        "direction": "LR",  # 从上到下的布局
+        "node_spacing": 400,  # 同一层级节点间的间距
+        "layer_spacing": 500,  # 不同层级间的间距
+        "margin_x": 20,  # 图形左右边距
+        "margin_y": 20,  # 图形上下边距
+    }
+
+    # 合并用户提供的选项
+    if options:
+        default_options.update(options)
+
+    # 构建邻接表
+    adjacency = {node.id: [] for node in nodes}
+    in_degree = {node.id: 0 for node in nodes}
+
+    for edge in edges:
+        if edge.source in adjacency:
+            adjacency[edge.source].append(edge.target)
+            in_degree[edge.target] = in_degree.get(edge.target, 0) + 1
+
+    # 找出所有入度为0的节点（根节点）
+    roots = [node_id for node_id, degree in in_degree.items() if degree == 0]
+
+    # 如果没有根节点，选择第一个节点作为起点
+    if not roots and nodes:
+        roots = [nodes[0].id]
+
+    # 按层级排列节点
+    layers = []
+    visited = set()
+
+    current_layer = roots
+    while current_layer:
+        layers.append(current_layer)
+        next_layer = []
+        for node_id in current_layer:
+            visited.add(node_id)
+            for neighbor in adjacency.get(node_id, []):
+                if neighbor not in visited and all(
+                    parent in visited for parent in [e.source for e in edges if e.target == neighbor]
+                ):
+                    next_layer.append(neighbor)
+        current_layer = next_layer
+
+    # 还有未访问的节点（可能是孤立节点或环的一部分）
+    remaining = [node.id for node in nodes if node.id not in visited]
+    if remaining:
+        layers.append(remaining)
+
+    # 根据层级信息设置节点位置
+    layer_spacing = default_options["layer_spacing"]
+    node_spacing = default_options["node_spacing"]
+    margin_x = default_options["margin_x"]
+    margin_y = default_options["margin_y"]
+
+    # 布局方向
+    is_vertical = default_options["direction"] in ["TB", "BT"]
+    is_reversed = default_options["direction"] in ["BT", "RL"]
+
+    for layer_idx, layer in enumerate(layers):
+        for node_idx, node_id in enumerate(layer):
+            # 根据布局方向计算位置
+            if is_vertical:
+                # 垂直布局 (TB 或 BT)
+                x = node_idx * node_spacing + margin_x
+                y = layer_idx * layer_spacing + margin_y
+                if is_reversed:  # BT 布局需要反转 y 坐标
+                    y = (len(layers) - 1 - layer_idx) * layer_spacing + margin_y
+            else:
+                # 水平布局 (LR 或 RL)
+                x = layer_idx * layer_spacing + margin_x
+                y = node_idx * node_spacing + margin_y
+                if is_reversed:  # RL 布局需要反转 x 坐标
+                    x = (len(layers) - 1 - layer_idx) * layer_spacing + margin_x
+
+            # 找到节点对象并设置位置
+            for node in nodes:
+                if node.id == node_id:
+                    # 确保节点有 position 属性
+                    if not hasattr(node, "position"):
+                        node.position = {"x": x, "y": y}
+                    else:
+                        # 如果已经有 position 属性，更新它
+                        if isinstance(node.position, dict):
+                            node.position.update({"x": x, "y": y})
+                        else:
+                            node.position = {"x": x, "y": y}
+                    break

{vectorvein-0.2.48.dist-info → vectorvein-0.2.50.dist-info}/METADATA

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

{vectorvein-0.2.48.dist-info → vectorvein-0.2.50.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.48.dist-info/METADATA,sha256=5zz-Cv6NkNsXQ8TqbojdH8VB4-HgbAgGtZX3IXJX3CE,4570
-vectorvein-0.2.48.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.48.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.50.dist-info/METADATA,sha256=ltuWwgnho44gDn6eGM8ED7_dtnefHd1KZ89yMM4bqrY,4570
+vectorvein-0.2.50.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.50.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
@@ -32,7 +32,7 @@ vectorvein/server/token_server.py,sha256=36F9PKSNOX8ZtYBXY_l-76GQTpUSmQ2Y8EMy1H7
 vectorvein/settings/__init__.py,sha256=3Kw3hbvqcIQepAR6Q2m2UXbBnwyJTUm8yAz-aHmbUTg,11163
 vectorvein/settings/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/types/__init__.py,sha256=ypg8c8AwF49FrFBMqmgH_eIBH4LFf0KN4kjqQa7zrvM,3376
-vectorvein/types/defaults.py,sha256=RA4bERvJQ9hy3UFmrF2K3qA7Pp1CKkgudH3PsX14lGs,27562
+vectorvein/types/defaults.py,sha256=MpDvT9vNwIqwHFKXTlBq3FY02PLr4BncCdvNhfbxM60,27563
 vectorvein/types/enums.py,sha256=LplSVkXLBK-t8TWtJKj_f7ktWTd6CSHWRLb67XKMm54,1716
 vectorvein/types/exception.py,sha256=KtnqZ-1DstHm95SZAyZdHhkGq1bJ4A9Aw3Zfdu-VIFo,130
 vectorvein/types/llm_parameters.py,sha256=2rF-CQsWcHqTzI2r5x55gSsFm7LW_iCxQxZTEnk0yF8,7843
@@ -44,7 +44,7 @@ vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM
 vectorvein/workflow/graph/edge.py,sha256=1ckyyjCue_PLm7P1ItUfKOy6AKkemOpZ9m1WJ8UXIHQ,1072
 vectorvein/workflow/graph/node.py,sha256=U3LAq01wOaDANVUuYWc2BhIq-Pl2-33FirJZkkNL_wA,5329
 vectorvein/workflow/graph/port.py,sha256=_QpHCBGAu657VhYAh0Wzjri3ZZ8-WYJp99J465mqmwo,6492
-vectorvein/workflow/graph/workflow.py,sha256=jDa8dedgtW39Gx0PrqgYrs7XQ_NfQbhgGge88xElksI,11159
+vectorvein/workflow/graph/workflow.py,sha256=rXLIKQiOFjyP3YwYf0deHUcRNBCHJcTGj2omM4a9SpE,12070
 vectorvein/workflow/nodes/__init__.py,sha256=dWrWtL3q0Vsn-MLgJ7gNgLCrwZ5BrqjrN2QFPNeBMuc,3240
 vectorvein/workflow/nodes/audio_generation.py,sha256=ZRFZ_ycMTSJ2LKmekctagQdJYTl-3q4TNOIKETpS9AM,5870
 vectorvein/workflow/nodes/control_flows.py,sha256=l8CjFQlsGV3fNGM6SVzS1Kz361K1xDv1fGT7acuDXuU,6613
@@ -62,4 +62,5 @@ vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs0
 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.48.dist-info/RECORD,,
+vectorvein/workflow/utils/layout.py,sha256=j0bRD3uaXu40xCS6U6BGahBI8FrHa5MiF55GbTrZ1LM,4565
+vectorvein-0.2.50.dist-info/RECORD,,