vectorvein 0.2.15__tar.gz → 0.2.17__tar.gz

{vectorvein-0.2.15 → vectorvein-0.2.17}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.2.15
-Summary: VectorVein python SDK
+Version: 0.2.17
+Summary: VectorVein Python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT
 Requires-Python: >=3.10

{vectorvein-0.2.15 → vectorvein-0.2.17}/pyproject.toml

@@ -14,11 +14,11 @@ dependencies = [
     "google-auth>=2.35.0",
     "pycryptodome>=3.21.0",
 ]
-description = "VectorVein python SDK"
+description = "VectorVein Python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.2.15"
+version = "0.2.17"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.15 → vectorvein-0.2.17}/src/vectorvein/chat_clients/__init__.py

@@ -10,6 +10,7 @@ from .xai_client import XAIChatClient, AsyncXAIChatClient
 from .groq_client import GroqChatClient, AsyncGroqChatClient
 from .qwen_client import QwenChatClient, AsyncQwenChatClient
 from .local_client import LocalChatClient, AsyncLocalChatClient
+from .ernie_client import ErnieChatClient, AsyncErnieChatClient
 from .gemini_client import GeminiChatClient, AsyncGeminiChatClient
 from .openai_client import OpenAIChatClient, AsyncOpenAIChatClient
 from .zhipuai_client import ZhiPuAIChatClient, AsyncZhiPuAIChatClient
@@ -49,6 +50,7 @@ BackendMap = {
         BackendType.Baichuan: BaichuanChatClient,
         BackendType.StepFun: StepFunChatClient,
         BackendType.XAI: XAIChatClient,
+        BackendType.Ernie: ErnieChatClient,
     },
     "async": {
         BackendType.Anthropic: AsyncAnthropicChatClient,
@@ -66,6 +68,7 @@ BackendMap = {
         BackendType.Baichuan: AsyncBaichuanChatClient,
         BackendType.StepFun: AsyncStepFunChatClient,
         BackendType.XAI: AsyncXAIChatClient,
+        BackendType.Ernie: AsyncErnieChatClient,
     },
 }
 
@@ -295,6 +298,21 @@ def create_chat_client(
 ) -> XAIChatClient: ...
 
 
+@overload
+def create_chat_client(
+    backend: Literal[BackendType.Ernie],
+    model: str | None = None,
+    stream: bool = False,
+    temperature: float | None | NotGiven = NOT_GIVEN,
+    context_length_control: ContextLengthControlType = defs.CONTEXT_LENGTH_CONTROL,
+    random_endpoint: bool = True,
+    endpoint_id: str = "",
+    http_client: httpx.Client | None = None,
+    settings: "Settings | SettingsDict | None" = None,  # Use default settings if not provided
+    **kwargs,
+) -> ErnieChatClient: ...
+
+
 @overload
 def create_chat_client(
     backend: BackendType,
@@ -566,6 +584,21 @@ def create_async_chat_client(
 ) -> AsyncXAIChatClient: ...
 
 
+@overload
+def create_async_chat_client(
+    backend: Literal[BackendType.Ernie],
+    model: str | None = None,
+    stream: bool = False,
+    temperature: float | None | NotGiven = NOT_GIVEN,
+    context_length_control: ContextLengthControlType = defs.CONTEXT_LENGTH_CONTROL,
+    random_endpoint: bool = True,
+    endpoint_id: str = "",
+    http_client: httpx.AsyncClient | None = None,
+    settings: "Settings | SettingsDict | None" = None,  # Use default settings if not provided
+    **kwargs,
+) -> AsyncErnieChatClient: ...
+
+
 @overload
 def create_async_chat_client(
     backend: BackendType,

vectorvein-0.2.17/src/vectorvein/chat_clients/ernie_client.py

@@ -0,0 +1,15 @@
+# @Author: Bi Ying
+# @Date:   2024-07-26 14:48:55
+from ..types.enums import BackendType
+from ..types.defaults import ERNIE_DEFAULT_MODEL
+from .openai_compatible_client import OpenAICompatibleChatClient, AsyncOpenAICompatibleChatClient
+
+
+class ErnieChatClient(OpenAICompatibleChatClient):
+    DEFAULT_MODEL = ERNIE_DEFAULT_MODEL
+    BACKEND_NAME = BackendType.Ernie
+
+
+class AsyncErnieChatClient(AsyncOpenAICompatibleChatClient):
+    DEFAULT_MODEL = ERNIE_DEFAULT_MODEL
+    BACKEND_NAME = BackendType.Ernie

{vectorvein-0.2.15 → vectorvein-0.2.17}/src/vectorvein/settings/__init__.py

@@ -56,6 +56,7 @@ class Backends(BaseModel):
     baichuan: BackendSettings = Field(default_factory=BackendSettings, description="Baichuan models settings.")
     stepfun: BackendSettings = Field(default_factory=BackendSettings, description="StepFun models settings.")
     xai: BackendSettings = Field(default_factory=BackendSettings, description="XAI models settings.")
+    ernie: BackendSettings = Field(default_factory=BackendSettings, description="Baidu Ernie models settings.")
 
 
 class Settings(BaseModel):
@@ -95,6 +96,9 @@ class Settings(BaseModel):
     )
     stepfun: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="StepFun models settings.")
     xai: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="XAI models settings.")
