vectorvein 0.2.13__py3-none-any.whl → 0.2.15__py3-none-any.whl

vectorvein/settings/__init__.py

@@ -1,6 +1,7 @@
 # @Author: Bi Ying
 # @Date:   2024-07-27 00:30:56
 import warnings
+from copy import deepcopy
 from typing import List, Optional, Literal
 
 from pydantic import BaseModel, Field
@@ -59,7 +60,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."
@@ -114,8 +115,14 @@ class Settings(BaseModel):
             "xai": defs.XAI_MODELS,
         }
 
+        data = deepcopy(data)
+
         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,13 +131,12 @@ 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:
-                model_settings = BackendSettings()
                 user_models = backends[model_type].get("models", {})
+                model_settings = BackendSettings()
                 model_settings.update_models(default_models, user_models)
                 backends[model_type] = model_settings
             else:
@@ -149,6 +155,10 @@ class Settings(BaseModel):
     def load(self, settings_dict: SettingsDict):
         self.__init__(**settings_dict)
 
+    @classmethod
+    def load_from_dict(cls, settings_dict: SettingsDict):
+        return cls(**settings_dict)
+
     def get_endpoint(self, endpoint_id: str) -> EndpointSetting:
         for endpoint in self.endpoints:
             if endpoint.id == endpoint_id:
@@ -165,5 +175,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/types/__init__.py

@@ -61,8 +61,13 @@ from .settings import (
     BackendSettingsDict,
     EndpointSettingDict,
     SettingsDict,
+    SettingsV1Dict,
+    SettingsV2Dict,
 )
 
+from anthropic.types import ThinkingConfigParam, ThinkingConfigEnabledParam
+from openai.types.chat.completion_create_params import ResponseFormat
+
 
 __all__ = [
     "CONTEXT_LENGTH_CONTROL",
@@ -126,4 +131,9 @@ __all__ = [
     "BackendSettingsDict",
     "EndpointSettingDict",
     "SettingsDict",
+    "SettingsV1Dict",
+    "SettingsV2Dict",
+    "ThinkingConfigParam",
+    "ThinkingConfigEnabledParam",
+    "ResponseFormat",
 ]

vectorvein/types/defaults.py

@@ -176,62 +176,6 @@ GROQ_MODELS: Final[Dict[str, Dict[str, Any]]] = {
 # Qwen models
 QWEN_DEFAULT_MODEL: Final[str] = "qwen2.5-72b-instruct"
 QWEN_MODELS: Final[Dict[str, Dict[str, Any]]] = {
-    "qwen1.5-1.8b-chat": {
-        "id": "qwen1.5-1.8b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen1.5-4b-chat": {
-        "id": "qwen1.5-4b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen1.5-7b-chat": {
-        "id": "qwen1.5-7b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen1.5-14b-chat": {
-        "id": "qwen1.5-14b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen1.5-32b-chat": {
-        "id": "qwen1.5-32b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen1.5-72b-chat": {
-        "id": "qwen1.5-72b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen1.5-110b-chat": {
-        "id": "qwen1.5-110b-chat",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
-    "qwen2-72b-instruct": {
-        "id": "qwen2-72b-instruct",
-        "context_length": 30000,
-        "max_output_tokens": 4096,
-        "function_call_available": False,
-        "response_format_available": True,
-    },
     "qwen2.5-7b-instruct": {
         "id": "qwen2.5-7b-instruct",
         "context_length": 131072,
@@ -282,6 +226,30 @@ QWEN_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "response_format_available": False,
         "native_multimodal": True,
     },
+    "qwen2.5-vl-72b-instruct": {
+        "id": "qwen2.5-vl-72b-instruct",
+        "context_length": 131072,
+        "max_output_tokens": 8192,
+        "function_call_available": False,
+        "response_format_available": False,
+        "native_multimodal": True,
+    },
+    "qwen2.5-vl-7b-instruct": {
+        "id": "qwen2.5-vl-7b-instruct",
+        "context_length": 131072,
+        "max_output_tokens": 8192,
+        "function_call_available": False,
+        "response_format_available": False,
+        "native_multimodal": True,
+    },
+    "qwen2.5-vl-3b-instruct": {
+        "id": "qwen2.5-vl-3b-instruct",
+        "context_length": 131072,
+        "max_output_tokens": 8192,
+        "function_call_available": False,
+        "response_format_available": False,
+        "native_multimodal": True,
+    },
     "qwen-max": {
         "id": "qwen-max",
         "context_length": 8096,

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]

{vectorvein-0.2.13.dist-info → vectorvein-0.2.15.dist-info}/METADATA

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

{vectorvein-0.2.13.dist-info → vectorvein-0.2.15.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.13.dist-info/METADATA,sha256=sEZT2h2XwDUIkIvdKzXomeSuniLUSr-HWlj88MmHRew,4414
-vectorvein-0.2.13.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.13.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.15.dist-info/METADATA,sha256=meMymU-CMwxurlkRGFjRPsQZkrSEFDLy8kfns0k7OZQ,4414
+vectorvein-0.2.15.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.15.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
@@ -28,15 +28,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=oBOLwG61RN9dq012Bgq3OfZyxyrJ5j0BYNmHbThhpjo,8097
+vectorvein/settings/__init__.py,sha256=x1V0zHMLmafoqpLlEr4_LSZ0nxQj_LDoGRB1pKUiMO4,9030
 vectorvein/settings/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/types/__init__.py,sha256=ie7H3rTMq_Fg836vOmy96m3wzjDkqfekQecPXXEDbcM,3005
-vectorvein/types/defaults.py,sha256=VrkQoyHqC_eK3g1b6egpPYLLo0ltwMHqxDscCX4y-N0,27417
+vectorvein/types/__init__.py,sha256=DJYGhlshgUQgzXPfMfKW5sTpBClZUCLhqmCqF44lVuU,3329
+vectorvein/types/defaults.py,sha256=hdCEY1bEPmPbsh335bXd4LWMeJXLWD3_EulURaWs4EU,26430
 vectorvein/types/enums.py,sha256=7KTJSVtQueImmbr1fSwv3rQVtc0RyMWXJmoE2tDOaso,1667
 vectorvein/types/exception.py,sha256=gnW4GnJ76jND6UGnodk9xmqkcbeS7Cz2rvncA2HpD5E,69
 vectorvein/types/llm_parameters.py,sha256=4SxDbJKVb9oGYymyxQtNZ66YZmUQd9_CpYYg81_Inkk,7650
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/types/settings.py,sha256=096mcaqtuZYCeJQYzF4ILqrLp_z5-upJAxwLrRZ5WQk,4165
+vectorvein/types/settings.py,sha256=B8eVJsJRrv-op8Enn-Rbour2b9vD01TLXz3xlXEF9vo,4485
 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.13.dist-info/RECORD,,
+vectorvein-0.2.15.dist-info/RECORD,,