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

vectorvein/chat_clients/anthropic_client.py

@@ -35,6 +35,7 @@ from anthropic import (
     AsyncAnthropicBedrock,
 )
 from anthropic._types import NOT_GIVEN
+from anthropic._exceptions import APIStatusError as AnthropicAPIStatusError
 from anthropic.types import (
     TextBlock,
     ThinkingBlock,
@@ -52,6 +53,7 @@ from ..types import defaults as defs
 from .utils import cutoff_messages, get_message_token_counts
 from .base_client import BaseChatClient, BaseAsyncChatClient
 from .openai_compatible_client import OpenAICompatibleChatClient, AsyncOpenAICompatibleChatClient
+from ..types.exception import APIStatusError
 from ..types.enums import ContextLengthControlType, BackendType
 from ..types.llm_parameters import (
     Usage,
@@ -601,18 +603,21 @@ class AnthropicChatClient(BaseChatClient):
         self._acquire_rate_limit(self.endpoint, self.model, messages)
 
         if self.stream:
-            stream_response = raw_client.messages.create(
-                model=self.model_id,
-                messages=messages,
-                system=system_prompt,
-                stream=True,
-                temperature=self.temperature,
-                max_tokens=max_tokens,
-                tools=tools_params,
-                tool_choice=tool_choice_param,
-                top_p=top_p,
-                thinking=thinking,
-            )
+            try:
+                stream_response = raw_client.messages.create(
+                    model=self.model_id,
+                    messages=messages,
+                    system=system_prompt,
+                    stream=True,
+                    temperature=self.temperature,
+                    max_tokens=max_tokens,
+                    tools=tools_params,
+                    tool_choice=tool_choice_param,
+                    top_p=top_p,
+                    thinking=thinking,
+                )
+            except AnthropicAPIStatusError as e:
+                raise APIStatusError(message=e.message, response=e.response, body=e.body)
 
             def generator():
                 result = {"content": "", "reasoning_content": "", "usage": {}, "tool_calls": []}
@@ -675,18 +680,21 @@ class AnthropicChatClient(BaseChatClient):
 
             return generator()
         else:
-            response = raw_client.messages.create(
-                model=self.model_id,
-                messages=messages,
-                system=system_prompt,
-                stream=False,
-                temperature=self.temperature,
-                max_tokens=max_tokens,
-                tools=tools_params,
-                tool_choice=tool_choice_param,
-                top_p=top_p,
-                thinking=thinking,
-            )
+            try:
+                response = raw_client.messages.create(
+                    model=self.model_id,
+                    messages=messages,
+                    system=system_prompt,
+                    stream=False,
+                    temperature=self.temperature,
+                    max_tokens=max_tokens,
+                    tools=tools_params,
+                    tool_choice=tool_choice_param,
+                    top_p=top_p,
+                    thinking=thinking,
+                )
+            except AnthropicAPIStatusError as e:
+                raise APIStatusError(message=e.message, response=e.response, body=e.body)
 
             result = {
                 "content": "",
@@ -1140,18 +1148,21 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
         await self._acquire_rate_limit(self.endpoint, self.model, messages)
 
         if self.stream:
-            stream_response = await raw_client.messages.create(
-                model=self.model_id,
-                messages=messages,
-                system=system_prompt,
-                stream=True,
-                temperature=self.temperature,
-                max_tokens=max_tokens,
-                tools=tools_params,
-                tool_choice=tool_choice_param,
-                top_p=top_p,
-                thinking=thinking,
-            )
+            try:
+                stream_response = await raw_client.messages.create(
+                    model=self.model_id,
+                    messages=messages,
+                    system=system_prompt,
+                    stream=True,
+                    temperature=self.temperature,
+                    max_tokens=max_tokens,
+                    tools=tools_params,
+                    tool_choice=tool_choice_param,
+                    top_p=top_p,
+                    thinking=thinking,
+                )
+            except AnthropicAPIStatusError as e:
+                raise APIStatusError(message=e.message, response=e.response, body=e.body)
 
             async def generator():
                 result = {"content": "", "reasoning_content": "", "usage": {}, "tool_calls": []}
@@ -1214,18 +1225,21 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
 
             return generator()
         else:
-            response = await raw_client.messages.create(
-                model=self.model_id,
-                messages=messages,
-                system=system_prompt,
-                stream=False,
-                temperature=self.temperature,
-                max_tokens=max_tokens,
-                tools=tools_params,
-                tool_choice=tool_choice_param,
-                top_p=top_p,
-                thinking=thinking,
-            )
+            try:
+                response = await raw_client.messages.create(
+                    model=self.model_id,
+                    messages=messages,
+                    system=system_prompt,
+                    stream=False,
+                    temperature=self.temperature,
+                    max_tokens=max_tokens,
+                    tools=tools_params,
+                    tool_choice=tool_choice_param,
+                    top_p=top_p,
+                    thinking=thinking,
+                )
+            except AnthropicAPIStatusError as e:
+                raise APIStatusError(message=e.message, response=e.response, body=e.body)
 
             result = {
                 "content": "",

vectorvein/workflow/graph/edge.py

@@ -6,17 +6,17 @@ class Edge:
         self,
         id: str,
         source: str,
-        sourceHandle: str,
+        source_handle: str,
         target: str,
-        targetHandle: str,
+        target_handle: str,
         animated: bool = True,
         type: str = "default",
     ) -> None:
         self.id: str = id
         self.source: str = source
-        self.sourceHandle: str = sourceHandle
+        self.source_handle: str = source_handle
         self.target: str = target
-        self.targetHandle: str = targetHandle
+        self.target_handle: str = target_handle
         self.animated: bool = animated
         self.type: str = type
         self.style: Dict[str, Union[str, int]] = {"stroke": "#28c5e5", "strokeWidth": 3}
@@ -25,9 +25,9 @@ class Edge:
         return {
             "id": self.id,
             "source": self.source,
-            "sourceHandle": self.sourceHandle,
+            "sourceHandle": self.source_handle,
             "target": self.target,
-            "targetHandle": self.targetHandle,
+            "targetHandle": self.target_handle,
             "animated": self.animated,
             "type": self.type,
             "style": self.style,

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] = []
@@ -58,6 +65,13 @@ class Workflow:
         if not target_node_obj.has_input_port(target_port):
             raise ValueError(f"目标节点 {target_node_id} 不存在输入端口: {target_port}")
 
+        # 检查目标端口是否已有被连接的线
+        for edge in self.edges:
+            if edge.target == target_node_id and edge.target_handle == target_port:
+                raise ValueError(
+                    f"目标节点 {target_node_id} 的输入端口 {target_port} 已经被连接: {edge.source}({edge.source_handle}) → {edge.target}({edge.target_handle})"
+                )
+
         # 创建并添加边
         edge_id = f"vueflow__edge-{source_node_id}{source_port}-{target_node_id}{target_port}"
         edge = Edge(edge_id, source_node_id, source_port, target_node_id, target_port)
@@ -105,7 +119,106 @@ class Workflow:
         for edge in self.edges:
             source_label = node_id_to_label[edge.source]
             target_label = node_id_to_label[edge.target]
-            label = f"{edge.sourceHandle} → {edge.targetHandle}"
+            label = f"{edge.source_handle} → {edge.target_handle}"
             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.26.dist-info}/METADATA

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

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

@@ -1,13 +1,13 @@
-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.26.dist-info/METADATA,sha256=FuIheW9ox95gDN74fCkhXrE9d0VUYdp5VuoJ4xvPUgY,4570
+vectorvein-0.2.26.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.26.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
 vectorvein/api/exceptions.py,sha256=uS_PAdx0ksC0r3dgfSGWdbLMZm4qdLeWSSqCv1g3_Gc,772
 vectorvein/api/models.py,sha256=xtPWMsB0yIJI7i-gY4B6MtvXv0ZIXnoeKspmeInH6fU,1449
 vectorvein/chat_clients/__init__.py,sha256=UIytpIgwo8qkZpIyrHVxLYTyliUOTp4J7C4iHRjbtWE,23850
-vectorvein/chat_clients/anthropic_client.py,sha256=URufNrSLKk24kaNPwvxjDfRsYcSSbAmPjyZJdSt2U7Q,58978
+vectorvein/chat_clients/anthropic_client.py,sha256=PVUP9UddiyFG7MXFXyon_PkQZxOgUc4GrsAmNkSu2co,59915
 vectorvein/chat_clients/baichuan_client.py,sha256=CVMvpgjdrZGv0BWnTOBD-f2ufZ3wq3496wqukumsAr4,526
 vectorvein/chat_clients/base_client.py,sha256=p7s-G4Wh9MSpDKEfG8wuFAeWy5DGvj5Go31hqrpQPhM,38817
 vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
@@ -41,16 +41,16 @@ vectorvein/types/settings.py,sha256=82RsherFSCc8s9-v0E7F_FK6avVh_XvC1b7EkNlFHbc,
 vectorvein/utilities/media_processing.py,sha256=7KtbLFzOYIn1e9QTN9G6C76NH8CBlV9kfAgiRKEIeXY,6263
 vectorvein/utilities/rate_limiter.py,sha256=dwolIUVw2wP83Odqpx0AAaE77de1GzxkYDGH4tM_u_4,10300
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
-vectorvein/workflow/graph/edge.py,sha256=xLZEJmBjAfVB53cd7CuRcKhgE6QqXv9nz32wJn8cmyk,1064
+vectorvein/workflow/graph/edge.py,sha256=1ckyyjCue_PLm7P1ItUfKOy6AKkemOpZ9m1WJ8UXIHQ,1072
 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=l0rSDZeSd6OtHMwxzhIENfOKS0fJKKB4JUow-dV-LUI,8610
 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.26.dist-info/RECORD,,