vectorvein 0.2.83__py3-none-any.whl → 0.2.84__py3-none-any.whl

vectorvein/settings/__init__.py

@@ -60,12 +60,8 @@ class Backends(BaseModel):
 
 
 class Settings(BaseModel):
-    VERSION: Optional[str] = Field(
-        default="2", description="Configuration version. If provided, will use the corresponding format."
-    )
-    endpoints: List[EndpointSetting] = Field(
-        default_factory=list, description="Available endpoints for the LLM service."
-    )
+    VERSION: Optional[str] = Field(default="2", description="Configuration version. If provided, will use the corresponding format.")
+    endpoints: List[EndpointSetting] = Field(default_factory=list, description="Available endpoints for the LLM service.")
     token_server: Optional[Server] = Field(default=None, description="Token server address. Format: host:port")
     rate_limit: Optional[RateLimitConfig] = Field(default=None, description="Rate limit settings.")
 
@@ -73,32 +69,22 @@ class Settings(BaseModel):
     backends: Optional[Backends] = Field(default=None, description="All model backends in one place (V2 format).")
 
     # V1 format: each model backend config
-    anthropic: Optional[BackendSettings] = Field(
-        default_factory=BackendSettings, description="Anthropic models settings."
-    )
-    deepseek: Optional[BackendSettings] = Field(
-        default_factory=BackendSettings, description="Deepseek models settings."
-    )
+    anthropic: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Anthropic models settings.")
+    deepseek: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Deepseek models settings.")
     gemini: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Gemini models settings.")
     groq: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Groq models settings.")
     local: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Local models settings.")
     minimax: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Minimax models settings.")
     mistral: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Mistral models settings.")
-    moonshot: Optional[BackendSettings] = Field(
-        default_factory=BackendSettings, description="Moonshot models settings."
-    )
+    moonshot: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Moonshot models settings.")
     openai: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="OpenAI models settings.")
     qwen: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Qwen models settings.")
     yi: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Yi models settings.")
     zhipuai: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
-    baichuan: Optional[BackendSettings] = Field(
-        default_factory=BackendSettings, description="Baichuan models settings."
-    )
+    baichuan: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Baichuan models settings.")
     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."
