vectorvein 0.2.6__py3-none-any.whl → 0.2.7__py3-none-any.whl

vectorvein/chat_clients/anthropic_client.py

@@ -17,6 +17,8 @@ from anthropic import (
 from anthropic._types import NOT_GIVEN
 from anthropic.types import (
     TextBlock,
+    ThinkingBlock,
+    RedactedThinkingBlock,
     MessageParam,
     ToolUseBlock,
     RawMessageDeltaEvent,
@@ -128,6 +130,8 @@ def refactor_into_openai_messages(messages: Iterable[MessageParam]):
             for item in content:
                 if isinstance(item, (TextBlock, ToolUseBlock)):
                     _content.append(item.model_dump())
+                elif isinstance(item, (ThinkingBlock, RedactedThinkingBlock)):
+                    continue
                 elif item.get("type") == "image":
                     image_data = item.get("source", {}).get("data", "")
                     media_type = item.get("source", {}).get("media_type", "")
@@ -338,6 +342,9 @@ class AnthropicChatClient(BaseChatClient):
 
             formatted_messages = refactor_into_openai_messages(messages)
 
+            if "thinking" in kwargs:
+                kwargs.pop("thinking")  # TODO: 暂时没看到如何处理 openai 接口的 thinking 参数，如 openrouter 中使用
+
             if self.stream:
 
                 def _generator():
@@ -451,7 +458,7 @@ class AnthropicChatClient(BaseChatClient):
             )
 
             def generator():
-                result = {"content": "", "usage": {}, "tool_calls": []}
+                result = {"content": "", "reasoning_content": "", "usage": {}, "tool_calls": []}
                 for chunk in stream_response:
                     message = {"content": "", "tool_calls": []}
                     if isinstance(chunk, RawMessageStartEvent):
@@ -472,11 +479,16 @@ class AnthropicChatClient(BaseChatClient):
                             ]
                         elif chunk.content_block.type == "text":
                             message["content"] = chunk.content_block.text
+                        elif chunk.content_block.type == "thinking":
+                            message["reasoning_content"] = chunk.content_block.thinking
                         yield ChatCompletionDeltaMessage(**message)
                     elif isinstance(chunk, RawContentBlockDeltaEvent):
                         if chunk.delta.type == "text_delta":
                             message["content"] = chunk.delta.text
                             result["content"] += chunk.delta.text
