vectorvein 0.2.11__py3-none-any.whl → 0.2.12__py3-none-any.whl

vectorvein/chat_clients/openai_compatible_client.py

@@ -99,7 +99,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
             return AzureOpenAI(
                 azure_endpoint=self.endpoint.api_base,
                 api_key=self.endpoint.api_key,
-                api_version="2024-12-01-preview",
+                api_version="2025-01-01-preview",
                 http_client=self.http_client,
             )
         else:
@@ -574,7 +574,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
             return AsyncAzureOpenAI(
                 azure_endpoint=self.endpoint.api_base,
                 api_key=self.endpoint.api_key,
-                api_version="2024-12-01-preview",
+                api_version="2025-01-01-preview",
                 http_client=self.http_client,
             )
         else:

vectorvein/types/llm_parameters.py

@@ -2,6 +2,7 @@
 # @Date:   2024-07-26 23:48:04
 from typing import List, Dict, Optional, Union, Iterable
 
+import httpx
 from pydantic import BaseModel, Field
 
 from anthropic._types import NotGiven as AnthropicNotGiven
@@ -10,6 +11,7 @@ from anthropic.types import ToolParam as AnthropicToolParam
 from anthropic.types import ThinkingConfigParam, ThinkingConfigEnabledParam
 from anthropic.types.message_create_params import ToolChoice as AnthropicToolChoice
 
+from openai import OpenAI, AsyncOpenAI, AzureOpenAI, AsyncAzureOpenAI
 from openai._types import NotGiven as OpenAINotGiven
 from openai._types import NOT_GIVEN as OPENAI_NOT_GIVEN
 from openai.types.chat.completion_create_params import ResponseFormat
@@ -45,6 +47,48 @@ class EndpointSetting(BaseModel):
     )
     proxy: Optional[str] = Field(None, description="The proxy URL for the endpoint.")
 
+    def model_list(self):
+        http_client = httpx.Client(proxy=self.proxy) if self.proxy is not None else None
+
+        if self.is_azure:
+            if self.api_base is None:
+                raise ValueError("Azure endpoint is not set")
+            _client = AzureOpenAI(
+                azure_endpoint=self.api_base,
+                api_key=self.api_key,
+                api_version="2025-01-01-preview",
+                http_client=http_client,
+            )
+        else:
+            _client = OpenAI(
+                api_key=self.api_key,
+                base_url=self.api_base,
+                http_client=http_client,
+            )
+
+        return _client.models.list().model_dump()
+
+    async def amodel_list(self):
+        http_client = httpx.AsyncClient(proxy=self.proxy) if self.proxy is not None else None
+
+        if self.is_azure:
+            if self.api_base is None:
+                raise ValueError("Azure endpoint is not set")
+            _client = AsyncAzureOpenAI(
+                azure_endpoint=self.api_base,
+                api_key=self.api_key,
+                api_version="2025-01-01-preview",
+                http_client=http_client,
+            )
+        else:
+            _client = AsyncOpenAI(
+                api_key=self.api_key,
+                base_url=self.api_base,
+                http_client=http_client,
+            )
+
+        return (await _client.models.list()).model_dump()
+
 
 class ModelSetting(BaseModel):
     id: str = Field(..., description="The id of the model.")

vectorvein/types/settings.py

@@ -21,8 +21,8 @@ class RateLimitConfigDict(TypedDict):
 
     enabled: bool
     backend: Literal["memory", "redis", "diskcache"]
-    redis: Optional[RedisConfigDict]
-    diskcache: Optional[DiskCacheConfigDict]
+    redis: NotRequired[RedisConfigDict]
+    diskcache: NotRequired[DiskCacheConfigDict]
     default_rpm: int
     default_tpm: int
 
@@ -32,7 +32,7 @@ class ServerDict(TypedDict):
 
     host: str
     port: int
-    url: Optional[str]
+    url: NotRequired[str]
 
 
 class EndpointOptionDict(TypedDict):
