vectorvein 0.2.29__py3-none-any.whl → 0.2.31__py3-none-any.whl

vectorvein/chat_clients/anthropic_client.py

@@ -36,6 +36,7 @@ from anthropic import (
 )
 from anthropic._types import NOT_GIVEN
 from anthropic._exceptions import APIStatusError as AnthropicAPIStatusError
+from anthropic._exceptions import APIConnectionError as AnthropicAPIConnectionError
 from anthropic.types import (
     TextBlock,
     ThinkingBlock,
@@ -53,7 +54,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.exception import APIStatusError, APIConnectionError
 from ..types.enums import ContextLengthControlType, BackendType
 from ..types.llm_parameters import (
     Usage,
@@ -618,6 +619,8 @@ class AnthropicChatClient(BaseChatClient):
                 )
             except AnthropicAPIStatusError as e:
                 raise APIStatusError(message=e.message, response=e.response, body=e.body)
+            except AnthropicAPIConnectionError as e:
+                raise APIConnectionError(message=e.message, request=e.request)
 
             def generator():
                 result = {"content": "", "reasoning_content": "", "usage": {}, "tool_calls": []}
@@ -695,6 +698,8 @@ class AnthropicChatClient(BaseChatClient):
                 )
             except AnthropicAPIStatusError as e:
                 raise APIStatusError(message=e.message, response=e.response, body=e.body)
+            except AnthropicAPIConnectionError as e:
+                raise APIConnectionError(message=e.message, request=e.request)
 
             result = {
                 "content": "",
@@ -1163,6 +1168,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                 )
             except AnthropicAPIStatusError as e:
                 raise APIStatusError(message=e.message, response=e.response, body=e.body)
+            except AnthropicAPIConnectionError as e:
+                raise APIConnectionError(message=e.message, request=e.request)
 
             async def generator():
                 result = {"content": "", "reasoning_content": "", "usage": {}, "tool_calls": []}
@@ -1240,6 +1247,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                 )
             except AnthropicAPIStatusError as e:
                 raise APIStatusError(message=e.message, response=e.response, body=e.body)