+                        elif chunk.delta.type == "thinking_delta":
+                            message["reasoning_content"] = chunk.delta.thinking
+                            result["reasoning_content"] += chunk.delta.thinking
                         elif chunk.delta.type == "input_json_delta":
                             result["tool_calls"][0]["function"]["arguments"] += chunk.delta.partial_json
                             message["tool_calls"] = [
@@ -521,6 +533,7 @@ class AnthropicChatClient(BaseChatClient):
 
             result = {
                 "content": "",
+                "reasoning_content": "",
                 "usage": {
                     "prompt_tokens": response.usage.input_tokens,
                     "completion_tokens": response.usage.output_tokens,
@@ -531,6 +544,8 @@ class AnthropicChatClient(BaseChatClient):
             for content_block in response.content:
                 if isinstance(content_block, TextBlock):
                     result["content"] += content_block.text
+                elif isinstance(content_block, ThinkingBlock):
+                    result["reasoning_content"] = content_block.thinking
                 elif isinstance(content_block, ToolUseBlock):
                     tool_calls.append(content_block.model_dump())
 
@@ -728,6 +743,9 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
 
             formatted_messages = refactor_into_openai_messages(messages)
 
+            if "thinking" in kwargs:
+                kwargs.pop("thinking")  # TODO: 暂时没看到如何处理 openai 接口的 thinking 参数，如 openrouter 中使用
+
             if self.stream:
 
                 async def _generator():
@@ -843,7 +861,7 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
             )
 
             async def generator():
-                result = {"content": "", "usage": {}, "tool_calls": []}
+                result = {"content": "", "reasoning_content": "", "usage": {}, "tool_calls": []}
                 async for chunk in stream_response:
                     message = {"content": "", "tool_calls": []}
                     if isinstance(chunk, RawMessageStartEvent):
@@ -864,11 +882,16 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                             ]
                         elif chunk.content_block.type == "text":
                             message["content"] = chunk.content_block.text
+                        elif chunk.content_block.type == "thinking":
+                            message["reasoning_content"] = chunk.content_block.thinking
                         yield ChatCompletionDeltaMessage(**message)
                     elif isinstance(chunk, RawContentBlockDeltaEvent):
                         if chunk.delta.type == "text_delta":
                             message["content"] = chunk.delta.text
                             result["content"] += chunk.delta.text
+                        elif chunk.delta.type == "thinking_delta":
+                            message["reasoning_content"] = chunk.delta.thinking
+                            result["reasoning_content"] += chunk.delta.thinking
                         elif chunk.delta.type == "input_json_delta":
                             result["tool_calls"][0]["function"]["arguments"] += chunk.delta.partial_json
                             message["tool_calls"] = [
@@ -913,6 +936,7 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
 
             result = {
                 "content": "",
+                "reasoning_content": "",
                 "usage": {
                     "prompt_tokens": response.usage.input_tokens,
                     "completion_tokens": response.usage.output_tokens,
@@ -923,6 +947,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
             for content_block in response.content:
                 if isinstance(content_block, TextBlock):
                     result["content"] += content_block.text
+                elif isinstance(content_block, ThinkingBlock):
+                    result["reasoning_content"] = content_block.thinking
                 elif isinstance(content_block, ToolUseBlock):
                     tool_calls.append(content_block.model_dump())
 

vectorvein/types/defaults.py

@@ -541,6 +541,14 @@ OPENAI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "response_format_available": False,
         "native_multimodal": False,
     },
+    "o3-mini": {
+        "id": "o3-mini",
+        "context_length": 200000,
+        "max_output_tokens": 100000,
+        "function_call_available": True,
+        "response_format_available": True,
+        "native_multimodal": False,
+    },
 }
 
 # Anthropic models

{vectorvein-0.2.6.dist-info → vectorvein-0.2.7.dist-info}/METADATA

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

{vectorvein-0.2.6.dist-info → vectorvein-0.2.7.dist-info}/RECORD

@@ -1,13 +1,13 @@
-vectorvein-0.2.6.dist-info/METADATA,sha256=DtsS36_CyacffEZFVJHYIXzCMMLYmith6vrRnlXkBSg,4413
-vectorvein-0.2.6.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.6.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.7.dist-info/METADATA,sha256=J67JgDpuGe0YZmLAxOKdkMYKaXD_6LjhSjAzmkzAfSM,4413
+vectorvein-0.2.7.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.7.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=omQuG4PRRPNflSAgtdU--rwsWG6vMpwMEyIGZyFVHVQ,18596
-vectorvein/chat_clients/anthropic_client.py,sha256=Zk6X1feIvv7Az5dgyipJXbm9TkgWgpFghSTxLiXKKA8,38405
+vectorvein/chat_clients/anthropic_client.py,sha256=8ns9oVFbGoRMWnizmskuDDW2C-xrbVVV_66s3uuhpNM,40079
 vectorvein/chat_clients/baichuan_client.py,sha256=CVMvpgjdrZGv0BWnTOBD-f2ufZ3wq3496wqukumsAr4,526
 vectorvein/chat_clients/base_client.py,sha256=QLvcGhjravPbvha6-spU-w6ugHU1LrsbdFUcs6NwMgE,18842
 vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
@@ -30,7 +30,7 @@ vectorvein/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/server/token_server.py,sha256=36F9PKSNOX8ZtYBXY_l-76GQTpUSmQ2Y8EMy1H7wtdQ,1353
 vectorvein/settings/__init__.py,sha256=ecGyrE_6YfX9z6Igb1rDCu1Q-qMTcVozWF3WEl_hiKA,4871
 vectorvein/settings/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/types/defaults.py,sha256=LhjCjeoR8AZxNmyEg_5LEETzfRQet-UrNQ8q08WxmZ8,27169
+vectorvein/types/defaults.py,sha256=VrkQoyHqC_eK3g1b6egpPYLLo0ltwMHqxDscCX4y-N0,27417
 vectorvein/types/enums.py,sha256=7KTJSVtQueImmbr1fSwv3rQVtc0RyMWXJmoE2tDOaso,1667
 vectorvein/types/exception.py,sha256=gnW4GnJ76jND6UGnodk9xmqkcbeS7Cz2rvncA2HpD5E,69
 vectorvein/types/llm_parameters.py,sha256=9KJ8z_7xVa0JsV0YPxRKz5n8Eb8PfvB9Y89e6ahjmjw,5989
@@ -59,4 +59,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.6.dist-info/RECORD,,
+vectorvein-0.2.7.dist-info/RECORD,,