+    ernie: Optional[BackendSettings] = Field(
+        default_factory=BackendSettings, description="Baidu Ernie models settings."
+    )
 
     def __init__(self, **data):
         model_types = {
@@ -113,6 +117,7 @@ class Settings(BaseModel):
             "baichuan": defs.BAICHUAN_MODELS,
             "stepfun": defs.STEPFUN_MODELS,
             "xai": defs.XAI_MODELS,
+            "ernie": defs.ERNIE_MODELS,
         }
 
         data = deepcopy(data)
@@ -152,7 +157,11 @@ class Settings(BaseModel):
 
         super().__init__(**data)
 
-    def load(self, settings_dict: SettingsDict):
+    def load(self, settings: SettingsDict | "Settings"):
+        if isinstance(settings, Settings):
+            settings_dict = settings.model_dump()
+        else:
+            settings_dict = settings
         self.__init__(**settings_dict)
 
     @classmethod
@@ -194,6 +203,7 @@ class Settings(BaseModel):
                     "baichuan",
                     "stepfun",
                     "xai",
+                    "ernie",
                 }
             )
         else:

{vectorvein-0.2.15 → vectorvein-0.2.17}/src/vectorvein/types/defaults.py

@@ -739,7 +739,14 @@ ERNIE_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "id": "ernie-4.0-8k-latest",
         "context_length": 5120,
         "max_output_tokens": 2048,
-        "function_call_available": False,
+        "function_call_available": True,
+        "response_format_available": True,
+    },
+    "ernie-4.0-8k": {
+        "id": "ernie-4.0-8k",
+        "context_length": 5120,
+        "max_output_tokens": 2048,
+        "function_call_available": True,
         "response_format_available": True,
     },
     "ernie-4.0-turbo-8k": {

{vectorvein-0.2.15 → vectorvein-0.2.17}/src/vectorvein/types/enums.py

@@ -29,7 +29,7 @@ class BackendType(str, Enum):
     # DeepSeek
     DeepSeek = "deepseek"
 
-    # Qwen
+    # Aliyun Qwen
     Qwen = "qwen"
 
     # Groq
@@ -53,6 +53,9 @@ class BackendType(str, Enum):
     # XAI
     XAI = "xai"
 
+    # Baidu Ernie
+    Ernie = "ernie"
+
     def __repr__(self):
         """Get a string representation."""
         return f'"{self.value}"'

{vectorvein-0.2.15 → vectorvein-0.2.17}/src/vectorvein/types/settings.py

@@ -99,6 +99,7 @@ class BackendsDict(TypedDict):
     baichuan: NotRequired[BackendSettingsDict]
     stepfun: NotRequired[BackendSettingsDict]
     xai: NotRequired[BackendSettingsDict]
+    ernie: NotRequired[BackendSettingsDict]
 
 
 class SettingsV1Dict(TypedDict):
@@ -124,6 +125,7 @@ class SettingsV1Dict(TypedDict):
     baichuan: NotRequired[BackendSettingsDict]
     stepfun: NotRequired[BackendSettingsDict]
     xai: NotRequired[BackendSettingsDict]
+    ernie: NotRequired[BackendSettingsDict]
 
 
 class SettingsV2Dict(TypedDict):