vectorvein 0.1.6__tar.gz → 0.1.7__tar.gz

{vectorvein-0.1.6 → vectorvein-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.6
+Version: 0.1.7
 Summary: Default template for PDM package
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.6 → vectorvein-0.1.7}/pyproject.toml

@@ -14,7 +14,7 @@ description = "Default template for PDM package"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.6"
+version = "0.1.7"
 
 [project.license]
 text = "MIT"

vectorvein-0.1.7/src/vectorvein/settings/__init__.py

@@ -0,0 +1,69 @@
+# @Author: Bi Ying
+# @Date:   2024-07-27 00:30:56
+from typing import List, Dict
+
+from pydantic import BaseModel, Field
+
+from ..types import defaults as defs
+from ..types.enums import BackendType
+from ..types.llm_parameters import BackendSettings, EndpointSetting
+
+
+class Settings(BaseModel):
+    endpoints: List[EndpointSetting] = Field(
+        default_factory=list, description="Available endpoints for the LLM service."
+    )
+
+    anthropic: BackendSettings = Field(default_factory=BackendSettings, description="Anthropic models settings.")
+    deepseek: BackendSettings = Field(default_factory=BackendSettings, description="Deepseek models settings.")
+    gemini: BackendSettings = Field(default_factory=BackendSettings, description="Gemini models settings.")
+    groq: BackendSettings = Field(default_factory=BackendSettings, description="Groq models settings.")
+    local: BackendSettings = Field(default_factory=BackendSettings, description="Local models settings.")
+    minimax: BackendSettings = Field(default_factory=BackendSettings, description="Minimax models settings.")
+    mistral: BackendSettings = Field(default_factory=BackendSettings, description="Mistral models settings.")
+    moonshot: BackendSettings = Field(default_factory=BackendSettings, description="Moonshot models settings.")
+    openai: BackendSettings = Field(default_factory=BackendSettings, description="OpenAI models settings.")
+    qwen: BackendSettings = Field(default_factory=BackendSettings, description="Qwen models settings.")
+    yi: BackendSettings = Field(default_factory=BackendSettings, description="Yi models settings.")
+    zhipuai: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
+
+    def __init__(self, **data):
+        model_types = {
+            "anthropic": defs.ANTHROPIC_MODELS,
+            "deepseek": defs.DEEPSEEK_MODELS,
+            "gemini": defs.GEMINI_MODELS,
+            "groq": defs.GROQ_MODELS,
+            "local": {},
+            "minimax": defs.MINIMAX_MODELS,
+            "mistral": defs.MISTRAL_MODELS,
+            "moonshot": defs.MOONSHOT_MODELS,
+            "openai": defs.OPENAI_MODELS,
+            "qwen": defs.QWEN_MODELS,
+            "yi": defs.YI_MODELS,
+            "zhipuai": defs.ZHIPUAI_MODELS,
+        }
+
+        for model_type, default_models in model_types.items():
+            if model_type in data:
+                model_settings = BackendSettings()
+                model_settings.update_models(default_models, data[model_type].get("models", {}))
+                data[model_type] = model_settings
+            else:
+                data[model_type] = BackendSettings(models=default_models)
+
+        super().__init__(**data)
+
+    def load(self, settings_dict: Dict):
+        self.__init__(**settings_dict)
+
+    def get_endpoint(self, endpoint_id: str) -> EndpointSetting:
+        for endpoint in self.endpoints:
+            if endpoint.id == endpoint_id:
+                return endpoint
+        return EndpointSetting()
+
+    def get_backend(self, backend: BackendType) -> BackendSettings:
+        return getattr(self, backend.value.lower())
+
+
+settings = Settings()