vectorvein 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

vectorvein/chat_clients/__init__.py

@@ -13,6 +13,7 @@ from .openai_client import OpenAIChatClient, AsyncOpenAIChatClient
 from .zhipuai_client import ZhiPuAIChatClient, AsyncZhiPuAIChatClient
 from .minimax_client import MiniMaxChatClient, AsyncMiniMaxChatClient
 from .mistral_client import MistralChatClient, AsyncMistralChatClient
+from .baichuan_client import BaichuanChatClient, AsyncBaichuanChatClient
 from .moonshot_client import MoonshotChatClient, AsyncMoonshotChatClient
 from .deepseek_client import DeepSeekChatClient, AsyncDeepSeekChatClient
 
@@ -36,6 +37,7 @@ BackendMap = {
         BackendType.Qwen: QwenChatClient,
         BackendType.Yi: YiChatClient,
         BackendType.ZhiPuAI: ZhiPuAIChatClient,
+        BackendType.Baichuan: BaichuanChatClient,
     },
     "async": {
         BackendType.Anthropic: AsyncAnthropicChatClient,
@@ -50,6 +52,7 @@ BackendMap = {
         BackendType.Qwen: AsyncQwenChatClient,
         BackendType.Yi: AsyncYiChatClient,
         BackendType.ZhiPuAI: AsyncZhiPuAIChatClient,
+        BackendType.Baichuan: AsyncBaichuanChatClient,
     },
 }
 

vectorvein/chat_clients/baichuan_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 BAICHUAN_DEFAULT_MODEL
+from .openai_compatible_client import OpenAICompatibleChatClient, AsyncOpenAICompatibleChatClient
+
+
+class BaichuanChatClient(OpenAICompatibleChatClient):
+    DEFAULT_MODEL = BAICHUAN_DEFAULT_MODEL
+    BACKEND_NAME = BackendType.Baichuan
+
+
+class AsyncBaichuanChatClient(AsyncOpenAICompatibleChatClient):
+    DEFAULT_MODEL = BAICHUAN_DEFAULT_MODEL
+    BACKEND_NAME = BackendType.Baichuan

vectorvein/chat_clients/base_client.py

@@ -52,6 +52,7 @@ class BaseChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> ChatCompletionMessage | Generator[ChatCompletionDeltaMessage, Any, None]:
         pass
 
@@ -63,6 +64,7 @@ class BaseChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> Generator[ChatCompletionDeltaMessage, Any, None]:
         return self.create_completion(
             messages=messages,
@@ -72,6 +74,7 @@ class BaseChatClient(ABC):
             max_tokens=max_tokens,
             tools=tools,
             tool_choice=tool_choice,
+            **kwargs,
         )
 
 
@@ -115,6 +118,7 @@ class BaseAsyncChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> ChatCompletionMessage | AsyncGenerator[ChatCompletionDeltaMessage, None]:
         pass
 
@@ -126,6 +130,7 @@ class BaseAsyncChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> AsyncGenerator[ChatCompletionDeltaMessage, None]:
         return await self.create_completion(
             messages=messages,
@@ -135,4 +140,5 @@ class BaseAsyncChatClient(ABC):
             max_tokens=max_tokens,
             tools=tools,
             tool_choice=tool_choice,
+            **kwargs,
         )

vectorvein/settings/__init__.py

@@ -26,6 +26,7 @@ class Settings(BaseModel):
     qwen: BackendSettings = Field(default_factory=BackendSettings, description="Qwen models settings.")
     yi: BackendSettings = Field(default_factory=BackendSettings, description="Yi models settings.")
     zhipuai: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
+    baichuan: BackendSettings = Field(default_factory=BackendSettings, description="Baichuan models settings.")
 
     def __init__(self, **data):
         model_types = {
@@ -41,6 +42,7 @@ class Settings(BaseModel):
             "qwen": defs.QWEN_MODELS,
             "yi": defs.YI_MODELS,
             "zhipuai": defs.ZHIPUAI_MODELS,
+            "baichuan": defs.BAICHUAN_MODELS,
         }
 
         for model_type, default_models in model_types.items():

vectorvein/types/defaults.py

@@ -52,6 +52,46 @@ DEEPSEEK_MODELS = {
 }
 DEEPSEEK_DEFAULT_MODEL = "deepseek-chat"
 