-    )
+    ernie: Optional[BackendSettings] = Field(default_factory=BackendSettings, description="Baidu Ernie models settings.")
 
     def __init__(self, **data):
         model_types = {
@@ -143,6 +129,9 @@ class Settings(BaseModel):
                 user_models = backends[model_type].get("models", {})
                 model_settings = BackendSettings()
                 model_settings.update_models(default_models, user_models)
+                default_endpoint = backends[model_type].get("default_endpoint", None)
+                if default_endpoint is not None:
+                    model_settings.default_endpoint = default_endpoint
                 backends[model_type] = model_settings
             else:
                 backends[model_type] = BackendSettings(models=default_models)

vectorvein/types/llm_parameters.py

@@ -101,9 +101,7 @@ class EndpointSetting(BaseModel):
 
 class ModelSetting(BaseModel):
     id: str = Field(..., description="The id of the model.")
-    endpoints: List[Union[str, EndpointOptionDict]] = Field(
-        default_factory=list, description="Available endpoints for the model."
-    )
+    endpoints: List[Union[str, EndpointOptionDict]] = Field(default_factory=list, description="Available endpoints for the model.")
     function_call_available: bool = Field(False, description="Indicates if function call is available.")
     response_format_available: bool = Field(False, description="Indicates if response format is available.")
     native_multimodal: bool = Field(False, description="Indicates if the model is a native multimodal model.")
@@ -113,6 +111,16 @@ class ModelSetting(BaseModel):
 
 class BackendSettings(BaseModel):
     models: Dict[str, ModelSetting] = Field(default_factory=dict)
+    default_endpoint: Optional[str] = Field(default_factory=lambda: None, description="The default endpoint for the model.")
+
+    def get_model_setting(self, model_name: str) -> ModelSetting:
+        if model_name in self.models:
+            model_setting = self.models[model_name]
+            if len(model_setting.endpoints) == 0 and self.default_endpoint is not None:
+                model_setting.endpoints = [self.default_endpoint]
+            return model_setting
+        else:
+            raise ValueError(f"Model {model_name} not found in {self.models}")
 
     def update_models(self, default_models: Dict[str, Dict], input_models: Dict[str, Dict]):
         updated_models = {}

vectorvein/types/settings.py

@@ -61,6 +61,7 @@ class BackendSettingsDict(TypedDict):
     """TypedDict representing the BackendSettings structure."""
 
     models: Dict[str, ModelConfigDict]
+    default_endpoint: NotRequired[Optional[str]]
 
 
 class EndpointSettingDict(TypedDict):

{vectorvein-0.2.83.dist-info → vectorvein-0.2.84.dist-info}/METADATA

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

{vectorvein-0.2.83.dist-info → vectorvein-0.2.84.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.83.dist-info/METADATA,sha256=_p3JvMcCL1vV4pyxPdxzvdtqEuyJk5sTAeLXcCM4AQo,4567
-vectorvein-0.2.83.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-vectorvein-0.2.83.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.84.dist-info/METADATA,sha256=kMut60wbAADgGMP32bt1tHyWzsGbWbIk3ILjGAOvD90,4567
+vectorvein-0.2.84.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+vectorvein-0.2.84.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
@@ -29,15 +29,15 @@ vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9s
 vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
 vectorvein/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/server/token_server.py,sha256=36F9PKSNOX8ZtYBXY_l-76GQTpUSmQ2Y8EMy1H7wtdQ,1353
-vectorvein/settings/__init__.py,sha256=3Kw3hbvqcIQepAR6Q2m2UXbBnwyJTUm8yAz-aHmbUTg,11163
+vectorvein/settings/__init__.py,sha256=_2783EWi-QRKp7OuxDg-rxj-gezfyb1h0hTcb-F4xS4,11260
 vectorvein/settings/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/types/__init__.py,sha256=ypg8c8AwF49FrFBMqmgH_eIBH4LFf0KN4kjqQa7zrvM,3376
 vectorvein/types/defaults.py,sha256=zaUusjyOaPJV-a-e-zOIpNljxk_TSQO5gvnadls-fUY,37065
 vectorvein/types/enums.py,sha256=LplSVkXLBK-t8TWtJKj_f7ktWTd6CSHWRLb67XKMm54,1716
 vectorvein/types/exception.py,sha256=KtnqZ-1DstHm95SZAyZdHhkGq1bJ4A9Aw3Zfdu-VIFo,130
-vectorvein/types/llm_parameters.py,sha256=Bdz9E_x0G96rvJ5TnEFPrU5QV4I2y0YFv7dY4Pq-MuU,7933
+vectorvein/types/llm_parameters.py,sha256=Fo_-_rFkBvl_dUsCZxy2xNMC9Q4qVGLzDBL9jRVbXIQ,8489
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/types/settings.py,sha256=zneZcebts-MHcYS1DXakulA_f0fGPzidOo1s9yl51zI,4794
+vectorvein/types/settings.py,sha256=5wINBiT9qj3ZwDiE6Kf4r6wD3yY1bKN40xxZ7F6jEfI,4844
 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
@@ -65,4 +65,4 @@ vectorvein/workflow/utils/analyse.py,sha256=msmvyz35UTYTwqQR5sg9H0sm1vxmGDSmep9X
 vectorvein/workflow/utils/check.py,sha256=B_NdwqIqnc7Ko2HHqFpfOmWVaAu21tPITe0szKfiZKc,11414
 vectorvein/workflow/utils/json_to_code.py,sha256=P8dhhSNgKhTnW17qXNjLO2aLdb0rA8qMAWxhObol2TU,7295
 vectorvein/workflow/utils/layout.py,sha256=j0bRD3uaXu40xCS6U6BGahBI8FrHa5MiF55GbTrZ1LM,4565
-vectorvein-0.2.83.dist-info/RECORD,,
+vectorvein-0.2.84.dist-info/RECORD,,