vectorvein 0.2.24__py3-none-any.whl → 0.2.25__py3-none-any.whl

vectorvein/workflow/graph/workflow.py

@@ -1,10 +1,17 @@
 import json
-from typing import List, Union
+from typing import List, Union, TypedDict
 
 from .node import Node
 from .edge import Edge
 
 
+class WorkflowCheckResult(TypedDict):
+    """工作流检查结果类型。"""
+
+    no_cycle: bool  # 工作流是否不包含环
+    no_isolated_nodes: bool  # 工作流是否不包含孤立节点
+
+
 class Workflow:
     def __init__(self) -> None:
         self.nodes: List[Node] = []
@@ -109,3 +116,102 @@ class Workflow:
             lines.append(f"    {source_label} -->|{label}| {target_label}")
 
         return "\n".join(lines)
+
+    def _check_dag(self) -> WorkflowCheckResult:
+        """检查流程图是否为有向无环图，并检测是否存在孤立节点。
+
+        Returns:
+            WorkflowCheckResult: 包含检查结果的字典
+                - no_cycle (bool): 如果流程图是有向无环图返回 True，否则返回 False
+                - no_isolated_nodes (bool): 如果不存在孤立节点返回 True，否则返回 False
+        """
+        result: WorkflowCheckResult = {"no_cycle": True, "no_isolated_nodes": True}
+
+        # 过滤掉触发器节点和辅助节点
+        trigger_nodes = [
+            node.id
+            for node in self.nodes
+            if hasattr(node, "category") and (node.category == "triggers" or node.category == "assistedNodes")
+        ]
+
+        # 获取需要检查的节点和边
+        regular_nodes = [node.id for node in self.nodes if node.id not in trigger_nodes]
+        regular_edges = [
+            edge for edge in self.edges if edge.source not in trigger_nodes and edge.target not in trigger_nodes
+        ]
+
+        # ---------- 检查有向图是否有环 ----------
+        # 构建邻接表
+        adjacency = {node_id: [] for node_id in regular_nodes}
+        for edge in regular_edges:
+            if edge.source in adjacency:  # 确保节点在字典中
+                adjacency[edge.source].append(edge.target)
+
+        # 三种状态: 0 = 未访问, 1 = 正在访问, 2 = 已访问完成
+        visited = {node_id: 0 for node_id in regular_nodes}
+
+        def dfs_cycle_detection(node_id):
+            # 如果节点正在被访问，说明找到了环
+            if visited[node_id] == 1:
+                return False
+
+            # 如果节点已经访问完成，无需再次访问
+            if visited[node_id] == 2:
+                return True
+
+            # 标记为正在访问
+            visited[node_id] = 1
+
+            # 访问所有邻居
+            for neighbor in adjacency[node_id]:
+                if neighbor in visited and not dfs_cycle_detection(neighbor):
+                    return False
+
+            # 标记为已访问完成
+            visited[node_id] = 2
+            return True
+
+        # 对每个未访问的节点进行 DFS 检测环
+        for node_id in regular_nodes:
+            if visited[node_id] == 0:
+                if not dfs_cycle_detection(node_id):
+                    result["no_cycle"] = False
+                    break
+
+        # ---------- 检查是否存在孤立节点 ----------
+        # 构建无向图邻接表
+        undirected_adjacency = {node_id: [] for node_id in regular_nodes}
+        for edge in regular_edges:
+            if edge.source in undirected_adjacency and edge.target in undirected_adjacency:
+                undirected_adjacency[edge.source].append(edge.target)
+                undirected_adjacency[edge.target].append(edge.source)
+
+        # 深度优先搜索来检测连通分量
+        undirected_visited = set()
+
+        def dfs_connected_components(node_id):
+            undirected_visited.add(node_id)
+            for neighbor in undirected_adjacency[node_id]:
+                if neighbor not in undirected_visited:
+                    dfs_connected_components(neighbor)
+
+        # 计算连通分量数量
+        connected_components_count = 0
+        for node_id in regular_nodes:
+            if node_id not in undirected_visited:
+                connected_components_count += 1
+                dfs_connected_components(node_id)
+
+        # 如果连通分量数量大于1，说明存在孤立节点
+        if connected_components_count > 1 and len(regular_nodes) > 0:
+            result["no_isolated_nodes"] = False
+
+        return result
+
+    def check(self) -> WorkflowCheckResult:
+        """检查流程图的有效性。
+
+        Returns:
+            WorkflowCheckResult: 包含各种检查结果的字典
+        """
+        return self._check_dag()

