vectorvein 0.1.60__tar.gz → 0.1.62__tar.gz

{vectorvein-0.1.60 → vectorvein-0.1.62}/PKG-INFO

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

{vectorvein-0.1.60 → vectorvein-0.1.62}/pyproject.toml

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

{vectorvein-0.1.60 → vectorvein-0.1.62}/src/vectorvein/chat_clients/anthropic_client.py

@@ -179,8 +179,8 @@ class AnthropicChatClient(BaseChatClient):
         )
         self.model_id = None
 
-    @cached_property
-    def raw_client(self):
+    @property
+    def raw_client(self):  # type: ignore
         if self.random_endpoint:
             self.random_endpoint = True
             endpoint = random.choice(self.backend_settings.models[self.model].endpoints)
@@ -233,6 +233,7 @@ class AnthropicChatClient(BaseChatClient):
                 aws_access_key=self.endpoint.credentials.get("access_key"),
                 aws_secret_key=self.endpoint.credentials.get("secret_key"),
                 aws_region=self.endpoint.region,
+                base_url=self.endpoint.api_base,
                 http_client=self.http_client,
             )
         elif self.endpoint.api_schema_type == "default":
@@ -400,7 +401,8 @@ class AnthropicChatClient(BaseChatClient):
                     **kwargs,
                 )
 
-        assert isinstance(self.raw_client, Anthropic | AnthropicVertex)
+        raw_client = self.raw_client  # 调用完 self.raw_client 后，self.model_id 会被赋值
+        assert isinstance(raw_client, Anthropic | AnthropicVertex | AnthropicBedrock)
 
         if isinstance(tools, OpenAINotGiven):
             tools = NOT_GIVEN
@@ -409,7 +411,6 @@ class AnthropicChatClient(BaseChatClient):
         if isinstance(top_p, OpenAINotGiven) or top_p is None:
             top_p = NOT_GIVEN
 
-        raw_client = self.raw_client  # 调用完 self.raw_client 后，self.model_id 会被赋值
         self.model_setting = self.backend_settings.models[self.model]
         if self.model_id is None:
             self.model_id = self.model_setting.id
@@ -569,8 +570,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
         )
         self.model_id = None
 
-    @cached_property
-    def raw_client(self):
+    @property
+    def raw_client(self):  # type: ignore
         if self.random_endpoint:
             self.random_endpoint = True
             endpoint = random.choice(self.backend_settings.models[self.model].endpoints)
@@ -792,7 +793,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                     **kwargs,
                 )
 
-        assert isinstance(self.raw_client, AsyncAnthropic | AsyncAnthropicVertex)
+        raw_client = self.raw_client  # 调用完 self.raw_client 后，self.model_id 会被赋值
+        assert isinstance(raw_client, AsyncAnthropic | AsyncAnthropicVertex | AsyncAnthropicBedrock)
 
         if isinstance(tools, OpenAINotGiven):
             tools = NOT_GIVEN
@@ -801,7 +803,6 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
         if isinstance(top_p, OpenAINotGiven) or top_p is None:
             top_p = NOT_GIVEN
 
-        raw_client = self.raw_client  # 调用完 self.raw_client 后，self.model_id 会被赋值
         self.model_setting = self.backend_settings.models[self.model]
         if self.model_id is None:
             self.model_id = self.model_setting.id