vectorvein 0.2.25__py3-none-any.whl → 0.2.27__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

@@ -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/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",

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

@@ -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.dist-info → vectorvein-0.2.27.dist-info}/RECORD

@@ -1,13 +1,13 @@
-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-0.2.27.dist-info/METADATA,sha256=eRQRtg4_0C11ovd8KFKMW2Z-5eCIllKhPMWyFlx9PE4,4570
+vectorvein-0.2.27.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.27.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,11 +41,11 @@ 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=XIoCHfJBNLEvdNf1xDur10o4cjAvhuqy2SjRLvRip1M,8200
-vectorvein/workflow/nodes/__init__.py,sha256=jd4O27kIJdOtkij1FYZ6aJnJy2OQa7xtL1r-Yv8ylO0,3103
+vectorvein/workflow/graph/workflow.py,sha256=l0rSDZeSd6OtHMwxzhIENfOKS0fJKKB4JUow-dV-LUI,8610
+vectorvein/workflow/nodes/__init__.py,sha256=a1fsitA0mf1UlDOSEz6QjSGjiFsvsQcuDE-p2q9SlhM,3181
 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
@@ -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.25.dist-info/RECORD,,
+vectorvein-0.2.27.dist-info/RECORD,,