vectorvein 0.2.13__tar.gz → 0.2.14__tar.gz

{vectorvein-0.2.13 → vectorvein-0.2.14}/PKG-INFO

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

{vectorvein-0.2.13 → vectorvein-0.2.14}/pyproject.toml

@@ -18,7 +18,7 @@ description = "VectorVein python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.2.13"
+version = "0.2.14"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.13 → vectorvein-0.2.14}/src/vectorvein/settings/__init__.py

@@ -59,7 +59,7 @@ class Backends(BaseModel):
 
 class Settings(BaseModel):
     VERSION: Optional[str] = Field(
-        default=None, description="Configuration version. If provided, will use the corresponding format."
+        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."
@@ -116,6 +116,10 @@ class Settings(BaseModel):
 
         version = data.get("VERSION")
 
+        if len(data) == 0:
+            version = "2"
+            data["backends"] = {}
+
         # If V2 format, model configs are in the backends dictionary
         if version == "2":
             if "backends" not in data:
@@ -124,8 +128,7 @@ class Settings(BaseModel):
             backends = data["backends"]
         else:
             backends = data
-            if len(data) > 0:
-                warnings.warn("You're using vectorvein's deprecated V1 format. Please use V2 format.")
+            warnings.warn("You're using vectorvein's deprecated V1 format. Please use V2 format.")
 
         for model_type, default_models in model_types.items():
             if model_type in backends:
@@ -165,5 +168,29 @@ class Settings(BaseModel):
         # Compatible with VERSION 1 format
         return getattr(self, backend_name)
 
+    def export(self):
+        if self.VERSION == "2":
+            return super().model_dump(
+                exclude={
+                    "anthropic",
+                    "deepseek",
+                    "gemini",
+                    "groq",
+                    "local",
+                    "minimax",
+                    "mistral",
+                    "moonshot",
+                    "openai",
+                    "qwen",
+                    "yi",
+                    "zhipuai",
+                    "baichuan",
+                    "stepfun",
+                    "xai",
+                }
+            )
+        else:
+            return super().model_dump(exclude={"backends"})
+
 
 settings = Settings()

{vectorvein-0.2.13 → vectorvein-0.2.14}/src/vectorvein/types/__init__.py

@@ -61,6 +61,8 @@ from .settings import (
     BackendSettingsDict,
     EndpointSettingDict,
     SettingsDict,
+    SettingsV1Dict,
+    SettingsV2Dict,
 )
 
 
@@ -126,4 +128,6 @@ __all__ = [
     "BackendSettingsDict",
     "EndpointSettingDict",
     "SettingsDict",
+    "SettingsV1Dict",
+    "SettingsV2Dict",
 ]

{vectorvein-0.2.13 → vectorvein-0.2.14}/src/vectorvein/types/settings.py

@@ -101,17 +101,13 @@ class BackendsDict(TypedDict):
     xai: NotRequired[BackendSettingsDict]
 
 
-class SettingsDict(TypedDict):
+class SettingsV1Dict(TypedDict):
     """TypedDict representing the expected structure of the settings dictionary."""
 
-    VERSION: NotRequired[str]
     endpoints: List[EndpointSettingDict]
     token_server: NotRequired[ServerDict]
     rate_limit: NotRequired[RateLimitConfigDict]
 
-    # V2 format: all model backend configs in a single dictionary
-    backends: NotRequired[BackendsDict]
-
     # V1 format: each model backend config
     anthropic: NotRequired[BackendSettingsDict]
     deepseek: NotRequired[BackendSettingsDict]
@@ -128,3 +124,18 @@ class SettingsDict(TypedDict):
     baichuan: NotRequired[BackendSettingsDict]
     stepfun: NotRequired[BackendSettingsDict]
     xai: NotRequired[BackendSettingsDict]
+
+
+class SettingsV2Dict(TypedDict):
+    """TypedDict representing the expected structure of the settings dictionary."""
+
+    VERSION: NotRequired[str]
+    endpoints: List[EndpointSettingDict]
+    token_server: NotRequired[ServerDict]
+    rate_limit: NotRequired[RateLimitConfigDict]
+
+    # V2 format: all model backend configs in a single dictionary
+    backends: NotRequired[BackendsDict]
+
+
+SettingsDict = Union[SettingsV1Dict, SettingsV2Dict]