vectorvein/workflow/nodes/llms.py

@@ -292,6 +292,8 @@ class Claude(Node):
                     port_type=PortType.SELECT,
                     value="claude-3-5-haiku",
                     options=[
+                        {"value": "claude-3-7-sonnet-thinking", "label": "claude-3-7-sonnet-thinking"},
+                        {"value": "claude-3-7-sonnet", "label": "claude-3-7-sonnet"},
                         {"value": "claude-3-5-sonnet", "label": "claude-3-5-sonnet"},
                         {"value": "claude-3-5-haiku", "label": "claude-3-5-haiku"},
                         {"value": "claude-3-opus", "label": "claude-3-opus"},
@@ -338,8 +340,8 @@ class Deepseek(Node):
                     value="deepseek-chat",
                     options=[
                         {"value": "deepseek-chat", "label": "deepseek-chat"},
-                        {"value": "deepseek-reasoner", "label": "deepseek-reasoner"},
-                        {"value": "deepseek-32k", "label": "deepseek-32k"},
+                        {"value": "deepseek-reasoner", "label": "deepseek-r1"},
+                        {"value": "deepseek-r1-distill-qwen-32b", "label": "deepseek-r1-distill-qwen-32b"},
                     ],
                 ),
                 "temperature": InputPort(
@@ -523,12 +525,10 @@ class LingYiWanWu(Node):
                     port_type=PortType.SELECT,
                     value="yi-lightning",
                     options=[
-                        {"value": "yi-lightning", "label": "yi-lightning"},
-                        {"value": "yi-large", "label": "yi-large"},
-                        {"value": "yi-large-turbo", "label": "yi-large-turbo"},
-                        {"value": "yi-medium", "label": "yi-medium"},
-                        {"value": "yi-medium-200k", "label": "yi-medium-200k"},
-                        {"value": "yi-spark", "label": "yi-spark"},
+                        {
+                            "value": "yi-lightning",
+                            "label": "yi-lightning",
+                        },
                     ],
                 ),
                 "temperature": InputPort(
@@ -746,6 +746,7 @@ class OpenAI(Node):
                         {"value": "gpt-4o-mini", "label": "gpt-4o-mini"},
                         {"value": "o1-mini", "label": "o1-mini"},
                         {"value": "o1-preview", "label": "o1-preview"},
+                        {"value": "o3-mini", "label": "o3-mini"},
                     ],
                 ),
                 "temperature": InputPort(
@@ -893,3 +894,91 @@ class XAi(Node):
                 ),
             },
         )
+
+
+class CustomModel(Node):
+    def __init__(self, id: Optional[str] = None):
+        super().__init__(
+            node_type="CustomModel",
+            category="llms",
+            task_name="llms.custom_model",
+            node_id=id,
+            ports={
+                "prompt": InputPort(
+                    name="prompt",
+                    port_type=PortType.TEXTAREA,
+                    value="",
+                ),
+                "model_family": InputPort(
+                    name="model_family",
+                    port_type=PortType.SELECT,
+                    value="",
+                    options=[],
+                ),
+                "llm_model": InputPort(
+                    name="llm_model",
+                    port_type=PortType.SELECT,
+                    value="",
+                    options=[],
+                ),
+                "temperature": InputPort(
+                    name="temperature",
+                    port_type=PortType.TEMPERATURE,
+                    value=0.7,
+                ),
+                "top_p": InputPort(
+                    name="top_p",
+                    port_type=PortType.NUMBER,
+                    value=0.95,
+                ),
+                "stream": InputPort(
+                    name="stream",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                ),
+                "system_prompt": InputPort(
+                    name="system_prompt",
+                    port_type=PortType.TEXTAREA,
+                    value="",
+                ),
+                "response_format": InputPort(
+                    name="response_format",
+                    port_type=PortType.SELECT,
+                    value="text",
+                    options=[
+                        {"value": "text", "label": "Text"},
+                        {"value": "json_object", "label": "JSON"},
+                    ],
+                ),
+                "use_function_call": InputPort(
+                    name="use_function_call",
+                    port_type=PortType.CHECKBOX,
+                    value=False,
+                ),
+                "functions": InputPort(
+                    name="functions",
+                    port_type=PortType.SELECT,
+                    value=[],
+                ),
+                "function_call_mode": InputPort(
+                    name="function_call_mode",
+                    port_type=PortType.SELECT,
+                    value="auto",
+                    options=[
+                        {"value": "auto", "label": "auto"},
+                        {"value": "none", "label": "none"},
+                    ],
+                ),
+                "output": OutputPort(
+                    name="output",
+                ),
+                "function_call_output": OutputPort(
+                    name="function_call_output",
+                    condition="return fieldsData.use_function_call.value",
+                ),
+                "function_call_arguments": OutputPort(
+                    name="function_call_arguments",
+                    condition="return fieldsData.use_function_call.value",
+                ),
+            },
+        )

