vectorvein 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (4) hide show
  1. vectorvein/settings/__init__.py +26 -28
  2. {vectorvein-0.1.6.dist-info → vectorvein-0.1.7.dist-info}/METADATA +1 -1
  3. {vectorvein-0.1.6.dist-info → vectorvein-0.1.7.dist-info}/RECORD +4 -4
  4. {vectorvein-0.1.6.dist-info → vectorvein-0.1.7.dist-info}/WHEEL +0 -0
vectorvein/settings/__init__.py CHANGED
@@ -14,41 +14,39 @@ class Settings(BaseModel):
14
14
  default_factory=list, description="Available endpoints for the LLM service."
15
15
  )
16
16
 
17
- anthropic_models: BackendSettings = Field(
18
- default_factory=BackendSettings, description="Anthropic models settings."
19
- )
20
- deepseek_models: BackendSettings = Field(default_factory=BackendSettings, description="Deepseek models settings.")
21
- gemini_models: BackendSettings = Field(default_factory=BackendSettings, description="Gemini models settings.")
22
- groq_models: BackendSettings = Field(default_factory=BackendSettings, description="Groq models settings.")
23
- local_models: BackendSettings = Field(default_factory=BackendSettings, description="Local models settings.")
24
- minimax_models: BackendSettings = Field(default_factory=BackendSettings, description="Minimax models settings.")
25
- mistral_models: BackendSettings = Field(default_factory=BackendSettings, description="Mistral models settings.")
26
- moonshot_models: BackendSettings = Field(default_factory=BackendSettings, description="Moonshot models settings.")
27
- openai_models: BackendSettings = Field(default_factory=BackendSettings, description="OpenAI models settings.")
28
- qwen_models: BackendSettings = Field(default_factory=BackendSettings, description="Qwen models settings.")
29
- yi_models: BackendSettings = Field(default_factory=BackendSettings, description="Yi models settings.")
30
- zhipuai_models: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
17
+ anthropic: BackendSettings = Field(default_factory=BackendSettings, description="Anthropic models settings.")
18
+ deepseek: BackendSettings = Field(default_factory=BackendSettings, description="Deepseek models settings.")
19
+ gemini: BackendSettings = Field(default_factory=BackendSettings, description="Gemini models settings.")
20
+ groq: BackendSettings = Field(default_factory=BackendSettings, description="Groq models settings.")
21
+ local: BackendSettings = Field(default_factory=BackendSettings, description="Local models settings.")
22
+ minimax: BackendSettings = Field(default_factory=BackendSettings, description="Minimax models settings.")
23
+ mistral: BackendSettings = Field(default_factory=BackendSettings, description="Mistral models settings.")
24
+ moonshot: BackendSettings = Field(default_factory=BackendSettings, description="Moonshot models settings.")
25
+ openai: BackendSettings = Field(default_factory=BackendSettings, description="OpenAI models settings.")
26
+ qwen: BackendSettings = Field(default_factory=BackendSettings, description="Qwen models settings.")
27
+ yi: BackendSettings = Field(default_factory=BackendSettings, description="Yi models settings.")
28
+ zhipuai: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
31
29
 
32
30
  def __init__(self, **data):
33
31
  model_types = {
34
- "anthropic_models": defs.ANTHROPIC_MODELS,
35
- "deepseek_models": defs.DEEPSEEK_MODELS,
36
- "gemini_models": defs.GEMINI_MODELS,
37
- "groq_models": defs.GROQ_MODELS,
38
- "local_models": {},
39
- "minimax_models": defs.MINIMAX_MODELS,
40
- "mistral_models": defs.MISTRAL_MODELS,
41
- "moonshot_models": defs.MOONSHOT_MODELS,
42
- "openai_models": defs.OPENAI_MODELS,
43
- "qwen_models": defs.QWEN_MODELS,
44
- "yi_models": defs.YI_MODELS,
45
- "zhipuai_models": defs.ZHIPUAI_MODELS,
32
+ "anthropic": defs.ANTHROPIC_MODELS,
33
+ "deepseek": defs.DEEPSEEK_MODELS,
34
+ "gemini": defs.GEMINI_MODELS,
35
+ "groq": defs.GROQ_MODELS,
36
+ "local": {},
37
+ "minimax": defs.MINIMAX_MODELS,
38
+ "mistral": defs.MISTRAL_MODELS,
39
+ "moonshot": defs.MOONSHOT_MODELS,
40
+ "openai": defs.OPENAI_MODELS,
41
+ "qwen": defs.QWEN_MODELS,
42
+ "yi": defs.YI_MODELS,
43
+ "zhipuai": defs.ZHIPUAI_MODELS,
46
44
  }
