vectorvein 0.2.5__tar.gz → 0.2.7__tar.gz

{vectorvein-0.2.5 → vectorvein-0.2.7}/PKG-INFO

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

{vectorvein-0.2.5 → vectorvein-0.2.7}/pyproject.toml

@@ -18,7 +18,7 @@ description = "VectorVein python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.2.5"
+version = "0.2.7"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.5 → vectorvein-0.2.7}/src/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-0.2.5 → vectorvein-0.2.7}/src/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
@@ -594,6 +602,14 @@ ANTHROPIC_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "response_format_available": False,
         "native_multimodal": True,
     },
+    "claude-3-7-sonnet-20250219": {
+        "id": "claude-3-7-sonnet-20250219",
+        "context_length": 128000,
+        "max_output_tokens": 8192,
+        "function_call_available": True,
+        "response_format_available": False,
+        "native_multimodal": True,
+    },
 }
 
 # Minimax models