{vectorvein-0.2.24.dist-info → vectorvein-0.2.25.dist-info}/METADATA

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

{vectorvein-0.2.24.dist-info → vectorvein-0.2.25.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.24.dist-info/METADATA,sha256=Vq9TrFThHAi0HNPZxp4-JuEjGB7vgMSbf2hOZNJXRXQ,4570
-vectorvein-0.2.24.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.24.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.25.dist-info/METADATA,sha256=Q5GwQcPe2_hMU26jkaFftOxDlEkbYEPN-vw6KvImgRQ,4570
+vectorvein-0.2.25.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.25.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
@@ -44,13 +44,13 @@ vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM
 vectorvein/workflow/graph/edge.py,sha256=xLZEJmBjAfVB53cd7CuRcKhgE6QqXv9nz32wJn8cmyk,1064
 vectorvein/workflow/graph/node.py,sha256=A3M_GghrSju1D3xc_HtPdGyr-7XSkplGPKJveOUiIF4,3256
 vectorvein/workflow/graph/port.py,sha256=Q6HmI2cUi6viJ98ec6-MmMPMRtKS1-OgaudP3LMwVLA,6054
-vectorvein/workflow/graph/workflow.py,sha256=uh8JzTbvaucKryvFUj3nAswtdOtaOw2Z4YfD9Q6r77s,3973
+vectorvein/workflow/graph/workflow.py,sha256=XIoCHfJBNLEvdNf1xDur10o4cjAvhuqy2SjRLvRip1M,8200
 vectorvein/workflow/nodes/__init__.py,sha256=jd4O27kIJdOtkij1FYZ6aJnJy2OQa7xtL1r-Yv8ylO0,3103
 vectorvein/workflow/nodes/audio_generation.py,sha256=ht2S0vnd0mIAt6FBaSWlADGbb7f_1DAySYrgYnvZT1Q,5726
 vectorvein/workflow/nodes/control_flows.py,sha256=Zc_uWuroYznLrU-BZCncyzvejC-zFl6EuN_VP8oq5mY,6573
 vectorvein/workflow/nodes/file_processing.py,sha256=Rsjc8al0z-2KuweO0nIybWvceqxbqOPQyTs0-pgy5m4,3980
 vectorvein/workflow/nodes/image_generation.py,sha256=fXOhLGodJ3OdKBPXO5a3rq4wN2GMJ0jwqKO_gJFdocU,32852
-vectorvein/workflow/nodes/llms.py,sha256=RD3z5hsVJ722zCgBaM7PncgqvBcFA-f_XFSaOvhgMhc,34859
+vectorvein/workflow/nodes/llms.py,sha256=rXN5Vgn6EvoglNb_BEzVUIrc4dCDxlinEYqznSE-Bek,38121
 vectorvein/workflow/nodes/media_editing.py,sha256=Od0X0SdcyRhcJckWpDM4WvgWEKxaIsgMXpMifN8Sc5M,29405
 vectorvein/workflow/nodes/media_processing.py,sha256=t-azYDphXmLRdOyHDfXFTS1tsEOyKqKskDyD0y232j8,19043
 vectorvein/workflow/nodes/output.py,sha256=_UQxiddHtGv2rkjhUFE-KDgrjnh0AGJQJyq9-4Aji5A,12567
@@ -62,4 +62,4 @@ 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=LsqomfXfqrXfHJDO1cl0Ox48f4St7X_SL12DSbAMSOw,5415
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.2.24.dist-info/RECORD,,
+vectorvein-0.2.25.dist-info/RECORD,,