vectorvein 0.2.41__tar.gz → 0.2.43__tar.gz

{vectorvein-0.2.41 → vectorvein-0.2.43}/PKG-INFO

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

{vectorvein-0.2.41 → vectorvein-0.2.43}/pyproject.toml

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

{vectorvein-0.2.41 → vectorvein-0.2.43}/src/vectorvein/chat_clients/anthropic_client.py

@@ -263,13 +263,13 @@ class AnthropicChatClient(BaseChatClient):
                 base_url=self.endpoint.api_base,
                 http_client=self.http_client,
             )
-        elif self.endpoint.api_schema_type == "default":
+        elif self.endpoint.endpoint_type in ("default", "anthropic"):
             return Anthropic(
                 api_key=self.endpoint.api_key,
                 base_url=self.endpoint.api_base,
                 http_client=self.http_client,
             )
-        elif self.endpoint.api_schema_type == "openai":
+        else:
             return OpenAICompatibleChatClient(
                 model=self.model,
                 stream=self.stream,
@@ -455,7 +455,7 @@ class AnthropicChatClient(BaseChatClient):
 
         self.endpoint, self.model_id = self._set_endpoint()
 
-        if self.endpoint.api_schema_type == "openai":
+        if self.endpoint.endpoint_type and self.endpoint.endpoint_type.startswith("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
 
@@ -810,13 +810,13 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
                 aws_region=self.endpoint.region,
                 http_client=self.http_client,
             )
-        elif self.endpoint.api_schema_type == "default":
+        elif self.endpoint.endpoint_type in ("default", "anthropic"):
             return AsyncAnthropic(
                 api_key=self.endpoint.api_key,
                 base_url=self.endpoint.api_base,
                 http_client=self.http_client,
             )
-        elif self.endpoint.api_schema_type == "openai":
+        else:
             return AsyncOpenAICompatibleChatClient(
                 model=self.model,
                 stream=self.stream,
@@ -1002,7 +1002,7 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
 
         self.endpoint, self.model_id = self._set_endpoint()
 
-        if self.endpoint.api_schema_type == "openai":
+        if self.endpoint.endpoint_type and self.endpoint.endpoint_type.startswith("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
 

{vectorvein-0.2.41 → vectorvein-0.2.43}/src/vectorvein/chat_clients/utils.py

@@ -238,7 +238,7 @@ def get_token_counts(text: str | dict, model: str = "", use_token_server_first:
                     or endpoint.endpoint_type == "anthropic_bedrock"
                 ):
                     continue
-                elif endpoint.api_schema_type == "default":
+                elif endpoint.endpoint_type in ("default", "anthropic"):
                     return (
                         Anthropic(
                             api_key=endpoint.api_key,

{vectorvein-0.2.41 → vectorvein-0.2.43}/src/vectorvein/types/defaults.py

@@ -778,6 +778,13 @@ ERNIE_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "function_call_available": False,
         "response_format_available": True,
     },
+    "ernie-4.5-8k-preview": {
+        "id": "ernie-4.5-8k-preview",
+        "context_length": 5120,
+        "max_output_tokens": 2048,
+        "function_call_available": False,
+        "response_format_available": True,
+    },
 }
 
 

{vectorvein-0.2.41 → vectorvein-0.2.43}/src/vectorvein/types/llm_parameters.py

@@ -44,10 +44,6 @@ class EndpointSetting(BaseModel):
         "default",
         description="The type of endpoint. Set to 'default' will determine the type automatically.",
     )
-    api_schema_type: Optional[Literal["default", "openai", "anthropic"]] = Field(
-        "default",
-        description="The type of client for the endpoint. Set to 'default' will determine the type automatically.",
-    )
     credentials: Optional[dict] = Field(None, description="Additional credentials if needed.")
     is_azure: bool = Field(False, description="Indicates if the endpoint is for Azure.")
     is_vertex: bool = Field(False, description="Indicates if the endpoint is for Vertex.")

{vectorvein-0.2.41 → vectorvein-0.2.43}/src/vectorvein/types/settings.py

@@ -80,13 +80,6 @@ class EndpointSettingDict(TypedDict):
             "anthropic_bedrock",
         ]
     ]
-    api_schema_type: NotRequired[
-        Literal[
-            "default",
-            "openai",
-            "anthropic",
-        ]
-    ]
     credentials: NotRequired[dict]
     is_azure: NotRequired[bool]
     is_vertex: NotRequired[bool]