@@ -48,13 +48,13 @@ class EndpointOptionDict(TypedDict):
 class ModelConfigDict(TypedDict):
     """TypedDict representing the model configuration structure."""
 
-    id: str
+    id: NotRequired[str]
     endpoints: List[Union[str, EndpointOptionDict]]
-    function_call_available: bool
-    response_format_available: bool
-    native_multimodal: bool
-    context_length: int
-    max_output_tokens: Optional[int]
+    function_call_available: NotRequired[bool]
+    response_format_available: NotRequired[bool]
+    native_multimodal: NotRequired[bool]
+    context_length: NotRequired[int]
+    max_output_tokens: NotRequired[int]
 
 
 class BackendSettingsDict(TypedDict):
@@ -67,39 +67,39 @@ class EndpointSettingDict(TypedDict):
     """TypedDict representing the EndpointSetting structure."""
 
     id: str
-    api_base: Optional[str]
-    api_key: str
-    region: Optional[str]
-    api_schema_type: Optional[str]
-    credentials: Optional[dict]
-    is_azure: Optional[bool]
-    is_vertex: Optional[bool]
-    is_bedrock: Optional[bool]
-    rpm: Optional[int]
-    tpm: Optional[int]
-    concurrent_requests: Optional[int]
-    proxy: Optional[str]
+    api_base: NotRequired[Optional[str]]
+    api_key: NotRequired[str]
+    region: NotRequired[str]
+    api_schema_type: NotRequired[str]
+    credentials: NotRequired[dict]
+    is_azure: NotRequired[bool]
+    is_vertex: NotRequired[bool]
+    is_bedrock: NotRequired[bool]
+    rpm: NotRequired[int]
+    tpm: NotRequired[int]
+    concurrent_requests: NotRequired[int]
+    proxy: NotRequired[str]
 
 
 class SettingsDict(TypedDict):
     """TypedDict representing the expected structure of the settings dictionary."""
 
     endpoints: List[EndpointSettingDict]
-    token_server: Optional[ServerDict]
-    rate_limit: Optional[RateLimitConfigDict]
+    token_server: NotRequired[ServerDict]
+    rate_limit: NotRequired[RateLimitConfigDict]
     # 各模型后端配置
-    anthropic: BackendSettingsDict
-    deepseek: BackendSettingsDict
-    gemini: BackendSettingsDict
-    groq: BackendSettingsDict
-    local: BackendSettingsDict
-    minimax: BackendSettingsDict
-    mistral: BackendSettingsDict
-    moonshot: BackendSettingsDict
-    openai: BackendSettingsDict
-    qwen: BackendSettingsDict
-    yi: BackendSettingsDict
-    zhipuai: BackendSettingsDict
-    baichuan: BackendSettingsDict
-    stepfun: BackendSettingsDict
-    xai: BackendSettingsDict
+    anthropic: NotRequired[BackendSettingsDict]
+    deepseek: NotRequired[BackendSettingsDict]
+    gemini: NotRequired[BackendSettingsDict]
+    groq: NotRequired[BackendSettingsDict]
+    local: NotRequired[BackendSettingsDict]
+    minimax: NotRequired[BackendSettingsDict]
+    mistral: NotRequired[BackendSettingsDict]
+    moonshot: NotRequired[BackendSettingsDict]
+    openai: NotRequired[BackendSettingsDict]
+    qwen: NotRequired[BackendSettingsDict]
+    yi: NotRequired[BackendSettingsDict]
+    zhipuai: NotRequired[BackendSettingsDict]
+    baichuan: NotRequired[BackendSettingsDict]
+    stepfun: NotRequired[BackendSettingsDict]
+    xai: NotRequired[BackendSettingsDict]

{vectorvein-0.2.11.dist-info → vectorvein-0.2.12.dist-info}/METADATA

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