+# Baichuan models
+BAICHUAN_MODELS = {
+    "Baichuan4": {
+        "id": "Baichuan4",
+        "context_length": 32768,
+        "max_output_tokens": 2048,
+        "function_call_available": True,
+        "response_format_available": True,
+    },
+    "Baichuan3-Turbo": {
+        "id": "Baichuan3-Turbo",
+        "context_length": 32768,
+        "max_output_tokens": 2048,
+        "function_call_available": True,
+        "response_format_available": True,
+    },
+    "Baichuan3-Turbo-128k": {
+        "id": "Baichuan3-Turbo-128k",
+        "context_length": 128000,
+        "max_output_tokens": 2048,
+        "function_call_available": True,
+        "response_format_available": True,
+    },
+    "Baichuan2-Turbo": {
+        "id": "Baichuan2-Turbo",
+        "context_length": 32768,
+        "max_output_tokens": 2048,
+        "function_call_available": True,
+        "response_format_available": False,
+    },
+    "Baichuan2-53B": {
+        "id": "Baichuan2-53B",
+        "context_length": 32768,
+        "max_output_tokens": 2048,
+        "function_call_available": False,
+        "response_format_available": False,
+    },
+}
+BAICHUAN_DEFAULT_MODEL = "Baichuan3-Turbo"
+
 # Groq models
 GROQ_DEFAULT_MODEL = "llama3-70b-8192"
 GROQ_MODELS = {

vectorvein/types/enums.py

@@ -44,6 +44,9 @@ class BackendType(str, Enum):
     # Gemini
     Gemini = "gemini"
 
+    # Baichuan
+    Baichuan = "baichuan"
+
     def __repr__(self):
         """Get a string representation."""
         return f'"{self.value}"'

{vectorvein-0.1.12.dist-info → vectorvein-0.1.14.dist-info}/METADATA

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

{vectorvein-0.1.12.dist-info → vectorvein-0.1.14.dist-info}/RECORD

@@ -1,9 +1,10 @@
-vectorvein-0.1.12.dist-info/METADATA,sha256=OxvgLVzQ5ilVePTHSj5ZhU9aKdyb7nXq7A7UNyDFuzo,502
-vectorvein-0.1.12.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
+vectorvein-0.1.14.dist-info/METADATA,sha256=Xm4rgWhrckl6Jik4gaGQDx6KC85l4lDFcy2UToEj5no,502
+vectorvein-0.1.14.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/chat_clients/__init__.py,sha256=rur7elYPL-HIQM46U5Kb8aO4tMXfYOOKLdIHFlKIw0M,4421
+vectorvein/chat_clients/__init__.py,sha256=oaEcDx5MM89oBOqmvWWDGAkMoaHihh892cA0tJU2u4g,4599
 vectorvein/chat_clients/anthropic_client.py,sha256=sECXynbReJs1xixFEHSlfEQo9Q1zGf5uCJ-V8emdoAM,20438
-vectorvein/chat_clients/base_client.py,sha256=CscLWDNd8pE0EKthvRRtxyjMhGGt7pwP4FOts9KNOLU,4469
+vectorvein/chat_clients/baichuan_client.py,sha256=CVMvpgjdrZGv0BWnTOBD-f2ufZ3wq3496wqukumsAr4,526
+vectorvein/chat_clients/base_client.py,sha256=4nGY5A7s7Ubgqb7XqFGkYwM6RNYDTuWkHZrH5KTJZiE,4585
 vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
 vectorvein/chat_clients/gemini_client.py,sha256=W-9Vu-GTE9wxStPznyNR0rBEgDG3LYBu2uQXd4sh1YQ,14425
 vectorvein/chat_clients/groq_client.py,sha256=Uow4pgdmFi93ZQSoOol2-0PhhqkW-S0XuSldvppz5U4,498
@@ -17,10 +18,10 @@ vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15
 vectorvein/chat_clients/utils.py,sha256=mnAew2Ie3nQHdEyDLKuJvXkQ5QdcSAJ6SpYk5JPbR1Q,20888
 vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
 vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
-vectorvein/settings/__init__.py,sha256=4mpccT7eZC3yI1vVnVViW4wHBnDEH9D2R5EsIP34VgU,3218
-vectorvein/types/defaults.py,sha256=lCzGOLybX8FzHX-Cv32BaQFZ8sHvPhGIIwDD-VksP20,13460
-vectorvein/types/enums.py,sha256=vzOenCnRlFXBwPh-lfFhjGfM-6yfDj7wZColHODqocI,1550
+vectorvein/settings/__init__.py,sha256=jVHbhHn1BuMcyfZGXrxWKiI4NdY9wzvYyGMvKYmUtqg,3378
+vectorvein/types/defaults.py,sha256=bwE-MLIuiDqMiOGsstacaUTGD-bN5pKL6lGP-gS-Kmg,14667
+vectorvein/types/enums.py,sha256=PNK_pTIyjJFy-yAG2PHaMIO1ey3W6fReMCkH8M8VRW4,1595
 vectorvein/types/llm_parameters.py,sha256=PWN18dDGrCTP4Bz7pX0XxO-wDUA7qTngppzEELrROmc,3496
 vectorvein/utilities/media_processing.py,sha256=BujciRmw1GMmc3ELRvafL8STcy6r5b2rVnh27-uA7so,2256
 vectorvein/utilities/retry.py,sha256=9ePuJdeUUGx-qMWfaFxmlOvG_lQPwCQ4UB1z3Edlo34,993
-vectorvein-0.1.12.dist-info/RECORD,,
+vectorvein-0.1.14.dist-info/RECORD,,