vectorvein 0.1.52__tar.gz → 0.1.53__tar.gz

{vectorvein-0.1.52 → vectorvein-0.1.53}/PKG-INFO

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

{vectorvein-0.1.52 → vectorvein-0.1.53}/pyproject.toml

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

{vectorvein-0.1.52 → vectorvein-0.1.53}/src/vectorvein/chat_clients/__init__.py

@@ -6,6 +6,7 @@ from typing import overload, Literal
 from .base_client import BaseChatClient, BaseAsyncChatClient
 
 from .yi_client import YiChatClient, AsyncYiChatClient
+from .xai_client import XAIChatClient, AsyncXAIChatClient
 from .groq_client import GroqChatClient, AsyncGroqChatClient
 from .qwen_client import QwenChatClient, AsyncQwenChatClient
 from .local_client import LocalChatClient, AsyncLocalChatClient
@@ -41,6 +42,7 @@ BackendMap = {
         BackendType.ZhiPuAI: ZhiPuAIChatClient,
         BackendType.Baichuan: BaichuanChatClient,
         BackendType.StepFun: StepFunChatClient,
+        BackendType.XAI: XAIChatClient,
     },
     "async": {
         BackendType.Anthropic: AsyncAnthropicChatClient,
@@ -57,6 +59,7 @@ BackendMap = {
         BackendType.ZhiPuAI: AsyncZhiPuAIChatClient,
         BackendType.Baichuan: AsyncBaichuanChatClient,
         BackendType.StepFun: AsyncStepFunChatClient,
+        BackendType.XAI: AsyncXAIChatClient,
     },
 }
 
@@ -257,6 +260,20 @@ def create_chat_client(
 ) -> StepFunChatClient: ...
 
 
+@overload
+def create_chat_client(
+    backend: Literal[BackendType.XAI],
+    model: str | None = None,
+    stream: bool = False,
+    temperature: float = 0.7,
+    context_length_control: ContextLengthControlType = defs.CONTEXT_LENGTH_CONTROL,
+    random_endpoint: bool = True,
+    endpoint_id: str = "",
+    http_client: httpx.Client | None = None,
+    **kwargs,
+) -> XAIChatClient: ...
+
+
 @overload
 def create_chat_client(
     backend: BackendType,
@@ -496,6 +513,20 @@ def create_async_chat_client(
 ) -> AsyncStepFunChatClient: ...
 
 
+@overload
+def create_async_chat_client(
+    backend: Literal[BackendType.XAI],
+    model: str | None = None,
+    stream: bool = False,
+    temperature: float = 0.7,
+    context_length_control: ContextLengthControlType = defs.CONTEXT_LENGTH_CONTROL,
+    random_endpoint: bool = True,
+    endpoint_id: str = "",
+    http_client: httpx.AsyncClient | None = None,
+    **kwargs,
+) -> AsyncXAIChatClient: ...
+
+
 @overload
 def create_async_chat_client(
     backend: BackendType,

vectorvein-0.1.53/src/vectorvein/chat_clients/xai_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 XAI_DEFAULT_MODEL
+from .openai_compatible_client import OpenAICompatibleChatClient, AsyncOpenAICompatibleChatClient
+
+
+class XAIChatClient(OpenAICompatibleChatClient):
+    DEFAULT_MODEL = XAI_DEFAULT_MODEL
+    BACKEND_NAME = BackendType.XAI
+
+
+class AsyncXAIChatClient(AsyncOpenAICompatibleChatClient):
+    DEFAULT_MODEL = XAI_DEFAULT_MODEL
+    BACKEND_NAME = BackendType.XAI

{vectorvein-0.1.52 → vectorvein-0.1.53}/src/vectorvein/settings/__init__.py

@@ -35,6 +35,7 @@ class Settings(BaseModel):
     zhipuai: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
     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.")
 
     def __init__(self, **data):
         model_types = {

{vectorvein-0.1.52 → vectorvein-0.1.53}/src/vectorvein/types/defaults.py

@@ -724,3 +724,14 @@ STEPFUN_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "native_multimodal": True,
     },
 }
+
+
+XAI_DEFAULT_MODEL: Final[str] = "grok-beta"
+XAI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
+    "grok-beta": {
+        "id": "grok-beta",
+        "context_length": 131072,
+        "function_call_available": True,
+        "response_format_available": True,
+    },
+}

{vectorvein-0.1.52 → vectorvein-0.1.53}/src/vectorvein/types/enums.py

@@ -50,6 +50,9 @@ class BackendType(str, Enum):
     # StepFun
     StepFun = "stepfun"
 
+    # XAI
+    XAI = "xai"
+
     def __repr__(self):
         """Get a string representation."""
         return f'"{self.value}"'