47
45
 
48
46
  for model_type, default_models in model_types.items():
49
47
  if model_type in data:
50
48
  model_settings = BackendSettings()
51
- model_settings.update_models(default_models, data[model_type])
49
+ model_settings.update_models(default_models, data[model_type].get("models", {}))
52
50
  data[model_type] = model_settings
53
51
  else:
54
52
  data[model_type] = BackendSettings(models=default_models)
@@ -65,7 +63,7 @@ class Settings(BaseModel):
65
63
  return EndpointSetting()
66
64
 
67
65
  def get_backend(self, backend: BackendType) -> BackendSettings:
68
- return getattr(self, f"{backend.value.lower()}_models")
66
+ return getattr(self, backend.value.lower())
69
67
 
70
68
 
71
69
  settings = Settings()
{vectorvein-0.1.6.dist-info → vectorvein-0.1.7.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vectorvein
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: Default template for PDM package
5
5
  Author-Email: Anderson <andersonby@163.com>
6
6
  License: MIT
{vectorvein-0.1.6.dist-info → vectorvein-0.1.7.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
- vectorvein-0.1.6.dist-info/METADATA,sha256=cfyPkv1PSdCL5Ve9zAIDTNVzrPDLN2GDbiADGewJN0g,423
2
- vectorvein-0.1.6.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
1
+ vectorvein-0.1.7.dist-info/METADATA,sha256=C_3UCv_92cL58DnFKRMs-1GbEbzGIdDPlNt-X4aIKF4,423
2
+ vectorvein-0.1.7.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
3
3
  vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  vectorvein/chat_clients/__init__.py,sha256=5j7W--jr-l2cqDJp38uXYvkydDK0rnzm7MYGSACHKmU,3976
5
5
  vectorvein/chat_clients/anthropic_client.py,sha256=pkk0yPH05WTOnKOAXkm60ZZw1kdT8pCydNkHx7QArh4,18707
@@ -17,9 +17,9 @@ vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15
17
17
  vectorvein/chat_clients/utils.py,sha256=tAQwfydj46sMxSHeaeOWXrTUY2q0h7482NbvZjbNz9A,17637
18
18
  vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
19
19
  vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
20
- vectorvein/settings/__init__.py,sha256=1__ILDpBh25S43d32knqLBe05tF4AQwT0ZALdBHw1OY,3396
20
+ vectorvein/settings/__init__.py,sha256=4mpccT7eZC3yI1vVnVViW4wHBnDEH9D2R5EsIP34VgU,3218
21
21
  vectorvein/types/defaults.py,sha256=Mg-Mj3_eBzKZn1N8x1V2GqyaYgLD13i-NdSYdQC28X4,11437
22
22
  vectorvein/types/enums.py,sha256=vzOenCnRlFXBwPh-lfFhjGfM-6yfDj7wZColHODqocI,1550
23
23
  vectorvein/types/llm_parameters.py,sha256=nBjStC2zndTY__yhD2WFXB09taxEhDLE3OHA6MICfgE,3494
24
24
  vectorvein/utilities/media_processing.py,sha256=BujciRmw1GMmc3ELRvafL8STcy6r5b2rVnh27-uA7so,2256
25
- vectorvein-0.1.6.dist-info/RECORD,,
25
+ vectorvein-0.1.7.dist-info/RECORD,,