vectorvein 0.2.25__tar.gz → 0.2.27__tar.gz

{vectorvein-0.2.25 → vectorvein-0.2.27}/PKG-INFO

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

{vectorvein-0.2.25 → vectorvein-0.2.27}/pyproject.toml

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

{vectorvein-0.2.25 → vectorvein-0.2.27}/src/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-0.2.25 → vectorvein-0.2.27}/src/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-0.2.25 → vectorvein-0.2.27}/src/vectorvein/workflow/graph/workflow.py

@@ -65,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)
@@ -112,7 +119,7 @@ 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)

{vectorvein-0.2.25 → vectorvein-0.2.27}/src/vectorvein/workflow/nodes/__init__.py

@@ -14,6 +14,7 @@ from .llms import (
     Moonshot,
     OpenAI,
     XAi,
+    CustomModel,
 )
 from .media_editing import (
     AudioEditing,
@@ -45,6 +46,7 @@ from .text_processing import (
     MarkdownToHtml,
     ListRender,
     TemplateCompose,
+    RegexExtract,
 )
 from .tools import CodebaseAnalysis, TextTranslation, TextSearch, ProgrammingFunction, ImageSearch, WorkflowInvoke
 from .vector_db import AddData, DeleteData, Search
@@ -65,6 +67,7 @@ __all__ = [
     "CodebaseAnalysis",
     "CogVideoX",
     "Conditional",
+    "CustomModel",
     "DallE",
     "Deepseek",
     "DeepseekVl",
@@ -109,6 +112,7 @@ __all__ = [
     "QwenVision",
     "RandomChoice",
     "Recraft",
+    "RegexExtract",
     "RunSql",
     "SmartQuery",
     "SoundEffects",