vectorvein 0.2.20__tar.gz → 0.2.21__tar.gz

{vectorvein-0.2.20 → vectorvein-0.2.21}/PKG-INFO

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

{vectorvein-0.2.20 → vectorvein-0.2.21}/pyproject.toml

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

{vectorvein-0.2.20 → vectorvein-0.2.21}/src/vectorvein/settings/__init__.py

@@ -209,5 +209,52 @@ class Settings(BaseModel):
         else:
             return super().model_dump(exclude={"backends"})
 
+    def upgrade_to_v2(self) -> "Settings":
+        """
+        Upgrade settings from v1 format to v2 format.
+        In v2 format, all backend settings are stored in the 'backends' field.
+
+        Returns:
+            Settings: Self with updated format
+        """
+        # If already v2, no need to upgrade
+        if self.VERSION == "2" and self.backends is not None:
+            return self
+
+        # Initialize backends if not exists
+        if self.backends is None:
+            self.backends = Backends()
+
+        # Move all backend settings to backends field
+        backend_names = [
+            "anthropic",
+            "deepseek",
+            "gemini",
+            "groq",
+            "local",
+            "minimax",
+            "mistral",
+            "moonshot",
+            "openai",
+            "qwen",
+            "yi",
+            "zhipuai",
+            "baichuan",
+            "stepfun",
+            "xai",
+            "ernie",
+        ]
+
+        for backend_name in backend_names:
+            backend_setting = getattr(self, backend_name)
+            if backend_setting is not None:
+                setattr(self.backends, backend_name, backend_setting)
+                delattr(self, backend_name)
+
+        # Set version to 2
+        self.VERSION = "2"
+
+        return self
+
 
 settings = Settings()