{vectorvein-0.2.11.dist-info → vectorvein-0.2.12.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.11.dist-info/METADATA,sha256=gSQgVDuAhXDNgYkTNflojyYrSZudUpV76aDOSrNsb_g,4414
-vectorvein-0.2.11.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.11.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.12.dist-info/METADATA,sha256=tceoZCMvKLrKIQMxR0iZlKF1WHIHxgYS5Kqmozm0b8U,4414
+vectorvein-0.2.12.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.12.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/api/__init__.py,sha256=lfY-XA46fgD2iIZTU0VYP8i07AwA03Egj4Qua0vUKrQ,738
 vectorvein/api/client.py,sha256=xF-leKDQzVyyy9FnIRaz0k4eElYW1XbbzeRLcpnyk90,33047
@@ -18,7 +18,7 @@ vectorvein/chat_clients/minimax_client.py,sha256=YOILWcsHsN5tihLTMbKJIyJr9TJREMI
 vectorvein/chat_clients/mistral_client.py,sha256=1aKSylzBDaLYcFnaBIL4-sXSzWmXfBeON9Q0rq-ziWw,534
 vectorvein/chat_clients/moonshot_client.py,sha256=gbu-6nGxx8uM_U2WlI4Wus881rFRotzHtMSoYOcruGU,526
 vectorvein/chat_clients/openai_client.py,sha256=Nz6tV45pWcsOupxjnsRsGTicbQNJWIZyxuJoJ5DGMpg,527
-vectorvein/chat_clients/openai_compatible_client.py,sha256=9ldl6TPinXLLH1kPtP-pVYO2D-VBy_s1Gf-XNNdd3CY,48498
+vectorvein/chat_clients/openai_compatible_client.py,sha256=L8SXCRA7OO_eXh6b-oya8k6XZZJ-j8SrJSvy8BFhCgs,48498
 vectorvein/chat_clients/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15uPoJQOHA0,513
 vectorvein/chat_clients/stepfun_client.py,sha256=zsD2W5ahmR4DD9cqQTXmJr3txrGuvxbRWhFlRdwNijI,519
@@ -34,9 +34,9 @@ vectorvein/types/__init__.py,sha256=ie7H3rTMq_Fg836vOmy96m3wzjDkqfekQecPXXEDbcM,
 vectorvein/types/defaults.py,sha256=VrkQoyHqC_eK3g1b6egpPYLLo0ltwMHqxDscCX4y-N0,27417
 vectorvein/types/enums.py,sha256=7KTJSVtQueImmbr1fSwv3rQVtc0RyMWXJmoE2tDOaso,1667
 vectorvein/types/exception.py,sha256=gnW4GnJ76jND6UGnodk9xmqkcbeS7Cz2rvncA2HpD5E,69
-vectorvein/types/llm_parameters.py,sha256=8ot02N3PS794bJTc0jlPk_XoXDGWQDTxyyF-1NmqVn0,6103
+vectorvein/types/llm_parameters.py,sha256=4SxDbJKVb9oGYymyxQtNZ66YZmUQd9_CpYYg81_Inkk,7650
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/types/settings.py,sha256=hzLtE7ryyIIGxcExAsqh9sMNKP8-r4WczhYWIBNm-uM,2880
+vectorvein/types/settings.py,sha256=71D94qqG4PVY59NDp1M29ZGcINOByilGcnCrNs5rbX8,3214
 vectorvein/utilities/media_processing.py,sha256=7KtbLFzOYIn1e9QTN9G6C76NH8CBlV9kfAgiRKEIeXY,6263
 vectorvein/utilities/rate_limiter.py,sha256=dwolIUVw2wP83Odqpx0AAaE77de1GzxkYDGH4tM_u_4,10300
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
@@ -61,4 +61,4 @@ vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs0
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=LsqomfXfqrXfHJDO1cl0Ox48f4St7X_SL12DSbAMSOw,5415
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.2.11.dist-info/RECORD,,
+vectorvein-0.2.12.dist-info/RECORD,,