+            except AnthropicAPIConnectionError as e:
+                raise APIConnectionError(message=e.message, request=e.request)
 
             result = {
                 "content": "",

vectorvein/chat_clients/utils.py

@@ -559,6 +559,7 @@ def format_workflow_messages(message: dict, content: str, backend: BackendType):
 def transform_from_openai_message(message: ChatCompletionMessageParam, backend: BackendType):
     role = message.get("role", "user")
     content = message.get("content", "")
+    tool_calls = message.get("tool_calls", [])
 
     if backend == BackendType.Anthropic:
         if role == "tool":
@@ -582,6 +583,16 @@ def transform_from_openai_message(message: ChatCompletionMessageParam, backend:
                         formatted_content.append(format_image_message(item["image_url"]["url"], backend))
                     else:
                         formatted_content.append(item)
+            if tool_calls:
+                for tool_call in tool_calls:
+                    formatted_content.append(
+                        {
+                            "type": "tool_use",
+                            "id": tool_call["id"],
+                            "name": tool_call["function"]["name"],
+                            "input": json.loads(tool_call["function"]["arguments"]),
+                        }
+                    )
             return {"role": role, "content": formatted_content}
         else:
             return {"role": role, "content": content}

vectorvein/types/__init__.py

@@ -36,7 +36,7 @@ from .defaults import (
     XAI_MODELS,
 )
 from .enums import BackendType, LLMType, ContextLengthControlType
-from .exception import APIStatusError
+from .exception import APIStatusError, APIConnectionError
 from .llm_parameters import (
     EndpointSetting,
     ModelSetting,
@@ -109,6 +109,7 @@ __all__ = [
     "LLMType",
     "ContextLengthControlType",
     "APIStatusError",
+    "APIConnectionError",
     "EndpointOptionDict",
     "EndpointSetting",
     "ModelSetting",

vectorvein/types/exception.py

@@ -1,4 +1,5 @@
 from openai import APIStatusError
+from openai import APIConnectionError
 
 
-__all__ = ["APIStatusError"]
+__all__ = ["APIStatusError", "APIConnectionError"]

{vectorvein-0.2.29.dist-info → vectorvein-0.2.31.dist-info}/METADATA

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

{vectorvein-0.2.29.dist-info → vectorvein-0.2.31.dist-info}/RECORD

@@ -1,13 +1,13 @@
-vectorvein-0.2.29.dist-info/METADATA,sha256=x_u1wwPKVV4ZhMXjLN0VmM3uG72meapuWpH1jCbBtsE,4570
-vectorvein-0.2.29.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.29.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.31.dist-info/METADATA,sha256=6bAhtke289AGWRwg2q8KMxxxgs78Mt8o2nfpO7444qk,4570
+vectorvein-0.2.31.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.31.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=PVUP9UddiyFG7MXFXyon_PkQZxOgUc4GrsAmNkSu2co,59915
+vectorvein/chat_clients/anthropic_client.py,sha256=MfROOOCPiuaW0N8DRTbkjHSL9Stv4I0xIMvILF-RARM,60547
 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
@@ -23,7 +23,7 @@ vectorvein/chat_clients/openai_compatible_client.py,sha256=d6V6T_FvPxEuOhHoQC-NH
 vectorvein/chat_clients/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15uPoJQOHA0,513
 vectorvein/chat_clients/stepfun_client.py,sha256=zsD2W5ahmR4DD9cqQTXmJr3txrGuvxbRWhFlRdwNijI,519
-vectorvein/chat_clients/utils.py,sha256=hb4fNgd1F3AI46qBqAT4IEQhLw6qkkdPjf9PDIEb9A8,25777
+vectorvein/chat_clients/utils.py,sha256=g7lIzuTrZgm3Tq6skTRu8_xUpml8EfRFFMuFEKQm7Vc,26267
 vectorvein/chat_clients/xai_client.py,sha256=eLFJJrNRJ-ni3DpshODcr3S1EJQLbhVwxyO1E54LaqM,491
 vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
 vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
@@ -31,10 +31,10 @@ vectorvein/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/server/token_server.py,sha256=36F9PKSNOX8ZtYBXY_l-76GQTpUSmQ2Y8EMy1H7wtdQ,1353
 vectorvein/settings/__init__.py,sha256=3Kw3hbvqcIQepAR6Q2m2UXbBnwyJTUm8yAz-aHmbUTg,11163
 vectorvein/settings/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/types/__init__.py,sha256=DJYGhlshgUQgzXPfMfKW5sTpBClZUCLhqmCqF44lVuU,3329
+vectorvein/types/__init__.py,sha256=ypg8c8AwF49FrFBMqmgH_eIBH4LFf0KN4kjqQa7zrvM,3376
 vectorvein/types/defaults.py,sha256=U8acSg7yBWz5rOuwqRY45XX6lqPkM7Y7RbuurX9cpxg,27335
 vectorvein/types/enums.py,sha256=LplSVkXLBK-t8TWtJKj_f7ktWTd6CSHWRLb67XKMm54,1716
-vectorvein/types/exception.py,sha256=gnW4GnJ76jND6UGnodk9xmqkcbeS7Cz2rvncA2HpD5E,69
+vectorvein/types/exception.py,sha256=KtnqZ-1DstHm95SZAyZdHhkGq1bJ4A9Aw3Zfdu-VIFo,130
 vectorvein/types/llm_parameters.py,sha256=1-8G6ieUj_Hb_hnJNTj7kt_7R6LkPmA6fcbNuIIiDDs,8070
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/types/settings.py,sha256=82RsherFSCc8s9-v0E7F_FK6avVh_XvC1b7EkNlFHbc,4918
@@ -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.29.dist-info/RECORD,,
+vectorvein-0.2.31.dist-info/RECORD,,