vectorvein 0.1.44__tar.gz → 0.1.45__tar.gz

{vectorvein-0.1.44 → vectorvein-0.1.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.44
+Version: 0.1.45
 Summary: Default template for PDM package
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.44 → vectorvein-0.1.45}/pyproject.toml

@@ -17,7 +17,7 @@ description = "Default template for PDM package"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.44"
+version = "0.1.45"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.44 → vectorvein-0.1.45}/src/vectorvein/chat_clients/anthropic_client.py

@@ -7,6 +7,7 @@ from typing import overload, Generator, AsyncGenerator, Any, Literal, Iterable
 
 import httpx
 from openai._types import NotGiven as OpenAINotGiven
+from openai._types import NOT_GIVEN as OPENAI_NOT_GIVEN
 from anthropic import Anthropic, AnthropicVertex, AsyncAnthropic, AsyncAnthropicVertex
 from anthropic._types import NOT_GIVEN
 from anthropic.types import (
@@ -282,6 +283,8 @@ class AnthropicChatClient(BaseChatClient):
             self.endpoint = settings.get_endpoint(self.endpoint_id)
 
         if self.endpoint.api_schema_type == "openai":
+            _tools = OPENAI_NOT_GIVEN if tools is NOT_GIVEN else tools
+            _tool_choice = OPENAI_NOT_GIVEN if tool_choice is NOT_GIVEN else tool_choice
             if self.stream:
 
                 def _generator():
@@ -295,7 +298,7 @@ class AnthropicChatClient(BaseChatClient):
                         http_client=self.http_client,
                         backend_name=self.BACKEND_NAME,
                     ).create_completion(
-                        messages, model, True, temperature, max_tokens, tools, tool_choice, response_format, **kwargs
+                        messages, model, True, temperature, max_tokens, _tools, _tool_choice, response_format, **kwargs
                     )
                     for chunk in response:
                         yield chunk
@@ -312,7 +315,7 @@ class AnthropicChatClient(BaseChatClient):
                     http_client=self.http_client,
                     backend_name=self.BACKEND_NAME,
                 ).create_completion(
-                    messages, model, False, temperature, max_tokens, tools, tool_choice, response_format, **kwargs
+                    messages, model, False, temperature, max_tokens, _tools, _tool_choice, response_format, **kwargs
                 )
 
         assert isinstance(self.raw_client, Anthropic | AnthropicVertex)
@@ -620,6 +623,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
             self.endpoint = settings.get_endpoint(self.endpoint_id)
 
         if self.endpoint.api_schema_type == "openai":
+            _tools = OPENAI_NOT_GIVEN if tools is NOT_GIVEN else tools
+            _tool_choice = OPENAI_NOT_GIVEN if tool_choice is NOT_GIVEN else tool_choice
             if self.stream:
 
                 async def _generator():
@@ -634,7 +639,7 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                         backend_name=self.BACKEND_NAME,
                     )
                     response = await client.create_completion(
-                        messages, model, True, temperature, max_tokens, tools, tool_choice, response_format, **kwargs
+                        messages, model, True, temperature, max_tokens, _tools, _tool_choice, response_format, **kwargs
                     )
                     async for chunk in response:
                         yield chunk
@@ -652,7 +657,7 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                     backend_name=self.BACKEND_NAME,
                 )
                 return await client.create_completion(
-                    messages, model, False, temperature, max_tokens, tools, tool_choice, response_format, **kwargs
+                    messages, model, False, temperature, max_tokens, _tools, _tool_choice, response_format, **kwargs
                 )
 
         assert isinstance(self.raw_client, AsyncAnthropic | AsyncAnthropicVertex)