vectorvein 0.1.30__tar.gz → 0.1.32__tar.gz

{vectorvein-0.1.30 → vectorvein-0.1.32}/PKG-INFO

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

{vectorvein-0.1.30 → vectorvein-0.1.32}/pyproject.toml

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

{vectorvein-0.1.30 → vectorvein-0.1.32}/src/vectorvein/chat_clients/__init__.py

@@ -257,6 +257,20 @@ def create_chat_client(
 ) -> StepFunChatClient: ...
 
 
+@overload
+def create_chat_client(
+    backend: BackendType,
+    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,
+) -> BaseChatClient: ...
+
+
 def create_chat_client(
     backend: BackendType,
     model: str | None = None,
@@ -482,6 +496,20 @@ def create_async_chat_client(
 ) -> AsyncStepFunChatClient: ...
 
 
+@overload
+def create_async_chat_client(
+    backend: BackendType,
+    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,
+) -> BaseAsyncChatClient: ...
+
+
 def create_async_chat_client(
     backend: BackendType,
     model: str | None = None,

{vectorvein-0.1.30 → vectorvein-0.1.32}/src/vectorvein/types/defaults.py

@@ -498,6 +498,22 @@ OPENAI_MODELS = {
         "function_call_available": True,
         "response_format_available": True,
     },
+    "o1-mini": {
+        "id": "o1-mini",
+        "context_length": 128000,
+        "max_output_tokens": 65536,
+        "function_call_available": False,
+        "response_format_available": False,
+        "native_multimodal": False,
+    },
+    "o1-preview": {
+        "id": "o1-preview",
+        "context_length": 128000,
+        "max_output_tokens": 32768,
+        "function_call_available": False,
+        "response_format_available": False,
+        "native_multimodal": False,
+    },
 }
 
 # Anthropic models